arXiv:2503.01633cs.CV2025-03被引 2

用草图标注提升医学图像分割效率,边界更准

SparseMamba-PCL: Scribble-Supervised Medical Image Segmentation via SAM-Guided Progressive Collaborative Learning

  • 通过新算法将草图扩展为近似边界,增强标签质量
  • 融合Med-SAM与Sparse Mamba特征,提升表示能力
  • 适合医疗图像标注少、边界难分的场景

草图标注能显著降低复杂解剖结构医学数据集的密集标注成本。但现有方法在稀疏标注向稠密分割掩码传播及边界精确分割方面仍有不足。为此,我们提出一种渐进式协同学习框架,利用新型算法与Med-SAM基础模型提升训练中信息质量:(1) 提出新算法,将原始草图传播以估计物体边界,丰富真实标签;(2) 通过融合Med-SAM与所提Sparse Mamba网络的特征嵌入,优化特征表征,并支持用增强草图微调Med-SAM解码器;(3) 推理阶段引入Sparse Mamba网络,采用跳采样机制替代传统顺序块处理,高效捕捉局部与全局依赖。在ACDC、CHAOS和MSCMRSeg数据集上的实验验证了该框架的有效性,优于九种现有先进方法。代码已开源。

原文摘要 · Abstract (English)

Scribble annotations significantly reduce the cost and labor required for dense labeling in large medical datasets with complex anatomical structures. However, current scribble-supervised learning methods are limited in their ability to effectively propagate sparse annotation labels to dense segmentation masks and accurately segment object boundaries. To address these issues, we propose a Progressive Collaborative Learning framework that leverages novel algorithms and the Med-SAM foundation model to enhance information quality during training. (1) We enrich ground truth scribble segmentation labels through a new algorithm, propagating scribbles to estimate object boundaries. (2) We enhance feature representation by optimizing Med-SAM-guided training through the fusion of feature embeddings from Med-SAM and our proposed Sparse Mamba network. This enriched representation also facilitates the fine-tuning of the Med-SAM decoder with enriched scribbles. (3) For inference, we introduce a Sparse Mamba network, which is highly capable of capturing local and global dependencies by replacing the traditional sequential patch processing method with a skip-sampling procedure. Experiments on the ACDC, CHAOS, and MSCMRSeg datasets validate the effectiveness of our framework, outperforming nine state-of-the-art methods. Our code is available at \href{https://github.com/QLYCode/SparseMamba-PCL}{SparseMamba-PCL.git}.

医学图像草图标注分割Mamba

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