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.

Architecture And Factor Design Of Fully Convolutional Neural Networks For Retinal Vessel Segmentation

Accurate Retinal Vessel Segmentation via Octave Convolution Neural Network

Retinal Vessel Segmentation Via Cross-attention Feature Fusion

Retinal Vessel Segmentation based on Fully Convolutional Networks

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

A High-Resolution Network with Strip Attention for Retinal Vessel Segmentation

DCFU‐Net: Rethinking an Effective Attention and Convolutional Architecture for Retinal Vessel Segmentation

[Segmentation of retinal image vessels based on fully convolutional network with depthwise separable convolution and channel weighting].

Lightweight Attention Convolutional Neural Network for Retinal Vessel Image Segmentation

An Improved U-Net Architecture for Simultaneous Arteriole and Venule Segmentation in Fundus Image.

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

Construction of Retinal Vessel Segmentation Models Based on Convolutional Neural Network

Automated segmentaiton and classification of arterioles and venules using Cascading Dilated Convolutional Neural Networks

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

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

PCRTAM-Net: A Novel Pre-Activated Convolution Residual and Triple Attention Mechanism Network for Retinal Vessel Segmentation

Retinal Vessels Segmentation Based on Dilated Multi-Scale Convolutional Neural Network

Efficient CNN-CRF Network for Retinal Image Segmentation.

Deep Dilated Convolutional Nets for the Automatic Segmentation of Retinal Vessels

Size-Invariant Fully Convolutional Neural Network for Vessel Segmentation of Digital Retinal Images

A Supervised Method Using Convolutional Neural Networks for Retinal Vessel Delineation