A retinal vessel detection approach using convolution neural network

Abstract

Computer-aided detection (CAD) provides an efficient way to assist doctors to interpret fundus images. In a CAD system, retinal vessel (RV) detection is an important step to identify the retinal disease regions automatically and accurately. However, RV detection is still a challenging problem due to variations in morphology of the vessels on a noisy background. In this paper, we formulate the detection task as a classification problem and solve it using a convolutional neural network (CNN) as a two-class classifier. The proposed model has 2 convolution layers, 2 pooling layers, 1 dropout layer and 1 loss layer. The proposed CNN achieves better performance and significantly outperforms the state-of-the-art for automatic retinal vessel segmentation on the DRIVE dataset with 91.78% accuracy and 0.96743 AUC score. We further compare our result with several state of the art methods based on AUC values. The comparison shows that our proposal yields the second best AUC value. This demonstrates the efficiency of the proposed method which has no pre-processing steps.