arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2109.07045eess.IVcs.AIcs.CV
收藏

多解码器U-Net医学图像分割中的不确定性量化

Uncertainty Quantification in Medical Image Segmentation with Multi-decoder U-Net

Harbin Institute of Technology at Shenzhen(哈尔滨工业大学(深圳));Peng Cheng Lab(鹏城实验室);MindsGo Co.Ltd(MindsGo 有限公司);Capital Medical University(首都医科大学);Xuanwu Hospital Capital Medical University(首都医科大学宣武医院)

Yanwu Yang, Xutao Guo, Yiwei Pan, Pengcheng Shi, Haiyan Lv, Ting Ma

摘要:

准确的医学图像分割是诊断和分析的关键。然而,没有校准的不确定性估计的模型可能会导致下游分析中的错误,并表现出较低的鲁棒性。估计测量中的不确定性对于做出明确的、有根据的结论至关重要。特别是,很难对模型和放射科医生的模糊区域和焦点边界进行准确预测,甚至更难在多个注释中达成共识。在这项工作中,这些领域下的不确定性进行了研究,这与解剖结构的重要信息,是同样重要的分割性能。我们利用医学图像分割的不确定性量化通过测量分割性能与多个注释在监督学习的方式,并提出了一个基于U-Net的架构与多个解码器,其中的图像表示是用相同的编码器编码,和分割参考每个注释估计与多个解码器。然而,交叉损失函数的建议,弥合不同分支之间的差距。所提出的架构以端到端的方式进行训练,能够提高预测不确定性估计。该模型以更少的参数实现了与在MICCAI-QUBIQ 2020挑战赛中排名亚军的综合训练模型相当的性能。

英文摘要:

Accurate medical image segmentation is crucial for diagnosis and analysis. However, the models without calibrated uncertainty estimates might lead to errors in downstream analysis and exhibit low levels of robustness. Estimating the uncertainty in the measurement is vital to making definite, informed conclusions. Especially, it is difficult to make accurate predictions on ambiguous areas and focus boundaries for both models and radiologists, even harder to reach a consensus with multiple annotations. In this work, the uncertainty under these areas is studied, which introduces significant information with anatomical structure and is as important as segmentation performance. We exploit the medical image segmentation uncertainty quantification by measuring segmentation performance with multiple annotations in a supervised learning manner and propose a U-Net based architecture with multiple decoders, where the image representation is encoded with the same encoder, and segmentation referring to each annotation is estimated with multiple decoders. Nevertheless, a cross-loss function is proposed for bridging the gap between different branches. The proposed architecture is trained in an end-to-end manner and able to improve predictive uncertainty estimates. The model achieves comparable performance with fewer parameters to the integrated training model that ranked the runner-up in the MICCAI-QUBIQ 2020 challenge.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑