23 June 2020 Hybrid machine learning architecture for automated detection and grading of retinal images for diabetic retinopathy
Barath Narayanan Narayanan, Russell C. Hardie, Manawaduge Supun De Silva, Nathaniel K. Kueterman
Author Affiliations +
Abstract

Purpose: Diabetic retinopathy is the leading cause of blindness, affecting over 93 million people. An automated clinical retinal screening process would be highly beneficial and provide a valuable second opinion for doctors worldwide. A computer-aided system to detect and grade the retinal images would enhance the workflow of endocrinologists.

Approach: For this research, we make use of a publicly available dataset comprised of 3662 images. We present a hybrid machine learning architecture to detect and grade the level of diabetic retinopathy (DR) severity. We also present and compare simple transfer learning-based approaches using established networks such as AlexNet, VGG16, ResNet, Inception-v3, NASNet, DenseNet, and GoogLeNet for DR detection. For the grading stage (mild, moderate, proliferative, or severe), we present an approach of combining various convolutional neural networks with principal component analysis for dimensionality reduction and a support vector machine classifier. We study the performance of these networks under different preprocessing conditions.

Results: We compare these results with various existing state-of-the-art approaches, which include single-stage architectures. We demonstrate that this architecture is more robust to limited training data and class imbalance. We achieve an accuracy of 98.4% for DR detection and an accuracy of 96.3% for distinguishing severity of DR, thereby setting a benchmark for future research efforts using a limited set of training images.

Conclusions: Results obtained using the proposed approach serve as a benchmark for future research efforts. We demonstrate as a proof-of-concept that an automated detection and grading system could be developed with a limited set of images and labels. This type of independent architecture for detection and grading could be used in areas with a scarcity of trained clinicians based on the necessity.

© 2020 Society of Photo-Optical Instrumentation Engineers (SPIE) 2329-4302/2020/$28.00 © 2020 SPIE
Barath Narayanan Narayanan, Russell C. Hardie, Manawaduge Supun De Silva, and Nathaniel K. Kueterman "Hybrid machine learning architecture for automated detection and grading of retinal images for diabetic retinopathy," Journal of Medical Imaging 7(3), 034501 (23 June 2020). https://doi.org/10.1117/1.JMI.7.3.034501
Received: 16 December 2019; Accepted: 10 June 2020; Published: 23 June 2020
Lens.org Logo
CITATIONS
Cited by 17 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Machine learning

Principal component analysis

Detection and tracking algorithms

Feature extraction

Computing systems

Network architectures

Data modeling

Back to Top