arXiv:2508.09014cs.CV2025-08被引 13

通过不确定性感知的交叉训练,提升医学图像分割的半监督性能。

Uncertainty-aware Cross-training for Semi-supervised Medical Image Segmentation

  • 设计双子网结构,用一致性约束减少模型认知偏差。
  • 基于不确定图生成高置信伪标签,提升未标注数据利用效率。
  • 在多模态医学影像上表现优异,适合标注稀缺场景使用。

半监督学习因能降低对专家标注的依赖,在医学图像分割中备受关注。现有基于均值教师(MT)的方法虽利用一致性正则化挖掘未标注数据信息,但过度依赖学生模型,忽视了模型内存在的认知偏差。部分方法采用伪标签进行协同训练,但在扰动输入下生成高置信伪标签仍具挑战。本文提出不确定性感知的交叉训练框架UC-Seg,通过两个不同子网协同建模,有效挖掘二者间关联性,缓解模型偏差。提出跨子网一致性保持(CCP)策略,增强特征表示能力并确保双子网特征一致性,使每个子网可自我修正并学习共享语义。同时,设计不确定性感知伪标签生成(UPG)组件,结合双子网的分割结果与对应不确定性图,生成高置信伪标签。在包含MRI、CT、超声、肠镜等多模态医学图像的任务上广泛验证,结果表明该方法在分割精度和泛化能力上均优于现有先进半监督方法。代码将公开于https://github.com/taozh2017/UCSeg。

原文摘要 · Abstract (English)

Semi-supervised learning has gained considerable popularity in medical image segmentation tasks due to its capability to reduce reliance on expert-examined annotations. Several mean-teacher (MT) based semi-supervised methods utilize consistency regularization to effectively leverage valuable information from unlabeled data. However, these methods often heavily rely on the student model and overlook the potential impact of cognitive biases within the model. Furthermore, some methods employ co-training using pseudo-labels derived from different inputs, yet generating high-confidence pseudo-labels from perturbed inputs during training remains a significant challenge. In this paper, we propose an Uncertainty-aware Cross-training framework for semi-supervised medical image Segmentation (UC-Seg). Our UC-Seg framework incorporates two distinct subnets to effectively explore and leverage the correlation between them, thereby mitigating cognitive biases within the model. Specifically, we present a Cross-subnet Consistency Preservation (CCP) strategy to enhance feature representation capability and ensure feature consistency across the two subnets. This strategy enables each subnet to correct its own biases and learn shared semantics from both labeled and unlabeled data. Additionally, we propose an Uncertainty-aware Pseudo-label Generation (UPG) component that leverages segmentation results and corresponding uncertainty maps from both subnets to generate high-confidence pseudo-labels. We extensively evaluate the proposed UC-Seg on various medical image segmentation tasks involving different modality images, such as MRI, CT, ultrasound, colonoscopy, and so on. The results demonstrate that our method achieves superior segmentation accuracy and generalization performance compared to other state-of-the-art semi-supervised methods. Our code will be released at https://github.com/taozh2017/UCSeg.

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

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