How to Design a Deep Neural Network for Retinal Vessel Segmentation: an Empirical Study

Yanzhou Su,Jian Cheng,Guiqun Cao,Haijun Liu
DOI: https://doi.org/10.1016/j.bspc.2022.103761
IF: 5.1
2022-01-01
Biomedical Signal Processing and Control
Abstract:Retinal vessel segmentation is a critical step towards the accurate visualization, diagnosis, early treatment and surgery planning of ocular diseases. However, due to limited densely annotated data, inhomogeneous lighting and poor target contrast, the accurate segmentation of the retinal vessels remains challenging. This paper presents an empirical study of retinal vessel segmentation in data processing, architecture design, attention mechanism and regularization strategy, which reveals several best practices for producing state-of-the-art retinal vessel segmentation performance. Some of them are first investigated in retinal vessels segmentation, and then thoroughly analyzed with various parameters. Furthermore, based on our empirical study, a well-designed deep neural network is proposed for retinal vessel segmentation on three digital retinal images benchmarks: DRIVE, STARE and CHASE_DB1. In addition, to verify the generalization capability of the proposed framework on different publicly available datasets, a cross-dataset evaluation between DRIVE and STARE is performed. Extensive experimental results suggest that our proposed framework not only achieves state-of-the-art performance among existing methods with AUCs of 98.279%, 98.862%, 98.724% on DRIVE, STARE, and CHASE_DB1, respectively, but also exhibits stronger generalization than previous models in retinal vessel segmentation.
What problem does this paper attempt to address?