通过迭代伪标签与自适应复制粘贴,提升小肿瘤分割的半监督学习效果。
Iterative pseudo-labeling based adaptive copy-paste supervision for semi-supervised tumor segmentation
- 基于不确定性自适应增强,动态优化数据增广策略。
- 迭代伪标签转移生成更可靠的未标注样本标签。
- 在小体积肿瘤分割上优于现有方法,适合医学影像研究者。
半监督学习(SSL)在医学图像处理中受到广泛关注。最新方法结合一致性正则化与伪标签实现显著成效,但多数研究聚焦于大器官分割,忽视了肿瘤数量多或体积小等复杂场景。此外,对标注与未标注数据的增强策略尚未充分探索。为此,本文提出一种简单而高效的方法——基于迭代伪标签的自适应复制粘贴监督(IPA-CP),用于CT扫描中的肿瘤分割。IPA-CP引入双向不确定性自适应增强机制,将均值教师架构中的肿瘤不确定性融入增强过程;同时采用迭代伪标签转移策略,生成更具鲁棒性和信息量的未标注样本伪标签。在自建及公开数据集上的大量实验表明,该框架在医学图像分割任务中超越现有最优的半监督方法。消融实验证明了各项技术贡献的有效性。
原文摘要 · Abstract (English)
Semi-supervised learning (SSL) has attracted considerable attention in medical image processing. The latest SSL methods use a combination of consistency regularization and pseudo-labeling to achieve remarkable success. However, most existing SSL studies focus on segmenting large organs, neglecting the challenging scenarios where there are numerous tumors or tumors of small volume. Furthermore, the extensive capabilities of data augmentation strategies, particularly in the context of both labeled and unlabeled data, have yet to be thoroughly investigated. To tackle these challenges, we introduce a straightforward yet effective approach, termed iterative pseudo-labeling based adaptive copy-paste supervision (IPA-CP), for tumor segmentation in CT scans. IPA-CP incorporates a two-way uncertainty based adaptive augmentation mechanism, aiming to inject tumor uncertainties present in the mean teacher architecture into adaptive augmentation. Additionally, IPA-CP employs an iterative pseudo-label transition strategy to generate more robust and informative pseudo labels for the unlabeled samples. Extensive experiments on both in-house and public datasets show that our framework outperforms state-of-the-art SSL methods in medical image segmentation. Ablation study results demonstrate the effectiveness of our technical contributions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。