Enhanced YOLOv8 with attention mechanisms for accurate detection of colorectal polyps

Shuangyuan Wang,Shengmao Lin,Fujia Sun,Xiaobo Li
DOI: https://doi.org/10.1016/j.bspc.2024.106942
IF: 5.1
2024-10-03
Biomedical Signal Processing and Control
Abstract:Colorectal cancer (CRC) is one of the most common and deadly cancers in the world, and most cases arise from polyps. Colonoscopy is a widely recognized and effective method for polyp diagnosis. However, clinical diagnosis has a high rate of missed polyps. Despite the capacity of deep learning methods to enhance the detection rate by extracting diverse features of polyps, the real-time performance, error rate, and misidentification ratio in actual clinical diagnosis have yet to meet the criteria for practical utilization. Here, we propose an improved structure for accurate polyp detection by enhancing the YOLOv8 algorithm to overcome these obstacles. Firstly, we introduce an enhanced Reverse Attention Mechanism Channel (RA-S) module to improve the algorithm's detection performance by fusing global feature information with local details of the image. Then, we integrate an attention mechanism into a Path Aggregation Network (PAnet) to improve the algorithm's ability to fuse multi-scale features to adapt to the variations in polyps. Finally, the proposed method was validated using the ETIS-LARIB dataset, which was not part of the training data. The proposed method achieved high precision (92.1 %), recall (84.5 %), and F1 (88.1 %) on the public ETIS-LARIB dataset, showcasing the robust detection performance and generalization ability of the proposed method.
engineering, biomedical
What problem does this paper attempt to address?