Predicting early return to the operating room in early-onset scoliosis patients using machine learning techniques

Purpose: Surgical treatment of early-onset scoliosis (EOS) is associated with high rates of complications, often requiring unplanned return to the operating room (UPROR). The aim of this study was to create and validate a machine learning model to predict which EOS patients will go on to require an UPROR during their treatment course. Methods: A retrospective review was performed of all surgical EOS patients with at least 2 years follow-up. Patients were stratified based on whether they had experienced an UPROR. Ten machine learning algorithms were trained using tenfold cross-validation on an independent training set of patients. Model performance was evaluated on a separate testing set via their area under the receiver operating characteristic curve (AUC). Relative feature importance was calculated for the top-performing model. Results: 257 patients were included in the study. 146 patients experienced at least one UPROR (57%). Five factors were identified as significant and included in model training: age at initial surgery, EOS etiology, initial construct type, and weight and height at initial surgery. The Gaussian naïve Bayes model demonstrated the best performance on the testing set (AUC: 0.79). Significant protective factors against experiencing an UPROR were weight at initial surgery, idiopathic etiology, initial definitive fusion construct, and height at initial surgery. Conclusions: The Gaussian naïve Bayes machine learning algorithm demonstrated the best performance for predicting UPROR in EOS patients. Heavier, taller, idiopathic patients with initial definitive fusion constructs experienced UPROR less frequently. This model can be used to better quantify risk, optimize patient factors, and choose surgical constructs. Level of evidence: Prognostic: III.