HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

ProMISe: Promptable Medical Image Segmentation using SAM

Jinfeng Wang Sifan Song Xinkun Wang Yiyi Wang Yiyi Miao Jionglong Su S. Kevin Zhou

ProMISe: Promptable Medical Image Segmentation using SAM

Abstract

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.

Code Repositories

xinkunwang111/promise
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
lesion-segmentation-on-isic-2018ProMISe
Mean IoU: 0.850
mean Dice: 0.921
medical-image-segmentation-on-cvc-colondbProMISe
mIoU: 0.789
mean Dice: 0.874
medical-image-segmentation-on-etisProMISe
mIoU: 0.750
mean Dice: 0.840
medical-image-segmentation-on-isic-2018-1ProMISe
DSC: 92.10
mIoU: 85.00
medical-image-segmentation-on-kvasir-segProMISe
mIoU: 0.851
mean Dice: 0.911

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp