ISSN: 1837-9664Journal of Cancer
Global reach, higher impact
J Cancer 2024; 15(1):41-53. doi:10.7150/jca.88684 This issue Cite
Research Paper
Lightweight colon polyp segmentation algorithm based on improved DeepLabV3+
1. School of Electrical EngineeringAnhui Polytechnic University, Wuhu 241000, China.
2. Anhui Key Laboratory of Electric Drive and Control, Wuhu 241000, China.
3. The First Affiliated Hospital of Wannan Medical College Wuhu, Wuhu 241001, China.
Abstract
To address the problems that the current polyp segmentation model is complicated and the segmentation accuracy needs to be further improved, a lightweight polyp segmentation network model Li-DeepLabV3+ is proposed. Firstly, the optimized MobileNetV2 network is used as the backbone network to reduce the model complexity. Secondly, an improved simple pyramid pooling module is used to replace the original Atrous Spatial Pyramid Pooling structure, which improves the model training efficiency of the model while reducing the model parameters. Finally, to enhance the feature representation, in the feature fusion module, the low-level feature and the high-level feature are fused using the improved Unified Attention Fusion Module, which applies both channel and spatial attention to enrich the fused features, thus obtaining more boundary information. The model was combined with transfer learning for training and validation on the CVC-ClinicDB and Kvasir SEG datasets, and the generalization of the model was verified across the datasets. The experiment results show that the Li-DeepLabV3+ model has superior advantages in segmentation accuracy and segmentation speed, and has certain generalization abilities.
Keywords: DeepLabV3+, Image segmentation, Feature fusion, Transfer learning
Citation styles
