SetPlanner: A Lightweight Plug-in Point-Set Planner for Frozen SAM
Organizations: School of Computer Science and Engineering, Southeast Univ., China · School of Computer Science, Univ. of Sydney, Australia
Abstract
Segment Anything Models provide reusable priors, yet they require user prompts and cannot support fully automatic instrument segmentation. Automatic prompting is difficult for thin, articulated, reflective, and partly occluded tools, where several configurations can be valid. We formulate automatic prompting as lightweight point-set planning and isolate the point source under a frozen pathway. To this end, we present SetPlanner, a 1.52M-parameter plug-in point-set planner for frozen SAM. The plug-in preserves SAM's point-prompt interface and enables reuse across backbones. SetPlanner plans complete unordered K-point sets from geometry-aware targets with a permutation-aware conditional flow. SAM decodes eight candidates; their consensus readout yields a ground-truth-free prediction. Across three endoscopic datasets, SetPlanner wins all six transfer routes over a LoRA-adapted system. Under our frozen-pathway protocol, SetPlanner reaches 0.934 Dice on Kvasir-Instrument and recovers 96% of a 44.4-point localization gap, while candidate disagreement ranks low-Dice cases at AUROC 0.969.
Figures & tables
| Point source | Dice | Mask IoU | Hit |
|---|---|---|---|
| Zero-training rules | |||
| Uniform random | 0.091 | 0.060 | 0.114 |
| Center Gaussian | 0.121 | 0.091 | 0.131 |
| Color rule | 0.509 | 0.436 | 0.564 |
| Learned sources ( , control), frozen decoder | |||
| Logistic-1 | 0.922 | 0.873 | 0.970 |
| Kvasir | Endoscapes | EndoVis | Kvasir (SAM1) | |
|---|---|---|---|---|
| rule floor | 0.509 | 0.181 | 0.198 | 0.594 |
| GT-ref | 0.953 | 0.901 | 0.929 | 0.937 |
| 44.4 | 72.1 | 73.0 | 34.3 |
| Variant | Oracle | Pair IoU | ||
|---|---|---|---|---|
| Points per set | ||||
| 0.855 | 0.911 | 0.932 | 0.865 | |
| 0.912 | 0.928 | 0.941 | 0.936 | |
| 0.925 | 0.934 | 0.947 | 0.951 | |
| 0.909 | 0.918 | 0.932 | 0.949 | |
| Candidate source | ||||
Promptable Image Segmentation
Promptable image segmentation uses points, boxes, text, or masks to specify objects and generate pixel-level regions.
See all 152 papers