arXiv:2509.13084cs.CV2025-09被引 4

用不确定性引导伪标签,提升少样本医学图像分割精度

Enhancing Dual Network Based Semi-Supervised Medical Image Segmentation with Uncertainty-Guided Pseudo-Labeling

  • 双网络架构结合交叉一致性与熵过滤,降低伪标签噪声
  • 在左心房数据集仅用10%标注数据时达到89.95%的Dice分数
  • 适合标注稀缺的医疗影像分割场景,尤其适用于3D体积数据

尽管监督式医学图像分割模型表现优异,但其依赖大量标注数据在真实场景中不现实。半监督学习通过生成伪标签利用未标注数据缓解此问题。然而现有方法仍存在伪标签噪声和特征空间监督不足的问题。本文提出一种基于双网络架构的新型3D医学图像分割框架。设计跨一致性增强模块,结合交叉伪标签与熵过滤监督以减少噪声;引入不确定性感知的动态权重策略(基于Kullback-Leibler散度)调节伪标签贡献;并采用自监督对比学习机制,将不确定体素特征对齐至可靠类别原型,有效区分可信与不确定预测,从而降低预测不确定性。在左心房、NIH胰腺和BraTS-2019三个3D分割数据集上进行大量实验,所提方法在多种设置下均优于当前最优方法,例如在左心房数据集仅使用10%标注数据时取得89.95%的Dice分数。消融实验进一步验证了各模块的有效性。

原文摘要 · Abstract (English)

Despite the remarkable performance of supervised medical image segmentation models, relying on a large amount of labeled data is impractical in real-world situations. Semi-supervised learning approaches aim to alleviate this challenge using unlabeled data through pseudo-label generation. Yet, existing semi-supervised segmentation methods still suffer from noisy pseudo-labels and insufficient supervision within the feature space. To solve these challenges, this paper proposes a novel semi-supervised 3D medical image segmentation framework based on a dual-network architecture. Specifically, we investigate a Cross Consistency Enhancement module using both cross pseudo and entropy-filtered supervision to reduce the noisy pseudo-labels, while we design a dynamic weighting strategy to adjust the contributions of pseudo-labels using an uncertainty-aware mechanism (i.e., Kullback-Leibler divergence). In addition, we use a self-supervised contrastive learning mechanism to align uncertain voxel features with reliable class prototypes by effectively differentiating between trustworthy and uncertain predictions, thus reducing prediction uncertainty. Extensive experiments are conducted on three 3D segmentation datasets, Left Atrial, NIH Pancreas and BraTS-2019. The proposed approach consistently exhibits superior performance across various settings (e.g., 89.95\% Dice score on left Atrial with 10\% labeled data) compared to the state-of-the-art methods. Furthermore, the usefulness of the proposed modules is further validated via ablation experiments.

医学图像分割半监督学习伪标签3D分割

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