arXiv:2411.11636cs.CVcs.AI2024-11

用超像素传播伪标签,仅需3%标注量实现高精度医学图像分割

SP${ }^3$ : Superpixel-propagated pseudo-label learning for weakly semi-supervised medical image segmentation

  • 利用超像素结构信息将稀疏涂鸦扩展为密集伪标签
  • 动态筛选优质超像素,使伪标签准确率提升至约80% Dice分数
  • 适合临床标注资源有限场景,显著降低医生标注负担

基于深度学习的医学图像分割可辅助诊断与加速治疗,但通常需大规模精细标注数据。弱监督半监督分割(WSSS)因其仅需少量涂鸦和大量无标签数据即可训练模型,极大减轻了临床标注压力。针对WSSS中监督信息不足的问题,本文提出超像素传播伪标签(SP³)方法,利用超像素中的结构信息补充监督信号。具体而言,将涂鸦标注传播至超像素,生成可用于监督训练的密集伪标签;由于伪标签质量受原始标注影响,通过动态阈值筛选高质量超像素以优化伪标签;同时引入超像素级别不确定性,引导伪标签监督,增强学习稳定性。在肿瘤与器官分割数据集上,本方法在仅使用3%标注量的情况下达到约80%的Dice分数,性能优于八种现有弱监督与半监督方法。大量实验验证了其有效性与标注效率,可帮助临床快速实现器官或肿瘤的自动化分割,最终惠及患者。

原文摘要 · Abstract (English)

Deep learning-based medical image segmentation helps assist diagnosis and accelerate the treatment process while the model training usually requires large-scale dense annotation datasets. Weakly semi-supervised medical image segmentation is an essential application because it only requires a small amount of scribbles and a large number of unlabeled data to train the model, which greatly reduces the clinician's effort to fully annotate images. To handle the inadequate supervisory information challenge in weakly semi-supervised segmentation (WSSS), a SuperPixel-Propagated Pseudo-label (SP${}^3$) learning method is proposed, using the structural information contained in superpixel for supplemental information. Specifically, the annotation of scribbles is propagated to superpixels and thus obtains a dense annotation for supervised training. Since the quality of pseudo-labels is limited by the low-quality annotation, the beneficial superpixels selected by dynamic thresholding are used to refine pseudo-labels. Furthermore, aiming to alleviate the negative impact of noise in pseudo-label, superpixel-level uncertainty is incorporated to guide the pseudo-label supervision for stable learning. Our method achieves state-of-the-art performance on both tumor and organ segmentation datasets under the WSSS setting, using only 3\% of the annotation workload compared to fully supervised methods and attaining approximately 80\% Dice score. Additionally, our method outperforms eight weakly and semi-supervised methods under both weakly supervised and semi-supervised settings. Results of extensive experiments validate the effectiveness and annotation efficiency of our weakly semi-supervised segmentation, which can assist clinicians in achieving automated segmentation for organs or tumors quickly and ultimately benefit patients.

医学图像分割弱监督学习伪标签超像素

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