Modified Depthwise Parallel Attention UNet for Retinal Vessel Segmentation

Abstract

Retinal fundus images contain highly informative geometrical features for detecting diabetic retinopathy (DR), including vessels, especially thin and low-contrast vessels, which are predominant features for accurately diagnosing diabetic retinopathy. Automatic segmentation methods have been developed based on deep convolutional neural networks to replace manual labeling. These methods have shown acceptable performance in fundus vessel segmentation. The UNet model is a well-known architecture of deep neural networks often used for vessel segmentation tasks and has achieved significant performance. However, segmentation tasks remain challenging due to multiple convolutions, down-sampling operations, and inadequate feature fusion in the encoder-decoder architecture. Also, traditional convolution increases the number of multiplications while performing convolution operations. These challenges lead to the loss of information related to thin and low-contrast vessels, eventually affecting the segmentation performance. To tackle this issue, we propose incorporating depthwise parallel attention in the existing UNet framework (DPA-UNet) to achieve accurate vessel segmentation. This approach entails the integration of a depthwise convolution block in the downsampling path and a parallel attention mechanism in the upsampling path of UNet. The primary benefit of depthwise convolution and global information embedding (GIE) is the ability to capture intricate information characteristics across channels. This helps to minimize the information degradation caused by conventional convolution and downsampling techniques. A parallel attention network is proposed in the upsampling path of the existing UNet to optimize the channel and spatial information acquired from the encoder-decoder. Extensive experiments are conducted on three publicly available datasets, namely DRIVE, STARE, and CHASE_DB1, to validate the performance of the proposed model. The findings indicate that the UNET model with depthwise parallel attention achieved a competitive performance with fewer network parameters in segmenting retinal vessels.

Keywords

• Diabetic retinopathy
• vessel segmentation
• depth-wise separable convolution
• UNet
• attention mechanism
• deep learning