Residual Spatial Attention Network for Retinal Vessel Segmentation

Reliable segmentation of retinal vessels can be employed as a way of monitoring and diagnosing certain diseases, such as diabetes and hypertension, as they affect the retinal vascular structure. In this work, we propose the Residual Spatial Attention Network (RSAN) for retinal vessel segmentation. RSAN employs a modified residual block structure that integrates DropBlock, which can not only be utilized to construct deep networks to extract more complex vascular features, but can also effectively alleviate the overfitting. Moreover, in order to further improve the representation capability of the network, based on this modified residual block, we introduce the spatial attention (SA) and propose the Residual Spatial Attention Block (RSAB) to build RSAN. We adopt the public DRIVE and CHASE DB1 color fundus image datasets to evaluate the proposed RSAN. Experiments show that the modified residual structure and the spatial attention are effective in this work, and our proposed RSAN achieves the state-of-the-art performance.

[1]  Juntang Zhuang,et al.  LadderNet: Multi-path networks based on U-Net for medical image segmentation , 2018, ArXiv.

[2]  Bo Wang,et al.  Dual Encoding U-Net for Retinal Vessel Segmentation , 2019, MICCAI.

[3]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[4]  Vijayan K. Asari,et al.  Nuclei Segmentation with Recurrent Residual Convolutional Neural Networks based U-Net (R2U-Net) , 2018, NAECON 2018 - IEEE National Aerospace and Electronics Conference.

[5]  In So Kweon,et al.  Convolutional Block Attention Module , 2018, ECCV 2018.

[6]  Quoc V. Le,et al.  DropBlock: A regularization method for convolutional networks , 2018, NeurIPS.

[7]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[8]  Vijayan K. Asari,et al.  Recurrent Residual Convolutional Neural Network based on U-Net (R2U-Net) for Medical Image Segmentation , 2018, ArXiv.

[9]  Dhimas Arief Dharmawan,et al.  Residual U-Net for Retinal Vessel Segmentation , 2019, 2019 IEEE International Conference on Image Processing (ICIP).

[10]  In-So Kweon,et al.  CBAM: Convolutional Block Attention Module , 2018, ECCV.

[11]  Yugen Yi,et al.  Dense Residual Network for Retinal Vessel Segmentation , 2020, ICASSP 2020 - 2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[12]  Chaoyi Zhang,et al.  Vessel-Net: Retinal Vessel Segmentation Under Multi-path Supervision , 2019, MICCAI.

[13]  Yugen Yi,et al.  SD-Unet: A Structured Dropout U-Net for Retinal Vessel Segmentation , 2019, 2019 IEEE 19th International Conference on Bioinformatics and Bioengineering (BIBE).

[14]  Yanning Zhang,et al.  Multiscale Network Followed Network Model for Retinal Vessel Segmentation , 2018, MICCAI.

[15]  Xin Yang,et al.  Joint Segment-Level and Pixel-Wise Losses for Deep Learning Based Retinal Vessel Segmentation , 2018, IEEE Transactions on Biomedical Engineering.

[16]  Matthew B. Blaschko,et al.  A Discriminatively Trained Fully Connected Conditional Random Field Model for Blood Vessel Segmentation in Fundus Images , 2017, IEEE Transactions on Biomedical Engineering.

[17]  Jian Sun,et al.  Identity Mappings in Deep Residual Networks , 2016, ECCV.