Sharpening Lightweight Models for Generalized Polyp Segmentation: A Boundary Guided Distillation from Foundation Models
Not on the map. Why?
Abstract (arXiv)
Automated polyp segmentation is critical for early colorectal cancer detection and its prevention, yet remains challenging due to weak boundaries, large appearance variations, and limited annotated data. Lightweight segmentation models such as U-Net, U-Net++, and PraNet offer practical efficiency for clinical deployment but struggle to capture the rich semantic and structural cues required for accurate delineation of complex polyp regions. In contrast, large Vision Foundation Models (VFMs), including SAM, OneFormer, Mask2Former, and DINOv2, exhibit strong generalization but transfer poorly to polyp segmentation due to domain mismatch, insufficient boundary sensitivity, and high computational cost. To bridge this gap, we propose textitLiteBounD, a underlineLighunderlinetwunderlineeight underlineBoundary-guided underlineDistillation framework that transfers complementary semantic and structural priors from multiple VFMs into compact segmentation backbones. LiteBounD introduces (i) a dual-path distillation mechanism that disentangles semantic and boundary-aware representations, (ii) a frequency-aware alignment strategy that supervises low-frequency global semantics and high-frequency boundary details separately, and (iii) a boundary-aware decoder that fuses multi-scale encoder features with distilled semantically rich boundary information for precise segmentation. Extensive experiments on both seen (Kvasir-SEG, CVC-ClinicDB) and unseen (ColonDB, CVC-300, ETIS) datasets demonstrate that LiteBounD consistently outperforms its lightweight baselines by a significant margin and achieves performance competitive with state-of-the-art methods, while maintaining the efficiency required for real-time clinical use. Our code is available at https://github.com/lostinrepo/LiteBounD.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 27 references that are in the map. First posted on arXiv 20 Apr 2026.
Builds on
18 of 27 references.
References · 18
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2023Segment Anythingcited by 8,158
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,338
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
- 2021Per-Pixel Classification is Not All You Need for Semantic Segmentationcited by 1,148
- 2019Kvasir-SEG: A Segmented Polyp Datasetcited by 588
- 2020PraNet: Parallel Reverse Attention Network for Polyp Segmentationcited by 460
- 2022OneFormer: One Transformer to Rule Universal Image Segmentationcited by 327
- 2016A Benchmark for Endoluminal Scene Segmentation of Colonoscopy Imagescited by 220
- 2019ResUNet++: An Advanced Architecture for Medical Image Segmentationcited by 211
6 more
- 2021Polyp-PVT: Polyp Segmentation with Pyramid Vision Transformerscited by 159
- 2021Shallow Attention Network for Polyp Segmentationcited by 107
- 2021Automatic Polyp Segmentation via Multi-scale Subtraction Networkcited by 77
- 2023MEGANet: Multi-Scale Edge-Guided Attention Network for Weak Boundary Polyp Segmentationcited by 28
- 2024SAM-Mamba: Mamba Guided SAM Architecture for Generalized Zero-Shot Polyp Segmentationcited by 7
- 2025From SAM to DINOv2: Towards Distilling Foundation Models to Lightweight Baselines for Generalized Polyp Segmentationcited by 2
Built on by
Not cited yet.