Establishment of a prediction tool for ocular trauma patients with machine learning algorithm

Abstract

AIM: To predict final visual acuity and analyze significant factors influencing open globe injury prognosis. METHODS: Prediction models were built using a supervised classification algorithm from Microsoft Azure Machine Learning Studio. The best algorithm was selected to analyze the predicted final visual acuity. We retrospectively reviewed the data of 171 patients with open globe injury who visited the Pusan National University Hospital between January 2010 and July 2020. We then applied cross-validation, the permutation feature importance method, and the synthetic minority over-sampling technique to enhance tool performance. RESULTS: The two-class boosted decision tree model showed the best predictive performance. The accuracy, precision, recall, F1 score, and area under the receiver operating characteristic curve were 0.925, 0.962, 0.833, 0.893, and 0.971, respectively. To increase the efficiency and efficacy of the prognostic tool, the top 14 features were finally selected using the permutation feature importance method: (listed in the order of importance) retinal detachment, location of laceration, initial visual acuity, iris damage, surgeon, past history, size of the scleral laceration, vitreous hemorrhage, trauma characteristics, age, corneal injury, primary diagnosis, wound location, and lid laceration. CONCLUSION: Here we devise a highly accurate model to predict the final visual acuity of patients with open globe injury. This tool is useful and easily accessible to doctors and patients, reducing the socioeconomic burden. With further multicenter verification using larger datasets and external validation, we expect this model to become useful worldwide.

Keywords

• machine learning
• ocular trauma
• open globe injury
• predictive model
• vision preservation