Deep learning-based postoperative visual acuity prediction in idiopathic epiretinal membrane

BACKGROUND: To develop a deep learning (DL) model based on preoperative optical coherence tomography (OCT) training to automatically predict the 6-month postoperative visual outcomes in patients with idiopathic epiretinal membrane (iERM). METHODS: In this retrospective cohort study, a total of 442 eyes (5304 images in total) were enrolled for the development of the DL and multimodal deep fusion network (MDFN) models. All eyes were randomized into a training dataset with 265 eyes (60.0%), a validation dataset with 89 eyes (20.1%), and an internal testing dataset with the remaining 88 eyes (19.9%). The input variables for prediction consisted of macular OCT images and diverse clinical data. Inception-Resnet-v2 network was utilized to estimate the 6-month postoperative best-corrected visual acuity (BCVA). Concurrently, a regression model was developed using the clinical data and OCT parameters in the training data set for predicting postoperative BCVA. The reliability of the models was subsequently evaluated using the testing dataset. RESULTS: The prediction DL algorithm exhibited a mean absolute error (MAE) of 0.070 logMAR and root mean square error (RMSE) of 0.11 logMAR in the testing dataset. The DL model demonstrated a robust promising performance with R(2) = 0.80, notably superior to R(2) = 0.49 of the regression model. The percentages of BCVA prediction errors within ± 0.20 logMAR amounted to 94.32% in the testing dataset. CONCLUSIONS: The OCT-based DL model demonstrated sensitivity and accuracy in predicting postoperative BCVA in iERM patients. This innovative DL model exhibits substantial potential for integration into surgical planning protocols. SUPPLEMENTARY INFORMATION: The online version contains supplementary material available at 10.1186/s12886-023-03079-w.