A novel retinal vessel detection approach based on multiple deep convolution neural networks

Comput Methods Programs Biomed. 2018 Dec:167:43-48. doi: 10.1016/j.cmpb.2018.10.021. Epub 2018 Oct 30.

Abstract

Background and objective: Computer aided detection (CAD) offers an efficient way to assist doctors to interpret fundus images. In a CAD system, retinal vessel (RV) detection is a crucial step to identify the retinal disease regions. However, RV detection is still a challenging problem due to variations in morphology of the vessels on noisy and low contrast fundus images.

Methods: In this paper, we formulate the detection task as a classification problem and solve it using a multiple classifier framework based on deep convolutional neural networks. The multiple deep convolutional neural network (MDCNN) is constructed and trained on fundus images with limited image quantity. The MDCNN is trained using an incremental learning strategy to improve the networks' performance. The final classification results are obtained from the voting procedure on the results of MDCNN.

Results: The MDCNN achieves better performance and significantly outperforms the state-of-the-art for automatic retinal vessel segmentation on the DRIVE dataset with 95.97% and 96.13% accuracy and 0.9726 and 0.9737 AUC (area below the operator receiver character curve) score on training and testing sets, respectively. Another public dataset, STARE, is also used to evaluate the proposed network. The experimental results demonstrate that the proposed MDCNN network achieves 95.39% accuracy and 0.9539 AUC score in STARE dataset. We further compare our result with several state-of-the-art methods based on AUC values. The comparison is shown that our proposal yields the third best AUC value.

Conclusions: Our method yields the better performance in the compared the state of the art methods. In addition, our proposal has no preprocessing stage, and the input color fundus images are fed into the CNN directly.

Keywords: Image segmentation; Multiple deep convolution neural network; Retinal vessels segmentation.

MeSH terms

  • Algorithms
  • Computer Systems
  • Diagnosis, Computer-Assisted
  • Fundus Oculi
  • Humans
  • Machine Learning
  • Models, Statistical
  • Neural Networks, Computer*
  • Reproducibility of Results
  • Retinal Diseases / diagnostic imaging
  • Retinal Vessels / diagnostic imaging*
  • Sensitivity and Specificity