RPG-SAM: Reliability-Weighted Prototypes and Geometric Adaptive Threshold Selection for Training-Free One-Shot Polyp Segmentation
Organizations: Shanghai Key Laboratory of Multidimensional Information Processing, East China Normal University
Abstract
Training-free one-shot segmentation offers a scalable alternative to expert annotations where knowledge is often transferred from support images and foundation models. But existing methods often treat all pixels in support images and query response intensities models in a homogeneous way. They ignore the regional heterogeity in support images and response heterogeity in query.To resolve this, we propose RPG-SAM, a framework that systematically tackles these heterogeneity gaps. Specifically, to address regional heterogeneity, we introduce Reliability-Weighted Prototype Mining (RWPM) to prioritize high-fidelity support features while utilizing background anchors as contrastive references for noise suppression. To address response heterogeneity, we develop Geometric Adaptive Selection (GAS) to dynamically recalibrate binarization thresholds by evaluating the morphological consensus of candidates. Finally, an iterative refinement loop method is designed to polishes anatomical boundaries. By accounting for multi-layered information heterogeneity, RPG-SAM achieves a 5.56% mIoU improvement on the Kvasir dataset. Code will be released.
Figures & tables
| Method | Kvasir | PolypGen | CVC-ClinicDB | CVC-ColonDB | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Center 1 | Center 2 | Center 3 | ||||||||||
| IoU | Dice | IoU | Dice | IoU | Dice | IoU | Dice | IoU | Dice | IoU | Dice | |
| SEGIC [ 12 ] | 22.23 | 32.81 | 13.36 | 20.73 | 15.47 | 22.79 | 15.73 | 23.77 | 17.87 | 26.98 | 12.00 | 18.25 |
| PerSAM [ 21 ] | 26.01 | 35.38 | 26.15 | 31.47 | 12.97 | 18.96 | 10.87 | 17.74 | 13.29 | 20.10 | 10.31 | 16.18 |
| OPSAM [ 11 ] | 63.95 | 71.43 | 57.16 | 62.53 | 59.29 | 65.33 | 74.91 | 81.67 | 58.75 | 65.60 | 56.44 | 62.86 |
| Matcher [ 10 ] | 71.71 | 78.91 | 56.25 | 63.06 | 60.79 | 66.30 | 65.49 | 71.42 | 63.39 | 70.20 | 52.01 | 58.42 |
| BG Supp. | RWPM | GAS | PIR | mIoU (%) | mDice (%) | AUC-PR |
|---|---|---|---|---|---|---|
| - | - | - | - | 63.68 | 74.94 | 0.8743 |
| ✓ | - | - | - | 67.89 | 78.72 | 0.8921 |
| ✓ | ✓ | - | - | 71.27 | 81.05 | 0.9068 |
| ✓ | ✓ | ✓ | - | 75.25 | 83.64 | 0.9068 |
| ✓ | ✓ | ✓ | ✓ | 78.65 | 85.65 | 0.9068 |
| SLIC params | PIR params | GAS params | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| K | m | mIoU | mDice | mIoU | mDice | Range | mIoU | mDice | ||
| 5 | 5 | 77.34 | 84.52 | 0.8 | 0.7 | 77.98 | 85.20 | 0.3-0.6 | 75.05 | 82.57 |
| 10 | 5 | 78.28 | 85.27 | 0.8 | 0.8 | 78.11 | 85.32 | 0.4-0.6 | 77.15 | 84.48 |
| 10 | 20 | 78.65 | 85.65 | 0.8 | 0.9 | 77.79 | 85.12 | 0.4 - 0.7 | 78.65 | 85.65 |
| 10 | 50 | 78.34 | 85.41 | 0.9 | 0.8 | 78.65 | 85.65 | 0.5-0.7 | 77.26 | 84.57 |
| 20 | 5 | 76.08 | 83.21 | 0.95 | 0.8 | 78.66 | 85.60 | 0.5-0.8 | 77.12 | 84.41 |
Image Segmentation
Image segmentation assigns pixels or voxels to objects and regions, supporting measurement, scene understanding, and decisions across image domains.
See all 399 papers