Enabling Real-Time Colonoscopic Polyp Segmentation on Commodity CPUs via Ultra-Lightweight Architecture
About
Early detection of colorectal cancer hinges on real-time, accurate polyp identification and resection. Yet current high-precision segmentation models rely on GPUs, making them impractical to deploy in primary hospitals, mobile endoscopy units, or capsule robots. To bridge this gap, we present the UltraSeg family, operating in an extreme-compression regime (<0.3 M parameters). UltraSeg-108K (0.108 M parameters) is optimized for single-center data, while UltraSeg-130K (0.13 M parameters) generalizes to multi-center, multi-modal images. By jointly optimizing encoder-decoder widths, incorporating constrained dilated convolutions to enlarge receptive fields, and integrating a cross-layer lightweight fusion module, the models achieve 90 FPS on a single CPU core without sacrificing accuracy. Evaluated on seven public datasets, UltraSeg retains >94% of the Dice score of a 31 M-parameter U-Net while utilizing only 0.4% of its parameters, establishing a strong, clinically viable baseline for the extreme-compression domain and offering an immediately deployable solution for resource-constrained settings. This work provides not only a CPU-native solution for colonoscopy but also a reproducible blueprint for broader minimally invasive surgical vision applications. Source code is publicly available to ensure reproducibility and facilitate future benchmarking.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Polyp Segmentation | Kvasir | Dice Score79.26 | 128 | |
| Polyp Segmentation | Colonoscopy Datasets Average | Avg Dice79.26 | 22 | |
| Polyp Segmentation | CVC | Dice Score84.75 | 17 | |
| Instrument Segmentation | Kvasir-Instrument | Dice90.65 | 11 | |
| Polyp Segmentation | ETIS-Larib External | Dice68.21 | 11 | |
| Polyp Segmentation | BKAI-IGH External | Dice80.01 | 11 | |
| Polyp Segmentation | PolypDB | Dice0.782 | 11 | |
| Polyp Segmentation | PolypGen | Dice Coefficient68.26 | 11 | |
| Polyp Segmentation | Mixed Dataset | Dice80.38 | 11 |