通过互补网络与证据融合,提升医疗图像分割中伪标签的可靠性。
Mutual Evidential Deep Learning for Medical Image Segmentation
- 用不同结构网络生成互补证据,改进伪标签融合方式。
- 在五个主流数据集上达到当前最佳性能,显著提升分割精度。
- 适合需要高可靠伪标签的医疗影像分割场景。
现有半监督医学图像分割的协同学习框架因低质量伪标签导致模型识别偏差而性能受限。由于其伪标签融合策略具有平均特性,无法挖掘来自不同来源伪标签的可靠性。本文提出一种相互证据深度学习(MEDL)框架,从两个角度解决伪标签生成问题:首先,采用不同架构的网络为无标签样本生成互补证据,并引入改进的类别感知证据融合机制,指导来自多种网络的证据预测的可信合成;其次,利用融合证据中的不确定性,设计基于渐近Fisher信息的证据学习策略,使模型初期聚焦于伪标签更可靠的样本,逐步关注低质量伪标签样本,同时避免在高不确定性样本中过度惩罚误标类别。对于有标签数据,仍采用不确定性驱动的渐近学习策略,逐步引导模型关注挑战性体素。在五个主流数据集上的大量实验表明,MEDL实现当前最优性能。
原文摘要 · Abstract (English)
Existing semi-supervised medical segmentation co-learning frameworks have realized that model performance can be diminished by the biases in model recognition caused by low-quality pseudo-labels. Due to the averaging nature of their pseudo-label integration strategy, they fail to explore the reliability of pseudo-labels from different sources. In this paper, we propose a mutual evidential deep learning (MEDL) framework that offers a potentially viable solution for pseudo-label generation in semi-supervised learning from two perspectives. First, we introduce networks with different architectures to generate complementary evidence for unlabeled samples and adopt an improved class-aware evidential fusion to guide the confident synthesis of evidential predictions sourced from diverse architectural networks. Second, utilizing the uncertainty in the fused evidence, we design an asymptotic Fisher information-based evidential learning strategy. This strategy enables the model to initially focus on unlabeled samples with more reliable pseudo-labels, gradually shifting attention to samples with lower-quality pseudo-labels while avoiding over-penalization of mislabeled classes in high data uncertainty samples. Additionally, for labeled data, we continue to adopt an uncertainty-driven asymptotic learning strategy, gradually guiding the model to focus on challenging voxels. Extensive experiments on five mainstream datasets have demonstrated that MEDL achieves state-of-the-art performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。