Lightweight Deep Learning Model for Real-Time Colorectal Polyp Segmentation

• Main Authors: Jeong, S.-M (Author), Lee, E.-C (Author), Lee, J.-Y (Author), Lee, S.-G (Author), Seok, C.-L (Author)
• Format: Article
• Language: English
• Published: MDPI 2023
• Subjects: deep learning; DeepLabV3+; lightweight; medical image segmentation; MobileNetV3; polyp segmentation
• Online Access: View Fulltext in Publisher; View in Scopus
• ISBN: 20799292 (ISSN)
• DOI: 10.3390/electronics12091962

In colonoscopy, computer vision and artificial intelligence technology have enabled the automatic detection of the location of polyps and their visualization. These advancements have facilitated considerable research in the field. However, deep learning models used in the segmentation problem for capturing various patterns of polyps are becoming increasingly complex, which has rendered their operation in real time difficult. To identify and overcome this problem, a study was conducted on a model capable of precise polyp segmentation while increasing its processing speed. First, an efficient, high-performance, and lightweight model suitable for the segmentation of polyps was sought; the performance of existing segmentation models was compared and combined to obtain a learning model that exhibited good accuracy and speed. Next, hyperparameters were found for the MobileNetV3-encoder-based DeepLabV3+ model and, after tuning the hyperparameters, quantitative and qualitative results were compared, and the final model was selected. The experimental results showed that this model achieved high accuracy, with a Dice coefficient of 93.79%, while using a limited number of parameters and computational resources. Specifically, the model used 6.18 million parameters and 1.623 giga floating point operations for the CVC-ClinicDB dataset. This study revealed that increasing the amount of computation and parameters of the model did not guarantee unconditional performance. Furthermore, for the search and removal of polyps in cases in which morphological information is critical, an efficient model with low model complexity and high accuracy was proposed for real-time segmentation. © 2023 by the authors.