Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SAIF: A Stability-Aware Inference Framework for Medical Image Segmentation with Segment Anything Model

About

Segment Anything Model (SAM) enable scalable medical image segmentation but suffer from inference-time instability when deployed as a frozen backbone. In practice, bounding-box prompts often contain localization errors, and fixed threshold binarization introduces additional decision uncertainty. These factors jointly cause high prediction variance, especially near object boundaries, degrading reliability. We propose the Stability-Aware Inference Framework (SAIF), a training-free and plug-and-play inference framework that improves robustness by explicitly modeling prompt and threshold uncertainty. SAIF constructs a joint uncertainty space via structured box perturbations and threshold variations, evaluates each hypothesis using decision stability and boundary consistency, and introduces a stability-consistency score to filter unstable candidates and perform stability-weighted fusion in probability space. Experiments on Synapse, CVC-ClinicDB, Kvasir-SEG, and CVC-300 demonstrate that SAIF consistently improves segmentation accuracy and robustness, achieving state-of-the-art performance without retraining or architectural modification. Our anonymous code is released at https://anonymous.4open.science/r/SAIF.

Ke Wu, Shiqi Chen, Yiheng Zhong, Hengxian Liu, Yingxue Su, Yifang Wang, Junhao Jin, Guangyu Ren• 2026

Related benchmarks

TaskDatasetResultRank
Polyp SegmentationCVC-ClinicDB (test)
DSC92.17
211
Polyp SegmentationKvasir-SEG (test)
mIoU0.8942
102
Polyp SegmentationCVC-300 (test)
mDice0.9475
38
Organ SegmentationSynapse multi-organ CT
DSC (Spleen)59.12
24
Showing 4 of 4 rows

Other info

Follow for update