arXiv:2502.07457cs.CV2025-02

通过双向不确定性学习,提升半监督医学图像分割精度

Bidirectional Uncertainty-Aware Region Learning for Semi-Supervised Medical Image Segmentation

  • 在高不确定区域用真实标签指导模型,低不确定区用伪标签训练
  • 实验显示该方法在多个医学图像分割任务中性能显著提升
  • 适合处理标注少、噪声多的医疗影像分割场景

在半监督医学图像分割中,未标注数据质量差且模型预测存在不确定性,导致伪标签错误累积,削弱模型性能。我们发现这些错误伪标签通常集中在高不确定性区域。传统方法直接丢弃这些区域的伪标签,可能损失有价值的数据。为此,我们提出双向不确定性感知区域学习策略:在标注数据训练时关注高不确定性区域,利用精确标签引导模型学习;在未标注数据训练时聚焦低不确定性区域,减少错误伪标签干扰。该双向学习策略显著提升了模型整体性能。大量实验表明,该方法在不同医学图像分割任务中均取得显著改进。

原文摘要 · Abstract (English)

In semi-supervised medical image segmentation, the poor quality of unlabeled data and the uncertainty in the model's predictions lead to models that inevitably produce erroneous pseudo-labels. These errors accumulate throughout model training, thereby weakening the model's performance. We found that these erroneous pseudo-labels are typically concentrated in high-uncertainty regions. Traditional methods improve performance by directly discarding pseudo-labels in these regions, which can also result in neglecting potentially valuable training data. To alleviate this problem, we propose a bidirectional uncertainty-aware region learning strategy to fully utilize the precise supervision provided by labeled data and stabilize the training of unlabeled data. Specifically, in the training labeled data, we focus on high-uncertainty regions, using precise label information to guide the model's learning in potentially uncontrollable areas. Meanwhile, in the training of unlabeled data, we concentrate on low-uncertainty regions to reduce the interference of erroneous pseudo-labels on the model. Through this bidirectional learning strategy, the model's overall performance has significantly improved. Extensive experiments show that our proposed method achieves significant performance improvement on different medical image segmentation tasks.

医学图像半监督分割不确定性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。