DilateMobileU-Net: an efficient hybrid segmentation model for polyp diagnoses
摘要
Accurate polyp segmentation is crucial for early colon cancer diagnosis. While deep learning has significantly improved segmentation performance, many state-of-the-art models suffer from high computational costs and slow inference speeds, limiting their practicality in clinical endoscopic examinations. To address this issue, we propose DilateMobileU-Net, a lightweight yet effective hybrid model that balances segmentation accuracy with computational efficiency. Our approach integrates MobileNetv2 (MV2) and DilateMobileFormer (DMF) blocks in the encoder to enhance both local feature extraction and global contextual understanding. To further explore the trade-off between segmentation accuracy and model efficiency, we design three encoder variants with different parameter sizes, allowing flexible deployment in resource-constrained environments. The decoder utilizes residual convolutional layers to refine multi-scale feature representations for precise segmentation. Extensive experiments were conducted on two public datasets, i.e., Kvasir-SEG and CVC-ClinicDB datasets. Experimental results have proven that DilateMobileU-Net achieves the best results in terms of segmentation accuracy on the Kvasir-SEG dataset, with DSC and IoU of 92.56% and 88.57%, and reaches the competitive performance on the CVC-ClinicDB, with DSC and IoU of 92.45% and 87.58%, respectively. The model balances inference speed with accuracy, achieving precise polyp segmentation with minimal model parameters.