轻量级GAN模型提升结肠镜图像息肉分割精度与速度
LGPS: A Lightweight GAN-Based Approach for Polyp Segmentation in Colonoscopy Images
- 用MobileNetV2+改进残差块和注意力模块,高效提取特征
- 在最大数据集上达到0.7299的Dice和0.7867的IoU,优于现有方法
- 仅107万参数,体积小17倍,适合临床实时应用
结直肠癌(CRC)是全球主要致死癌症之一,早期通过结肠镜发现并切除息肉可有效预防。尽管深度学习在息肉分割中展现潜力,仍面临计算成本高、小尺寸或低对比度息肉难分割、跨数据集泛化能力差等问题。为此,我们提出LGPS——一种轻量级基于GAN的息肉分割框架。其核心创新包括:(1) 采用改进的MobileNetV2骨干网络,结合修改的残差块与挤压-激励(ResE)模块以实现高效特征提取;(2) 引入卷积条件随机场(ConvCRF)进行边界精修;(3) 设计融合二值交叉熵、加权交并比损失与Dice损失的混合损失函数,缓解类别不平衡问题。LGPS在五个基准数据集上验证,与当前最优方法比较。在最大且最具挑战性的PolypGen测试集上,取得0.7299的Dice和0.7867的IoU,优于所有现有SOTA方法,并展现出强泛化能力。模型仅含107万参数,为最小现有模型的1/17,极适合实时临床部署。代码已开源。
原文摘要 · Abstract (English)
Colorectal cancer (CRC) is a major global cause of cancer-related deaths, with early polyp detection and removal during colonoscopy being crucial for prevention. While deep learning methods have shown promise in polyp segmentation, challenges such as high computational costs, difficulty in segmenting small or low-contrast polyps, and limited generalizability across datasets persist. To address these issues, we propose LGPS, a lightweight GAN-based framework for polyp segmentation. LGPS incorporates three key innovations: (1) a MobileNetV2 backbone enhanced with modified residual blocks and Squeeze-and-Excitation (ResE) modules for efficient feature extraction; (2) Convolutional Conditional Random Fields (ConvCRF) for precise boundary refinement; and (3) a hybrid loss function combining Binary Cross-Entropy, Weighted IoU Loss, and Dice Loss to address class imbalance and enhance segmentation accuracy. LGPS is validated on five benchmark datasets and compared with state-of-the-art(SOTA) methods. On the largest and challenging PolypGen test dataset, LGPS achieves a Dice of 0.7299 and an IoU of 0.7867, outperformed all SOTA works and demonstrating robust generalization. With only 1.07 million parameters, LGPS is 17 times smaller than the smallest existing model, making it highly suitable for real-time clinical applications. Its lightweight design and strong performance underscore its potential for improving early CRC diagnosis. Code is available at https://github.com/Falmi/LGPS/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。