arXiv:2601.14039cs.CVcs.AI2026-01

让医学图像分割模型学会主动忽略错误标注,提升抗噪声能力。

Generalizing Abstention for Noise-Robust Learning in Medical Image Segmentation

  • 引入可适配多种损失函数的拒识框架,自动识别并跳过噪声样本。
  • 在高噪声条件下,新方法在两个医学数据集上显著优于传统模型。
  • 适合需要高可靠性、标注质量不高的医疗影像分析场景。

标签噪声是医学图像分割中的关键挑战,常源于人工标注的困难。在含噪数据上训练的模型易过拟合,影响泛化性能。尽管已有若干方法缓解分割任务中的标签噪声问题,该领域仍研究不足。拒识机制在分类任务中已被证明能增强交叉熵表现,但在分割中尚未验证。本文提出一种通用且模块化的拒识框架,可提升多种损失函数的抗噪能力。框架包含两项改进:用于引导拒识行为的有信息正则项,以及基于幂律的自适应调节算法以优化拒识惩罚。我们系统性地将该框架与三种不同损失函数结合,构建出三种新型抗噪变体:GAC、SAC 和 ADS。在 CaDIS 与 DSAD 医学数据集上的实验表明,新方法在高噪声条件下始终显著优于非拒识基线。本工作证实,允许模型选择性忽略错误样本是一种强大且通用的策略,有助于构建更可靠的分割模型。代码已公开于 https://github.com/wemous/abstention-for-segmentation。

原文摘要 · Abstract (English)

Label noise is a critical problem in medical image segmentation, often arising from the inherent difficulty of manual annotation. Models trained on noisy data are prone to overfitting, which degrades their generalization performance. While a number of methods and strategies have been proposed to mitigate noisy labels in the segmentation domain, this area remains largely under-explored. The abstention mechanism has proven effective in classification tasks by enhancing the capabilities of Cross Entropy, yet its potential in segmentation remains unverified. In this paper, we address this gap by introducing a universal and modular abstention framework capable of enhancing the noise-robustness of a diverse range of loss functions. Our framework improves upon prior work with two key components: an informed regularization term to guide abstention behaviour, and a more flexible power-law-based auto-tuning algorithm for the abstention penalty. We demonstrate the framework's versatility by systematically integrating it with three distinct loss functions to create three novel, noise-robust variants: GAC, SAC, and ADS. Experiments on the CaDIS and DSAD medical datasets show our methods consistently and significantly outperform their non-abstaining baselines, especially under high noise levels. This work establishes that enabling models to selectively ignore corrupted samples is a powerful and generalizable strategy for building more reliable segmentation models. Our code is publicly available at https://github.com/wemous/abstention-for-segmentation.

医学图像分割抗噪声拒识

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