国内刊号:31-1289/TP
国际刊号:1000-3428
发布日期:
作者:刘丽霞, 宣士斌, 刘畅, 李嘉祥
单位:广西民族大学 人工智能学院, 南宁 530006
关键词:不确定性估计,多解码器,多专家注释,视杯视盘分割,软注意机制
基金:国家自然科学基金(61866003)。
现有基于深度学习的视杯和视盘分割方法在模型训练时,仅使用图像的单个注释或从多个注释中获取唯一的注释信息,忽略原始多专家标注中嵌入的一致性或差异性信息,从而导致模型和预测结果过度自信等问题。提出一种基于多解码器不确定性感知体系的模型MUA-Net。通过引入专业知识推断模块,将各个专家注释的专业知识水平作为先验知识嵌入编码器和解码器的瓶颈中,以形成包含专家线索的高级语义特征。利用可同时学习多个注释的多解码器结构调节多专家之间的分歧,重构多专家注释过程,并对不确定或分歧区域进行量化。提出一种双分支软注意机制,增强多解码器分割预测的模糊区域,得到最终校准的分割结果。实验结果表明,该模型在RIGA数据集上能以较高的不确定性预测合理的区域,与MRNet模型相比,该模型在视杯分割中的平均精度、Dice系数、交并比分别提升了0.75、0.39、0.41个百分点。
来源:2023年第1期
《计算机工程》期刊编辑部