统一主动学习与半监督学习,提升稀缺标注下医学图像分割性能
Unifying Active Learning and Semi-Supervised Learning for Medical Image Segmentation

- 通过拓扑感知帕累托优化,同步选择待标注样本与利用无标签数据
- 在仅少量标注情况下,Dice等指标显著优于现有方法
- 适合标注资源极度有限的医疗影像研究者使用
实际应用中,医学图像分割常面临标注数据极少的情况,训练往往始于极低标注率阶段。此时需同时决定哪些病例应标注、如何利用剩余无标签数据。尽管主动学习(AL)与半监督学习(SSL)均应对标注稀缺问题,但二者通常独立设计,导致目标不一致且早期训练不稳定。本文提出RegAL,一种由共享拓扑感知帕累托优化驱动的统一主动半监督框架,兼顾样本选取与无标签数据利用。RegAL沿体素级不确定性、特征多样性及新型拓扑一致性三个互补维度筛选解剖学信息丰富的边缘案例用于标注;同一标准亦用于识别几何稳定图谱,支持微分同胚注册引导的数据增强,以训练自监督的Mean Teacher分割网络。在BraTS 2021、dHCP和ProstateX数据集上,RegAL在极低标注量下保持稳定,且在Dice、边界距离(ASD、HD95)指标上持续超越主流AL、SSL及主动半监督基线方法。
原文摘要 · Abstract (English)
In practical settings, medical image segmentation models are often developed with limited annotated data rather than fully labeled datasets. Training frequently begins in ultra-low labeled regimes where only a small number of volumes are annotated. In such scenarios, practitioners must simultaneously decide which cases to annotate and how to best use the remaining unlabeled data. Although active learning (AL) and semi-supervised learning (SSL) both target annotation scarcity, they are typically designed and optimized independently, resulting in objective mismatch and unstable training during early-stage "cold start" conditions. We propose RegAL, a unified active semi-supervised framework governed by a shared topology-aware Pareto optimization that couples sample acquisition with unlabeled data utilization. RegAL evaluates images along three complementary axes, voxel-wise uncertainty, feature diversity, and a novel topological consistency metric, to select anatomically informative edge cases for annotation. On the other hand, the same criteria are used to identify geometrically stable atlas candidates for diffeomorphic registration-guided augmentation to train a self-supervised Mean Teacher segmentation network. Across BraTS 2021, dHCP, and ProstateX, RegAL remains stable with few labeled volumes and consistently outperforms state-of-the-art AL, SSL, and active semi-supervised baselines across Dice and boundary-distance (ASD, HD95) metrics under extreme annotation scarcity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。