用不确定性过滤伪标签,单次训练实现高效医学图像分割。
nnFilterMatch: A Unified Semi-Supervised Learning Framework with Uncertainty-Aware Pseudo-Label Filtering for Efficient Medical Segmentation
- 基于熵值筛选伪标签,动态排除高置信度预测
- 仅用5%~20%标注数据即达全监督模型性能
- 无需重训练循环,适合临床场景快速部署
半监督学习(SSL)在医学图像分割中展现出巨大潜力,可显著减少人工标注需求。结合主动学习(AL)后,通过选择最富信息量样本进一步降低标注负担。然而,传统SSL_AL混合方法依赖每轮标注后的迭代重训,计算开销大,限制了其在临床中的可扩展性。本文提出nnFilterMatch框架,将基于熵的伪标签过滤(FilterMatch)机制嵌入单次遍历的nnU-Net训练流程中。通过在训练中主动排除高置信度伪标签,避免重复重训,同时保留不确定性引导学习的优势。在多个临床分割基准上验证表明,该方法仅使用5%~20%标注数据即可达到或超越全监督模型性能。本工作提供了一种可扩展、端到端的低标注需求医学图像分割方案。代码已公开:https://github.com/Ordi117/nnFilterMatch.git。
原文摘要 · Abstract (English)
Semi-supervised learning (SSL) has emerged as a promising paradigm in medical image segmentation, offering competitive performance while substantially reducing the need for extensive manual annotation. When combined with active learning (AL), these strategies further minimize annotation burden by selectively incorporating the most informative samples. However, conventional SSL_AL hybrid approaches often rely on iterative and loop-based retraining cycles after each annotation round, incurring significant computational overhead and limiting scalability in clinical applications. In this study, we present a novel, annotation-efficient, and self-adaptive deep segmentation framework that integrates SSL with entropy-based pseudo-label filtering (FilterMatch), an AL-inspired mechanism, within the single-pass nnU-Net training segmentation framework (nnFilterMatch). By selectively excluding high-confidence pseudo-labels during training, our method circumvents the need for retraining loops while preserving the benefits of uncertainty-guided learning. We validate the proposed framework across multiple clinical segmentation benchmarks and demonstrate that it achieves performance comparable to or exceeding fully supervised models, even with only 5\%--20\% labeled data. This work introduces a scalable, end-to-end learning strategy for reducing annotation demands in medical image segmentation without compromising accuracy. Code is available here: https://github.com/Ordi117/nnFilterMatch.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。