Machine learning models for orthokeratology lens fitting and axial length prediction

Abstract

PurposeIn order to improve the efficiency of orthokeratology (OK) lens fitting and predict the axial length after 1 year of OK lens wear, machine learning models were proposed.MethodsClinical data from 1302 myopic subjects were collected retrospectively, and two machine learning models were implemented. Demographic and corneal topographic data were collected as input variables. The output variables were the parameters of the OK lens and the axial length after 1 year. Eighty percent of input variables was used as the training set and the remaining 20% was used as the validation set. The first alignment curve (AC1) of the OK lenses, deduced using machine learning models and formula calculation, were compared. Multiple regression models (support vector machine, Gaussian process, decision tree and random forest) were used to predict the axial length after 1 year. In addition, we classified data based on lens brand, and carried out more detailed parameter fitting and analysis for spherical and toric OK lenses.ResultsThe OK lens fitting model showed higher (R2 = 0.93) and lower errors (mean absolute error [MAE] = 0.19, mean square error [MSE] = 0.09) when predicting AC1, compared with the formula calculation (R2 = 0.66, MAE = 0.44, MSE = 0.25). The machine learning model still had high R2 values ranging from 0.91 to 0.96 when considering the brand and design of the OK lenses. Further, the R2 value for the axial length prediction model was 0.94, which indicated that the machine learning model had high accuracy and good robustness.ConclusionThe OK lens fitting model and the axial length prediction model played an important role in guiding OK lens fitting, with high accuracy and robustness in prediction performance.