CBAM Attention Gate-Based Lightweight Deep Neural Network Model for Improved Retinal Vessel Segmentation

Abstract

Over the years, researchers have been using deep learning in different fields of science including disease diagnosis. Retinal vessel segmentation has seen significant advancements through deep learning techniques, resulting in high accuracy. Despite this progress, challenges remain in automating the segmentation process. One of the most pressing and often overlooked issues is computational complexity, which is critical for developing portable diagnostic systems. To address this, this study introduces a CBAM-Attention Gate-based U-Netmodel aimed at reducing computational complexity without sacrificing performance on evaluation metrics. The performance of the model was analyzed using four publicly available fundus image datasets: CHASE_DB1, DRIVE, STARE, and HRF, and it achieved sensitivity, specificity, accuracy, AUC, and MCC performances (0.7909, 0.9975, 0.9723, 0.9867, and 0.8011), (0.8217, 0.9816, 0.9674, 0.9849, and 0.9778), (0.8346, 0.9790, 0.9680, 0.9855, and 0.7810), and (0.8082, 0.9769, 0.9638, 0.9723, and 0.7575), respectively. Moreover, this model comprises of only 0.8 million parameters, which makes it one of the lightest available models used for retinal vessel segmentation. This lightweight yet efficient model is most suitable for use in low-end hardware devices. The attributes of significantly lower computational complexity along with improved evaluation metrics advocates for its deployment in portable embedded devices to be used for population-level screening programs.