arXiv:2409.07172cs.CV2024-09CVPR被引 13

轻量级医学图像分割模型,兼顾速度与精度。

Swin-LiteMedSAM: A Lightweight Box-Based Segment Anything Model for Large-Scale Medical Image Datasets

  • 用微型Swin Transformer作图像编码器,降低计算开销。
  • 支持框提示和草图提示,提升多场景适应性。
  • 在医疗图像挑战赛中排名第四,性能速度快。

医学影像对疾病诊断与治疗至关重要,其中图像分割作为关键任务备受关注。然而,现有自动分割模型多为任务专用,难以应对不同成像模态和感兴趣区域。随着通用分割模型SAM的出现,构建适用于多种临床场景的统一模型成为可能。尽管已有若干医学SAM(MedSAM)方法提出,但多数依赖重型图像编码器,导致计算资源消耗大、推理速度慢,限制了实际应用。为此,我们提出轻量级版本LiteMedSAM——Swin-LiteMedSAM。该模型采用微型Swin Transformer作为图像编码器,融合框提示点与由边界框生成的草图提示,并在图像编码器与掩码解码器间建立跳跃连接。在CVPR 2024“笔记本端医学图像分割”挑战赛中,该方法在分割性能与速度间取得良好平衡,相比基准模型在多个模态上均有显著提升。验证集上达到DSC 0.8678、NSD 0.8844;测试集上DSC为0.8193,NSD为0.8461,位列第四。

原文摘要 · Abstract (English)

Medical imaging is essential for the diagnosis and treatment of diseases, with medical image segmentation as a subtask receiving high attention. However, automatic medical image segmentation models are typically task-specific and struggle to handle multiple scenarios, such as different imaging modalities and regions of interest. With the introduction of the Segment Anything Model (SAM), training a universal model for various clinical scenarios has become feasible. Recently, several Medical SAM (MedSAM) methods have been proposed, but these models often rely on heavy image encoders to achieve high performance, which may not be practical for real-world applications due to their high computational demands and slow inference speed. To address this issue, a lightweight version of the MedSAM (LiteMedSAM) can provide a viable solution, achieving high performance while requiring fewer resources and less time. In this work, we introduce Swin-LiteMedSAM, a new variant of LiteMedSAM. This model integrates the tiny Swin Transformer as the image encoder, incorporates multiple types of prompts, including box-based points and scribble generated from a given bounding box, and establishes skip connections between the image encoder and the mask decoder. In the \textit{Segment Anything in Medical Images on Laptop} challenge (CVPR 2024), our approach strikes a good balance between segmentation performance and speed, demonstrating significantly improved overall results across multiple modalities compared to the LiteMedSAM baseline provided by the challenge organizers. Our proposed model achieved a DSC score of \textbf{0.8678} and an NSD score of \textbf{0.8844} on the validation set. On the final test set, it attained a DSC score of \textbf{0.8193} and an NSD score of \textbf{0.8461}, securing fourth place in the challenge.

医学图像轻量化分割模型SwinTransformer

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