Predicting curve progression for adolescent idiopathic scoliosis using random forest model

Abstract

Background Adolescent Idiopathic Scoliosis (AIS) is a three-dimensional (3D) spinal deformity characterized by coronal curvature and rotational deformity. Predicting curve progression is important for the selection and timing of treatment. Although there is a consensus in the literature regarding prognostic factors associated with curve progression, the order of importance, as well as the combination of factors that are most predictive of curve progression is unknown. Objectives (1) create an ordered list of prognostic factors that most contribute to curve progression, and (2) develop and validate a Machine Learning (ML) model to predict the final major Cobb angle in AIS patients. Methods 193 AIS patients were selected for the current study. Preoperative PA, lateral and lateral bending radiographs were retrospectively obtained from the Shriners Hospitals for Children. Demographic and radiographic features, previously reported to be associated with curve progression, were collected. Sequential Backward Floating Selection (SBFS) was used to select a subset of the most predictive features. Based on the performance of several machine learning methods, a Random Forest (RF) regressor model was used to provide the importance rank of prognostic features and to predict the final major Cobb angle. Results The seven most predictive prognostic features in the order of importance were initial major Cobb angle, flexibility, initial lumbar lordosis angle, initial thoracic kyphosis angle, age at last visit, number of levels involved, and Risser "+" stage at the first visit. The RF model predicted the final major Cobb angle with a Mean Absolute Error (MAE) of 4.64 degrees. Conclusion A RF model was developed and validated to identify the most important prognostic features for curve progression and predict the final major Cobb angle. It is possible to predict the final major Cobb angle value within 5 degrees error from 2D radiographic features. Such methods could be directly applied to guide intervention timing and optimization for AIS treatment.