动态感知不确定性的新方法,提升医学图像分割的准确性与鲁棒性
DyCON: Dynamic Uncertainty-aware Consistency and Contrastive Learning for Semi-supervised Medical Image Segmentation
- 根据体素不确定性动态调整一致性损失权重,保留高不确定性区域
- 在4个数据集上超越现有最优方法,显著改善小病灶分割效果
- 适合处理病理差异大、类别不平衡的3D医学图像分割任务
医学图像分割中的半监督学习通过利用未标注数据减轻标注负担,但现有方法在应对类别不平衡和病理变异带来的高不确定性时表现不佳,尤其在3D医学图像中导致分割不准确。为此,我们提出DyCON框架,采用两种互补的损失函数:不确定性感知一致性损失(UnCL)和焦点熵感知对比损失(FeCL)。UnCL基于体素不确定性动态调整一致性损失贡献,初期降低对不确定体素的惩罚以鼓励模型探索困难区域,随训练进展转向对置信体素施加更高惩罚以确保全局一致性。FeCL引入双重焦点机制和自适应置信度调整,强化不平衡区域的局部特征区分能力,优先关注难样本对并捕捉细微病变差异。在四个不同数据集(ISLES'22、BraTS'19、LA、Pancreas)上的广泛评估表明,DyCON性能优于当前最先进方法。
原文摘要 · Abstract (English)
Semi-supervised learning in medical image segmentation leverages unlabeled data to reduce annotation burdens through consistency learning. However, current methods struggle with class imbalance and high uncertainty from pathology variations, leading to inaccurate segmentation in 3D medical images. To address these challenges, we present DyCON, a Dynamic Uncertainty-aware Consistency and Contrastive Learning framework that enhances the generalization of consistency methods with two complementary losses: Uncertainty-aware Consistency Loss (UnCL) and Focal Entropy-aware Contrastive Loss (FeCL). UnCL enforces global consistency by dynamically weighting the contribution of each voxel to the consistency loss based on its uncertainty, preserving high-uncertainty regions instead of filtering them out. Initially, UnCL prioritizes learning from uncertain voxels with lower penalties, encouraging the model to explore challenging regions. As training progress, the penalty shift towards confident voxels to refine predictions and ensure global consistency. Meanwhile, FeCL enhances local feature discrimination in imbalanced regions by introducing dual focal mechanisms and adaptive confidence adjustments into the contrastive principle. These mechanisms jointly prioritizes hard positives and negatives while focusing on uncertain sample pairs, effectively capturing subtle lesion variations under class imbalance. Extensive evaluations on four diverse medical image segmentation datasets (ISLES'22, BraTS'19, LA, Pancreas) show DyCON's superior performance against SOTA methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。