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

PGE-SAM: Prompt-Guided Feature Enhancement for Interactive Segmentation under Degradation

About

Segment Anything Model (SAM) has revolutionized promptable image segmentation with strong zero-shot generalization. However, its performance degrades substantially under real-world imaging artifacts such as noise, blur, and compression. Existing methods restore features globally without focusing on segmentation-relevant regions and neglect SAM's iterative refinement mechanism, leading to suboptimal performance in interactive settings. We propose Prompt-Guided Feature Enhancement SAM (PGE-SAM), a framework that explicitly leverages user prompts and prior mask predictions to spatially guide the feature restoration process toward regions of interest through a Prompt Guidance Generator. To recover fine-grained details lost under degradation, we introduce Multi-Scale Features Interaction to incorporate low-level encoder features, along with a Foreground Reconstruction Loss that restricts feature-level supervision to the segmentation target. Furthermore, we present DM-Seg, a benchmark for interactive segmentation on degraded medical images, spanning multiple imaging modalities with both general and modality-specific degradations at varying severity levels. Extensive experiments demonstrate that PGE-SAM achieves SOTA robustness on both medical and natural image domains across multiple degradation levels, while maintaining generalization to clean images and adding less than one-fifth of the parameters of prior methods.

Tuan-Duc Nguyen, Anh-Tuan Mai, Duc-Trong Le• 2026

Related benchmarks

TaskDatasetResultRank
Interactive SegmentationDM-Seg LQ-3+ (seen)
IoU82.22
12
Interactive SegmentationDM-Seg LQ-3 (seen)
IoU82.85
12
Interactive SegmentationDM-Seg LQ-2 (seen)
IoU84.65
12
Interactive SegmentationDM-Seg LQ-1 (seen)
IoU85.31
12
Interactive SegmentationDM-Seg Clear (seen)
mIoU85.82
12
Interactive SegmentationDM-Seg Average (seen split)
IoU84.17
12
Medical Image SegmentationDM-Seg LQ-3+ (unseen)
IoU71.64
12
Medical Image SegmentationDM-Seg LQ-3 (unseen)
IoU72.09
12
Medical Image SegmentationDM-Seg LQ-2 (unseen)
mIoU73.77
12
Medical Image SegmentationDM-Seg LQ-1 (unseen)
IoU75.74
12
Showing 10 of 28 rows

Other info

Follow for update