SERR-U-Net: Squeeze-and-Excitation Residual and Recurrent Block-Based U-Net for Automatic Vessel Segmentation in Retinal Image.

Journal: Computational and mathematical methods in medicine
Published Date:

Abstract

METHODS: A new SERR-U-Net framework for retinal vessel segmentation is proposed, which leverages technologies including Squeeze-and-Excitation (SE), residual module, and recurrent block. First, the convolution layers of encoder and decoder are modified on the basis of U-Net, and the recurrent block is used to increase the network depth. Second, the residual module is utilized to alleviate the vanishing gradient problem. Finally, to derive more specific vascular features, we employed the SE structure to introduce attention mechanism into the U-shaped network. In addition, enhanced super-resolution generative adversarial networks (ESRGANs) are also deployed to remove the noise of retinal image.

Authors

  • Jinke Wang
    Rongcheng College, Harbin University of Science and Technology, Rongcheng 264300, China.
  • Xiang Li
    Department of Radiology, Massachusetts General Hospital and Harvard Medical School, Boston, MA, United States.
  • Peiqing Lv
    School of Automation, Harbin University of Science and Technology, Harbin 150080, China.
  • Changfa Shi
    Mobile E-Business Collaborative Innovation Center of Hunan Province, Hunan University of Technology and Business, Changsha 410205, China.