English

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

Computer Vision and Pattern Recognition 2026-06-29 v1

Abstract

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.

Keywords

Cite

@article{arxiv.2606.30477,
  title  = {PGE-SAM: Prompt-Guided Feature Enhancement for Interactive Segmentation under Degradation},
  author = {Tuan-Duc Nguyen and Anh-Tuan Mai and Duc-Trong Le},
  journal= {arXiv preprint arXiv:2606.30477},
  year   = {2026}
}

Comments

54 pages