超轻量模型在普通CPU上实现实时肠镜息肉分割,精度媲美大型模型。
Enabling Real-Time Colonoscopic Polyp Segmentation on Commodity CPUs via Ultra-Lightweight Architecture
- 用分组多速率空洞卷积和注意力融合替代大参数模块,专为CPU优化。
- 0.13万参数模型在256×256下达50+帧率,Dice超0.8,超越所有同类小模型。
- 仅用1.7%参数逼近776万参数模型性能,适合资源有限的医院部署。
实时息肉分割对早期结直肠癌筛查至关重要,但临床应用仍受限于对GPU的依赖。本文提出UltraSeg系列,一系列参数低于0.3M的原生CPU分割模型。UltraSeg-108K(0.108M)达到极致压缩,UltraSeg-130K(0.130M)引入跨层轻量融合以提升多中心泛化能力。架构用分组多速率空洞卷积与注意力门控跨层融合替代高参数组件,在单核CPU上实现实时推理(256×256下超50 FPS,352×352下30 FPS),且保持临床级精度。在七个公开数据集上,UltraSeg-130K在两种分辨率下均取得超过0.8的Dice分数,显著优于所有现有<0.3M模型。尤其在零样本外部验证中,其表现接近或超过UNet-Medium(7.76M参数),仅使用其1.7%参数,建立首个可落地的原生CPU实时息肉分割基准。当扩展至4.38M参数时,模型精度媲美顶尖大模型,同时保持数量级参数优势,证明所提设计原则具备全效率范围内的表征增益。该工作首次提供可立即部署的原生CPU实时医疗AI方案,适用于资源匮乏场景,并为内镜以外的实时医学人工智能提供可复现蓝图。源代码已公开。
原文摘要 · Abstract (English)
Real-time polyp segmentation is essential for early colorectal cancer detection, yet clinical deployment remains blocked by GPU dependency. We introduce the UltraSeg family, a set of CPU-native segmentation models operating below 0.3M parameters. UltraSeg-108K (0.108M) establishes the extreme-compression frontier, while UltraSeg-130K (0.130M) integrates cross-layer lightweight fusion for enhanced multi-center generalization. The architecture replaces parameter-heavy components with grouped multi-rate dilated convolutions and attention-gated cross-layer fusion, achieving real-time throughput on a single CPU core (exceeding 50 FPS at 256*256 and 30 FPS at 352*352) without sacrificing clinical-grade accuracy. Evaluated on seven public datasets, UltraSeg-130K attains Dice scores exceeding 0.8 at both resolutions, substantially outperforming all existing sub-0.3M competitors. Notably, it approaches or exceeds UNet-Medium (7.76M parameters) on zero-shot external validations while using only 1.7% of its parameters, establishing the first strong baseline for CPU-native real-time polyp segmentation. When scaled to 4.38M parameters, UltraSeg achieves accuracy competitive with heavyweight state-of-the-art models while maintaining an order-of-magnitude parameter advantage, demonstrating that the proposed design principles yield intrinsic representational gains across the entire efficiency spectrum. By delivering the first clinically deployable, CPU-native real-time solution, this work provides an immediately usable tool for resource-limited settings and a reproducible blueprint for real-time medical AI beyond endoscopy. Source code is publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。