Automation of Macular Degeneration Classification in the AREDS Dataset, Using a Novel Neural Network Design

PURPOSE: To create an ensemble of Convolutional Neural Networks (CNNs), capable of detecting and stratifying the risk of progressive age-related macular degeneration (AMD) from retinal photographs. DESIGN: Retrospective cohort study. METHODS: Three individual CNNs are trained to accurately detect 1) advanced AMD, 2) drusen size and 3) the presence or otherwise of pigmentary abnormalities, from macular centered retinal images were developed. The CNNs were then arranged in a “cascading” architecture to calculate the Age-related Eye Disease Study (AREDS) Simplified 5-level risk Severity score (Risk Score 0 – Risk Score 4), for test images. The process was repeated creating a simplified binary “low risk” (Scores 0–2) and “high risk” (Risk Score 3–4) classification. PARTICIPANTS: There were a total of 188,006 images, of which 118,254 images were deemed gradable, representing 4591 patients, from the AREDS1 dataset. The gradable images were split into 50%/25%/25% ratios for training, validation and test purposes. MAIN OUTCOME MEASURES: The ability of the ensemble of CNNs using retinal images to predict an individual’s risk of experiencing progression of their AMD based on the AREDS 5-step Simplified Severity Scale. RESULTS: When assessed against the 5-step Simplified Severity Scale, the results generated by the ensemble of CNN’s achieved an accuracy of 80.43% (quadratic kappa 0.870). When assessed against a simplified binary (Low Risk/High Risk) classification, an accuracy of 98.08%, sensitivity of ≥85% and specificity of ≥99% was achieved. CONCLUSION: We have created an ensemble of neural networks, trained on the AREDS 1 dataset, that is able to accurately calculate an individual’s score on the AREDS 5-step Simplified Severity Scale for AMD. If the results presented were replicated, then this ensemble of CNNs could be used as a screening tool that has the potential to significantly improve health outcomes by identifying asymptomatic individuals who would benefit from AREDS2 macular supplements.