Abstract:Precise segmentation of retinal vessels plays an important role in computer-assisted diagnosis. Deep learning models have been applied to retinal vessel segmentation, but the efficacy is limited by the significant scale variation of vascular structures and the intricate background of retinal images. This paper supposes a cross-channel spatial attention U-Net (CCS-UNet) for accurate retinal vessel segmentation. In comparison to other models based on U-Net, our model employes a ResNeSt block for the encoder-decoder architecture. The block has a multi-branch structure that enables the model to extract more diverse vascular features. It facilitates weight distribution across channels through the incorporation of soft attention, which effectively aggregates contextual information in vascular images. Furthermore, we suppose an attention mechanism within the skip connection. This mechanism serves to enhance feature integration across various layers, thereby mitigating the degradation of effective information. It helps acquire cross-channel information and enhance the localization of regions of interest, ultimately leading to improved recognition of vascular structures. In addition, the feature fusion module (FFM) module is used to provide semantic information for a more refined vascular segmentation map. We evaluated CCS-UNet based on five benchmark retinal image datasets, DRIVE, CHASEDB1, STARE, IOSTAR and HRF. Our proposed method exhibits superior segmentation efficacy compared to other state-of-the-art techniques with a global accuracy of 0.9617/0.9806/0.9766/0.9786/0.9834 and AUC of 0.9863/0.9894/0.9938/0.9902/0.9855 on DRIVE, CHASEDB1, STARE, IOSTAR and HRF respectively. Ablation studies are also performed to evaluate the the relative contributions of different architectural components. Our proposed model is potential for diagnostic aid of retinal diseases.

Accurate Retinal Vessel Segmentation via Octave Convolution Neural Network

A Multi-Scale Residual Attention Network for Retinal Vessel Segmentation

Multi-Scale and Multi-Branch Convolutional Neural Network for Retinal Image Segmentation

Accurate Retinal Vessel Segmentation in Color Fundus Images Via Fully Attention-Based Networks

Automatic Retinal Vessel Segmentation Based on an Improved U-Net Approach

Fundus Image Based Retinal Vessel Segmentation Utilizing a Fast and Accurate Fully Convolutional Network

Multi-Scale Retina Vessel Segmentation in OCTA with a Vascular Connectivity Module in the Convolutional Neural Network.

A Retinal Vessel Segmentation Algorithm with Convolutional Neural Network

Retinal Vessel Image Segmentation Based on Improved Convolutional Neural Network

Automatic Retinal Blood Vessel Segmentation Based on Fully Convolutional Neural Networks

CCS-UNet: a cross-channel spatial attention model for accurate retinal vessel segmentation

Improved Fully Convolutional Neuron Networks on Small Retinal Vessel Segmentation Using Local Phase As Attention.

Contextual Information Enhanced Convolutional Neural Networks for Retinal Vessel Segmentation in Color Fundus Images

Deep Dilated Convolutional Nets for the Automatic Segmentation of Retinal Vessels

Retinal Vessel Segmentation Algorithm Based on Residual Convolution Neural Network.

Multi-Scale Retinal Vessel Segmentation Based on Fully Convolutional Neural Network

Retinal vessel segmentation based on Fully Convolutional Neural Networks

FS-Net: Full Scale Network and Adaptive Threshold for Improving Extraction of Micro-Retinal Vessel Structures

Attention residual convolution neural network based on U-net (AttentionResU-Net) for retina vessel segmentation

Retinal Blood Vessels Segmentation With Improved SE‐UNet Model

ADD-Net: Attention U-Net with Dilated Skip Connection and Dense Connected Decoder for Retinal Vessel Segmentation