With the proposal of the Segment Anything Model (SAM), fine-tuning SAM for medical image segmentation (MIS) has become popular. However, due to the large size of the SAM model and the significant domain gap between natural and medical images, fine-tuning-based strategies are costly with potential risk of instability, feature damage and catastrophic forgetting. Furthermore, some methods of transferring SAM to a domain-specific MIS through fine-tuning strategies disable the model's prompting capability, severely limiting its utilization scenarios. In this paper, we propose an Auto-Prompting Module (APM), which provides SAM-based foundation model with Euclidean adaptive prompts in the target domain. Our experiments demonstrate that such adaptive prompts significantly improve SAM's non-fine-tuned performance in MIS. In addition, we propose a novel non-invasive method called Incremental Pattern Shifting (IPS) to adapt SAM to specific medical domains. Experimental results show that the IPS enables SAM to achieve state-of-the-art or competitive performance in MIS without the need for fine-tuning. By coupling these two methods, we propose ProMISe, an end-to-end non-fine-tuned framework for Promptable Medical Image Segmentation. Our experiments demonstrate that both using our methods individually or in combination achieves satisfactory performance in low-cost pattern shifting, with all of SAM's parameters frozen.
翻译:随着Segment Anything Model(SAM)的提出,针对医学图像分割(MIS)对SAM进行微调的方法日益流行。然而,由于SAM模型规模庞大且自然图像与医学图像之间存在显著领域差距,基于微调的策略成本高昂,并存在不稳定性、特征损伤及灾难性遗忘等潜在风险。此外,部分通过微调策略将SAM迁移至特定领域MIS的方法会破坏模型的提示能力,严重限制其应用场景。本文提出了一种自动提示模块(APM),该模块能够为基于SAM的基础模型提供目标域中的欧几里得自适应提示。实验表明,此类自适应提示显著提升了SAM在未经微调情况下对MIS的性能。此外,我们提出了一种名为增量模式移位(IPS)的新型非侵入式方法,用于将SAM适配至特定医学领域。实验结果显示,IPS无需微调即可使SAM在MIS任务中达到最优或具有竞争力的性能。通过结合这两种方法,我们提出了ProMISe——一个端到端且无需微调的可提示医学图像分割框架。实验证明,单独或组合使用我们的方法均能在低成本模式移位中取得满意效果,且SAM所有参数均保持冻结状态。