Predicting Visual Acuity by Using Machine Learning in Patients Treated for Neovascular Age-Related Macular Degeneration

Purpose To predict, by using machine learning, visual acuity (VA) at 3 and 12 months in patients with neovascular age-related macular degeneration (AMD) after initial upload of 3 anti–vascular endothelial growth factor (VEGF) injections. Design Database study. Participants For the 3-month VA forecast, 653 patients (379 female) with 738 eyes and an average age of 74.1 years were included. The baseline VA before the first injection was 0.54 logarithm of the minimum angle of resolution (logMAR) (±0.39). A total of 456 of these patients (270 female, 508 eyes, average age: 74.2 years) had sufficient follow-up data to be included for a 12-month VA prediction. The baseline VA before the first injection was 0.56 logMAR (±0.42). Methods Five different machine-learning algorithms (AdaBoost.R2, Gradient Boosting, Random Forests, Extremely Randomized Trees, and Lasso) were used to predict VA in patients with neovascular AMD after treatment with 3 anti-VEGF injections. Clinical data features came from a data warehouse (DW) containing electronic medical records (41 features, e.g., VA) and measurement features from OCT (124 features, e.g., central retinal thickness). The VA of patient eyes excluded from machine learning was predicted and compared with the ground truth, namely, the actual VA of these patients as recorded in the DW. Main Outcome Measures Difference in logMAR VA after 3 and 12 months upload phase between prediction and ground truth as defined. Results For the 3-month VA forecast, the difference between the prediction and ground truth was between 0.11 logMAR (5.5 letters) mean absolute error (MAE)/0.14 logMAR (7 letters) root mean square error (RMSE) and 0.18 logMAR (9 letters) MAE/0.2 logMAR (10 letters) RMSE. For the 12-month VA forecast, the difference between the prediction and ground truth was between 0.16 logMAR (8 letters) MAE/0.2 logMAR (10 letters) RMSE and 0.22 logMAR (11 letters) MAE/0.26 logMAR (13 letters) RMSE. The best performing algorithm was the Lasso protocol. Conclusions Machine learning allowed VA to be predicted for 3 months with a comparable result to VA measurement reliability. For a forecast after 12 months of therapy, VA prediction may help to encourage patients adhering to intravitreal therapy