无需人工标注,用支持集自动生成提示,提升医学影像分割精度
SAM2-SGP: Enhancing SAM2 for Medical Image Segmentation via Support-Set Guided Prompting
- 用支持集图像-掩码对自动生成伪掩码,替代人工提示
- 引入伪掩码注意力机制,显著提升定位与特征提取能力
- 适配多模态医学影像,对齐自然图像与医学图像的分布差异
尽管如Segment Anything Model 2(SAM2)等新视觉基础模型显著提升了零样本图像分割能力,但其依赖人工提示在医学图像分割任务中面临挑战。此外,由于原始训练数据为自然图像和视频,存在领域偏移问题,限制了SAM2在医学图像中的表现。为此,我们提出基于支持集引导提示的SAM2-SGP框架,消除对人工提示的依赖。该模型利用SAM2的记忆机制,通过支持集中的图像-掩码对,借助伪掩码生成(PMG)模块生成伪掩码;进一步设计伪掩码注意力(PMA)模块,以伪掩码自动生成边界框并引导注意力聚焦于相关区域,增强局部特征提取。同时采用低秩适应(LoRA)策略缓解领域偏移问题。该框架在包括眼底摄影、X射线、计算机断层扫描(CT)、磁共振成像(MRI)、正电子发射断层扫描(PET)及超声在内的多种2D与3D医学影像数据集上进行了评估,结果表明其性能显著优于当前先进模型(如nnUNet、SwinUNet)以及基础模型(如SAM2、MedSAM2),验证了方法的有效性。代码已开源:https://github.com/astlian9/SAM_Support。
原文摘要 · Abstract (English)
Although new vision foundation models such as Segment Anything Model 2 (SAM2) have significantly enhanced zero-shot image segmentation capabilities, reliance on human-provided prompts poses significant challenges in adapting SAM2 to medical image segmentation tasks. Moreover, SAM2's performance in medical image segmentation was limited by the domain shift issue, since it was originally trained on natural images and videos. To address these challenges, we proposed SAM2 with support-set guided prompting (SAM2-SGP), a framework that eliminated the need for manual prompts. The proposed model leveraged the memory mechanism of SAM2 to generate pseudo-masks using image-mask pairs from a support set via a Pseudo-mask Generation (PMG) module. We further introduced a novel Pseudo-mask Attention (PMA) module, which used these pseudo-masks to automatically generate bounding boxes and enhance localized feature extraction by guiding attention to relevant areas. Furthermore, a low-rank adaptation (LoRA) strategy was adopted to mitigate the domain shift issue. The proposed framework was evaluated on both 2D and 3D datasets across multiple medical imaging modalities, including fundus photography, X-ray, computed tomography (CT), magnetic resonance imaging (MRI), positron emission tomography (PET), and ultrasound. The results demonstrated a significant performance improvement over state-of-the-art models, such as nnUNet and SwinUNet, as well as foundation models, such as SAM2 and MedSAM2, underscoring the effectiveness of the proposed approach. Our code is publicly available at https://github.com/astlian9/SAM_Support.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。