DEAF-Net: Detail-Enhanced Attention Feature Fusion Network for Retinal Vessel Segmentation.

Journal: Journal of imaging informatics in medicine
Published Date:

Abstract

Retinal vessel segmentation is crucial for the diagnosis of ophthalmic and cardiovascular diseases. However, retinal vessels are densely and irregularly distributed, with many capillaries blending into the background, and exhibit low contrast. Moreover, the encoder-decoder-based network for retinal vessel segmentation suffers from irreversible loss of detailed features due to multiple encoding and decoding, leading to incorrect segmentation of the vessels. Meanwhile, the single-dimensional attention mechanisms possess limitations, neglecting the importance of multidimensional features. To solve these issues, in this paper, we propose a detail-enhanced attention feature fusion network (DEAF-Net) for retinal vessel segmentation. First, the detail-enhanced residual block (DERB) module is proposed to strengthen the capacity for detailed representation, ensuring that intricate features are efficiently maintained during the segmentation of delicate vessels. Second, the multidimensional collaborative attention encoder (MCAE) module is proposed to optimize the extraction of multidimensional information. Then, the dynamic decoder (DYD) module is introduced to preserve spatial information during the decoding process and reduce the information loss caused by upsampling operations. Finally, the proposed detail-enhanced feature fusion (DEFF) module composed of DERB, MCAE and DYD modules fuses feature maps from both encoding and decoding and achieves effective aggregation of multi-scale contextual information. The experiments conducted on the datasets of DRIVE, CHASEDB1, and STARE, achieving Sen of 0.8305, 0.8784, and 0.8654, and AUC of 0.9886, 0.9913, and 0.9911 on DRIVE, CHASEDB1, and STARE, respectively, demonstrate the performance of our proposed network, particularly in the segmentation of fine retinal vessels.

Authors

  • Pengfei Cai
    School of Electronic Engineering, Tianjin University of Technology and Education, Tianjin, 300222, China.
  • Biyuan Li
    School of Electronic Engineering, Tianjin University of Technology and Education, Tianjin, 300222, China. lby@tute.edu.cn.
  • Gaowei Sun
    School of Electronic Engineering, Tianjin University of Technology and Education, Tianjin, 300222, China.
  • Bo Yang
    Center for Cognition and Brain Disorders, Hangzhou Normal University, Hangzhou, Zhejiang Province 311121, China.
  • Xiuwei Wang
    School of Electronic Engineering, Tianjin University of Technology and Education, Tianjin, 300222, China.
  • Chunjie Lv
    School of Electronic Engineering, Tianjin University of Technology and Education, Tianjin, 300222, China.
  • Jun Yan
    Department of Statistics, University of Connecticut, Storrs, CT 06269, USA.