Abstract
SAM achieves strong segmentation but at high inference cost dominated by its ViT image encoder. Token merging accelerates ViTs without retraining, yet directly applying it to SAM is nontrivial: SAM mixes windowed and global attention and requires dense, prompt-conditioned features for precise boundary prediction. We systematically evaluate representative token merging methods on the SAM family in a strict off-the-shelf setting and find that existing destination-selection heuristics erode boundaries and leak prompt information as merge rates increase. We propose StructSAM, a resolution-preserving framework that computes lightweight token-energy scores from first-order feature gradients, protects boundary and prompt regions via grid-based flatness screening, and merges flat-region tokens toward low-energy targets with explicit recovery. We further present a spectral graph coarsening analysis showing that score-guided merging yields bounded Laplacian spectral distortion relative to random or window-restricted baselines. Across five natural and medical benchmarks, StructSAM reduces encoder FLOPs by 25–30% (up to 40%+ with prompt-aware merging) with minor drops in mIoU/Dice, consistently outperforming recent merging techniques at the same compute.
| Original language | English |
|---|---|
| Title of host publication | Proceedings of the 43rd International Conference on Machine Learning 2026 |
| Number of pages | 28 |
| Publication status | Accepted/In press - 2026 |
| Event | 43rd International Conference on Machine Learning - Seoul, Korea, Republic of Duration: 6 Jul 2026 → 11 Jul 2026 |
Conference
| Conference | 43rd International Conference on Machine Learning |
|---|---|
| Country/Territory | Korea, Republic of |
| City | Seoul |
| Period | 06/07/2026 → 11/07/2026 |
Fingerprint
Dive into the research topics of 'StructSAM: Structure- and Spectrum-Preserving Token Merging for Segment Anything Models'. Together they form a unique fingerprint.Cite this
- APA
- Author
- BIBTEX
- Harvard
- Standard
- RIS
- Vancouver