Machine Learning Model Trained with Finite Element Modeling Can Predict the Risk of Osteoarthritis: Data from the Osteoarthritis Initiative

Abstract

Objective: Despite long simulation times, recently developed finite element analysis (FEA) models of knee joints have demonstrated their suitability for predicting individual risk of onset and progression of knee osteoarthritis. Therefore, the objective of this study was to assess the feasibility of machine learning (ML) to replicate outcomes obtained from FEA when simulating mechanical responses and predicting cartilage degeneration within the knee joint. Design: Two ML models based on the Gaussian Process Regression (GPR) algorithms were developed. The first model (GPR1) utilized age, weight, and anatomical joint dimensions as predictor variables to predict tissue mechanical responses and cartilage degeneration based on FEA data. The second model (GPR2) utilized age, weight, height, and gender to predict anatomical joint dimensions, which were then used as inputs in the GPR1 model. Finally, the GPR1 and combined GPR1+GPR2 models were used to investigate the importance of clinical imaging when making personalized predictions for knees from healthy subjects with no history of knee injuries. Results: In the GPR1 model, R2 of 0.9 was exceeded for most of the predicted mechanical parameters. The GPR2 model was able to predict knee shape with R2 of 0.67–0.9. Both GPR1 and combined GPR1+GPR2 models offered equally good performances (AUC = 0.73–0.74) in classifying patients at high risk for the onset and development of knee osteoarthritis. Conclusions: In the future, real-time and easy-to-use GPR models may provide a rapid technology to evaluate mechanical responses within the knee for researchers or clinicians who have no former knowledge of FEA.

Keywords

• knee
• cartilage
• finite element analysis
• osteoarthritis
• machine learning