Applying Machine Learning to Predict Complex Clinical Course in Youth With Eating Disorders.

Abstract

Objective: To compare the predictive performance of supervised machine learning models to logistic regression in identifying youth with eating disorders at risk of a complex clinical course based on clinical characteristics from the first treatment episode.

Methods: Clinical data from 327 youth treated at any level of care at the Children's Hospital of Eastern Ontario Eating Disorders Program (2018-2024) were extracted. Complex clinical course outcome was defined as either readmission after discharge or a treatment trajectory deviating from the expected step-down in intensity, including return to the same or escalation to a higher level of care. Thirty-four intake and discharge variables from the first treatment episode were used to train seven machine learning models and logistic regression using repeated nested cross-validation. Performance was assessed by AUC and brier scores. Models using intake-only versus intake plus discharge data were compared. A parsimonious model using the top 10 predictors was also evaluated.

Results: Random forest model with intake and discharge data achieved the best performance (AUC = 0.723; Brier = 0.176) that was significantly superior to logistic regression. Models trained on intake-only data showed poor discrimination (AUCs < 0.6). Including discharge data improved model performance across all algorithms. The most important predictor was weight change throughout treatment. Random forest performance declined when restricted to the top 10 predictors.

Discussion: Supervised machine learning demonstrates improved predictive performance for eating disorder disease course outcomes compared to traditional statistical methods, especially in higher-dimensionality settings. These findings support future application of machine learning to complex biopsychosocial datasets to advance precision medicine initiatives in the eating disorder field and better understand the etiology of disease trajectory.