arXiv:2502.11456cs.CVcs.AI2025-02被引 9

用多原型协作修正伪标签,提升少标注3D医学图像分割精度

Leveraging Labelled Data Knowledge: A Cooperative Rectification Learning Network for Semi-supervised 3D Medical Image Segmentation

  • 构建多原型网络,动态修正体素级伪标签
  • 在三个公开数据集上达到领先性能,显著减少标注依赖
  • 适合需要降低标注成本的医疗影像分割研究者

半监督3D医学图像分割旨在仅使用少量标注数据和大量未标注数据实现精准分割。其核心挑战在于如何有效利用未标注数据。本文提出一种新方法,通过生成高质量伪标签来增强一致性学习。首先,引入协作修正学习网络(CRLN),为每类学习多个原型作为外部先验知识,自适应地在体素级别修正伪标签。其次,设计动态交互模块(DIM),促进原型与多尺度图像特征间的成对及跨类交互,生成精确的体素级修正线索。第三,提出协作正向监督(CPS),优化不确定表示,使其与同类别非确定性表征对齐,提升模型对模糊区域的分类准确性。在三个公开3D医学分割数据集上的大量实验表明,该方法具有显著有效性与优越性。

原文摘要 · Abstract (English)

Semi-supervised 3D medical image segmentation aims to achieve accurate segmentation using few labelled data and numerous unlabelled data. The main challenge in the design of semi-supervised learning methods consists in the effective use of the unlabelled data for training. A promising solution consists of ensuring consistent predictions across different views of the data, where the efficacy of this strategy depends on the accuracy of the pseudo-labels generated by the model for this consistency learning strategy. In this paper, we introduce a new methodology to produce high-quality pseudo-labels for a consistency learning strategy to address semi-supervised 3D medical image segmentation. The methodology has three important contributions. The first contribution is the Cooperative Rectification Learning Network (CRLN) that learns multiple prototypes per class to be used as external knowledge priors to adaptively rectify pseudo-labels at the voxel level. The second contribution consists of the Dynamic Interaction Module (DIM) to facilitate pairwise and cross-class interactions between prototypes and multi-resolution image features, enabling the production of accurate voxel-level clues for pseudo-label rectification. The third contribution is the Cooperative Positive Supervision (CPS), which optimises uncertain representations to align with unassertive representations of their class distributions, improving the model's accuracy in classifying uncertain regions. Extensive experiments on three public 3D medical segmentation datasets demonstrate the effectiveness and superiority of our semi-supervised learning method.

3D分割半监督医学影像

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