arXiv:2601.13677cs.CV2026-01中稿 · TMLR被引 1

提出简单有效方法,让3D生物医学图像分割主动学习超越随机采样基线。

Finally Outshining the Random Baseline: A Simple and Effective Solution for Active Learning in 3D Biomedical Imaging

  • 按类别分层+幂噪声衰减策略,解决类别不平衡和早期冗余问题。
  • 在24组实验中均显著优于改进的随机采样,提升分割精度。
  • 无需调参即可跨数据集泛化,适合真实场景部署使用。

主动学习(AL)有望大幅降低3D生物医学图像分割中的标注成本,但现有方法无法持续超越适配3D数据的改进随机采样基线。本文提出类分层调度幂预测熵(ClaSP PE),通过类分层查询确保稀有结构覆盖,并结合对数尺度幂噪声与衰减调度,在早期保证查询多样性、后期促进利用。在nnActive基准的四个3D生物医学数据集上,24组实验表明,ClaSP PE是唯一在分割质量上显著优于改进随机基线且保持高效标注的方法。进一步在四个未见过的数据集上模拟真实应用,所有参数按预设指南设置,结果证实其无需调参即可稳健泛化。本研究为3D分割主动学习提供了可落地的解决方案。代码开源:https://github.com/MIC-DKFZ/nnActive。

原文摘要 · Abstract (English)

Active learning (AL) has the potential to drastically reduce annotation costs in 3D biomedical image segmentation, where expert labeling of volumetric data is both time-consuming and expensive. Yet, existing AL methods are unable to consistently outperform improved random sampling baselines adapted to 3D data, leaving the field without a reliable solution. We introduce Class-stratified Scheduled Power Predictive Entropy (ClaSP PE), a simple and effective query strategy that addresses two key limitations of standard uncertainty-based AL methods: class imbalance and redundancy in early selections. ClaSP PE combines class-stratified querying to ensure coverage of underrepresented structures and log-scale power noising with a decaying schedule to enforce query diversity in early-stage AL and encourage exploitation later. In our evaluation on 24 experimental settings using four 3D biomedical datasets within the comprehensive nnActive benchmark, ClaSP PE is the only method that generally outperforms improved random baselines in terms of both segmentation quality with statistically significant gains, whilst remaining annotation efficient. Furthermore, we explicitly simulate the real-world application by testing our method on four previously unseen datasets without manual adaptation, where all experiment parameters are set according to predefined guidelines. The results confirm that ClaSP PE robustly generalizes to novel tasks without requiring dataset-specific tuning. Within the nnActive framework, we present compelling evidence that an AL method can consistently outperform random baselines adapted to 3D segmentation, in terms of both performance and annotation efficiency in a realistic, close-to-production scenario. Our open-source implementation and clear deployment guidelines make it readily applicable in practice. Code is at https://github.com/MIC-DKFZ/nnActive.

主动学习3D分割生物医学标注效率

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