Real-time prediction of postoperative spinal shape with machine learning models trained on finite element biomechanical simulations

Purpose: Adolescent idiopathic scoliosis is a chronic disease that may require correction surgery. The finite element method (FEM) is a popular option to plan the outcome of surgery on a patient-based model. However, it requires considerable computing power and time, which may discourage its use. Machine learning (ML) models can be a helpful surrogate to the FEM, providing accurate real-time responses. This work implements ML algorithms to estimate post-operative spinal shapes. Methods: The algorithms are trained using features from 6400 simulations generated using the FEM from spine geometries of 64 patients. The features are selected using an autoencoder and principal component analysis. The accuracy of the results is evaluated by calculating the root-mean-squared error and the angle between the reference and predicted position of each vertebra. The processing times are also reported. Results: A combination of principal component analysis for dimensionality reduction, followed by the linear regression model, generated accurate results in real-time, with an average position error of 3.75 mm and orientation angle error below 2.74 degrees in all main 3D axes, within 3 ms. The prediction time is considerably faster than simulations based on the FEM alone, which require seconds to minutes. Conclusion: It is possible to predict post-operative spinal shapes of patients with AIS in real-time by using ML algorithms as a surrogate to the FEM. Clinicians can compare the response of the initial spine shape of a patient with AIS to various target shapes, which can be modified interactively. These benefits can encourage clinicians to use software tools for surgical planning of scoliosis.