Quality-Guided Semi-Supervised Learning for Medical Image Segmentation
质量引导的半监督学习用于医学图像分割
Kumar Abhishek, Ghassan Hamarneh
AI总结 提出一种质量引导的半监督学习框架,通过专用网络估计分割质量,并利用质量感知正则化和伪标签重加权提升医学图像分割性能。
详情
- Comments
- Early Accept at MICCAI 2026, 13 pages, 2 figures
训练准确的医学图像分割模型需要大量密集标注的数据,这既昂贵又耗时。半监督学习通过从大量未标注数据和少量标注数据中学习来缓解这一问题。然而,大多数现代半监督学习方法依赖未标注数据的伪标签,并通常通过模型置信度或不确定性来评估其可靠性,这些度量是自我指涉的,缺乏对分割质量的明确基础。相反,我们提出了一种质量引导的半监督学习框架,训练一个专用网络从图像-掩膜对中估计分割质量。该预测器在通过合成损坏生成的变质量掩膜上进行训练,这些损坏结合了部分训练分割模型产生的不完美输出,捕捉训练中遇到的真实错误模式。我们通过两种互补机制将质量预测器集成到半监督学习中:质量感知正则化损失和基于质量的伪标签样本重新加权方案。我们表明,我们的方法可以作为现有半监督学习框架的即插即用增强。在五个数据集和多种架构上的大量实验表明,与竞争性的半监督学习方法相比,我们的方法取得了一致的改进,推进了半监督医学图像分割的最新水平。
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.