arXiv:2606.20765eess.IVcs.LG2026-06

针对医学影像标注难题,提出数据感知的冷启动主动学习方法

Dataset-Aware Cold-Start Active Learning for Annotation-Efficient 3D Medical Image Segmentation

论文配图:Dataset-Aware Cold-Start Active Learning for Annotation-Efficient 3D Medical Image Segmentation
图 1 · 摘自论文原文
  • 基于自监督信号融合典型性与重构不确定性,动态选择初始标注样本
  • 在4个3D医学图像数据集上,低中等标注预算下性能显著优于基线方法
  • 适合资源受限场景下医学图像分割模型的高效训练,尤其适合冷启动阶段

3D医学图像分割依赖大量人工标注,是体积医学成像的主要瓶颈。主动学习通过选择有信息量的样本减少标注负担,但多数方法假设已有初始标注集,未解决冷启动问题:如何从全未标注池中首次选择样本。本文提出CSCS(课程分层冷启动框架),根据未标注数据集结构自适应初始样本选择。CSCS结合两种无标签自监督信号:局部典型性(衡量嵌入空间代表性)和基于重构的不确定性(作为样本难度代理)。两者通过加权几何评分融合,权重由闭式步进规则确定,该规则基于有效标注预算与难度-覆盖比(池级统计量,衡量难度与代表性的一致性)。在四个3D医学图像分割基准(BraTS、FeTA、Spleen及内部胎儿MRI数据集)上评估,以nnU-Net为下游模型,CSCS在各数据集和标注预算下表现稳定且具有竞争力,尤其在低至中等标注预算下增益最明显。结果表明,数据感知的冷启动初始化可提升主动学习在3D医学图像分割中的鲁棒性,使样本选择更契合未标注池的几何特性。

原文摘要 · Abstract (English)

Deep learning for 3D medical image segmentation requires extensive manual annotations, a major bottleneck in volumetric medical imaging. Active learning aims to reduce this burden by selecting informative samples for annotation, but most methods assume that an initial labeled set is already available. This leaves the cold-start problem largely unresolved: how to select the first volumes from a fully unlabeled pool before any task-specific model is trained. We propose CSCS, a Curriculum-Stratified Cold-Start framework that adapts initial sample selection to the structure of the unlabeled dataset. CSCS combines two self-supervised, label-free signals: local typicality, measuring representativeness in the embedding space, and reconstruction-based uncertainty, used as a proxy for sample difficulty. These signals are combined through a weighted geometric score, where the weighting is determined by a closed-form pacing rule based on the effective annotation budget and the Difficulty-Coverage Ratio, a pool-level statistic measuring the alignment between difficulty and representativeness. We evaluate CSCS on four 3D medical image segmentation benchmarks: BraTS, FeTA, Spleen, and an in-house fetal MRI dataset. Using nnU-Net as downstream segmentation model, CSCS shows consistently competitive performance across datasets and annotation budgets, with the strongest gains in low-to-mid annotation regimes. These results suggest that dataset-aware cold-start initialization can improve the robustness of active learning for 3D medical image segmentation by adapting sample selection to the geometry of the unlabeled pool.

医学图像主动学习冷启动3D分割

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