用分割质量评估指导半监督学习,提升医学图像分割精度。
Quality-Guided Semi-Supervised Learning for Medical Image Segmentation

- 训练专用网络预测分割质量,捕捉真实错误模式。
- 在五个数据集上显著优于现有半监督方法,提升分割精度。
- 适合需要高质量标注的医学图像分割场景。
训练精准的医学图像分割模型需大量密集标注数据,获取成本高且耗时。半监督学习(SSL)通过结合大量未标注数据与少量标注数据缓解此问题。然而,现有方法依赖伪标签,其可靠性通常基于模型置信度或不确定性,这些指标自我参照,缺乏对分割质量的明确依据。为此,我们提出一种质量引导的半监督学习框架,训练专用网络从图像-掩码对中估计分割质量。该预测器在通过合成扰动及部分训练模型的不完美输出生成的变质掩码上进行训练,捕获训练过程中遇到的真实错误模式。我们将质量预测器融入SSL,采用两种互补机制:质量感知正则化损失和基于质量的伪标签样本重加权方案。实验表明,该方法可作为现有SSL框架的即插即用增强模块。在五个数据集和多种架构上的广泛实验验证了其一致性提升,推动了半监督医学图像分割的最新水平。
原文摘要 · Abstract (English)
Training accurate medical image segmentation models requires large amounts of densely annotated data, which is costly and time-consuming to obtain. Semi-supervised learning (SSL) alleviates this by learning from both abundant unlabeled data and limited labeled data. However, most modern SSL methods rely on pseudolabels for unlabeled data, and typically assess their reliability through model confidence or uncertainty, measures that are self-referential and lack explicit grounding in segmentation quality. Instead, we propose a quality-guided SSL framework that trains a dedicated network to estimate segmentation quality from image-mask pairs. The predictor is trained on variable-quality masks generated through synthetic corruptions augmented with imperfect outputs from partially trained segmentation models, capturing realistic error patterns encountered during training. We integrate the quality predictor into SSL through two complementary mechanisms: a quality-aware regularization loss and a quality-based pseudolabel sample reweighting scheme. We show that our method serves as a drop-in enhancement to existing SSL frameworks. Extensive experiments across five datasets and multiple architectures demonstrate consistent improvements over competing SSL methods, advancing the state-of-the-art in semi-supervised medical image segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。