A deep learning model for the detection of both advanced and early glaucoma using fundus photography

PLoS One. 2018 Nov 27;13(11):e0207982. doi: 10.1371/journal.pone.0207982. eCollection 2018.

Abstract

Purpose: To build a deep learning model to diagnose glaucoma using fundus photography.

Design: Cross sectional case study Subjects, Participants and Controls: A total of 1,542 photos (786 normal controls, 467 advanced glaucoma and 289 early glaucoma patients) were obtained by fundus photography.

Method: The whole dataset of 1,542 images were split into 754 training, 324 validation and 464 test datasets. These datasets were used to construct simple logistic classification and convolutional neural network using Tensorflow. The same datasets were used to fine tune pre-trained GoogleNet Inception v3 model.

Results: The simple logistic classification model showed a training accuracy of 82.9%, validation accuracy of 79.9% and test accuracy of 77.2%. Convolutional neural network achieved accuracy and area under the receiver operating characteristic curve (AUROC) of 92.2% and 0.98 on the training data, 88.6% and 0.95 on the validation data, and 87.9% and 0.94 on the test data. Transfer-learned GoogleNet Inception v3 model achieved accuracy and AUROC of 99.7% and 0.99 on training data, 87.7% and 0.95 on validation data, and 84.5% and 0.93 on test data.

Conclusion: Both advanced and early glaucoma could be correctly detected via machine learning, using only fundus photographs. Our new model that is trained using convolutional neural network is more efficient for the diagnosis of early glaucoma than previously published models.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Cross-Sectional Studies
  • Deep Learning*
  • Diagnostic Techniques, Ophthalmological*
  • Disease Progression
  • Fundus Oculi
  • Glaucoma / diagnostic imaging*
  • Humans
  • Image Interpretation, Computer-Assisted / methods*
  • Logistic Models
  • ROC Curve

Grants and funding

PDXen partially funded and sponsored the study. There was no additional external funding received for this study. SHC and KSA: Employees – PDXen. PDXen provided support in the form of salary for authors SHC and KSA, and participated in data analysis, and interpretation of the results, and in the preparation of the manuscript. The specific roles of these authors are articulated in the ‘author contributions’ section.