ECU-Net: an Efficient Cascade U-Net for Retinal Vessel Segmentation

Xin Yang,Zhiqiang Li,Yingqing Guo,Dake Zhou,Tao Li
DOI: https://doi.org/10.21203/rs.3.rs-327331/v1
2022-01-01
Abstract:Abstract Early detection of retinal vessel lesions by fundus image examination is essential to prevent and screen the related diseases. However, due to retinal vessels’ complex structure, it is time-consuming and laborious to extract retinal vessels manually by professionals. Therefore, retinal vessel automatic segmentation has great application value in modern medicine. In recent years, deep learning technology has achieved remarkable results in medical image segmentation, and the segmentation accuracy and speed are significantly improved compared with traditional methods. In this paper, we propose an efficient retinal vessel segmentation model ECU-Net. In this model, a feature extraction module combining edge attention module and efficient channel attention mechanism is designed to improve the vessel structure’s recognition ability. To improve the fusion efficiency of the texture information in the shallow layers and the semantic information in the deep layers, ECU-Net constructs an up-sampling fusion module. In addition, we design a multi-scale channel attention module to connect U-Net’s subnetwork to achieve efficient feature extraction. Compared with the state-of-the-art models, our ECU-Net has reached the highest level in some classical evaluation criteria and has a certain application value in the medical field.
What problem does this paper attempt to address?