Abstract
BACKGROUND: There is substantial evidence from previous studies that abnormalities in sleep parameters associated with depression are demonstrated in almost all stages of sleep architecture. Patients with symptoms of sleep-wake disorders have a much higher risk of developing major depressive disorders (MDD) compared to those without. OBJECTIVE: The aim of the present study is to establish and compare the performance of different machine learning models based on sleep-wake disorder symptoms data and to select the optimal model to interpret the importance of sleep-wake disorder symptoms to predict MDD occurrence in adolescents. METHODS: We derived data for this work from 2020 to 2021 Assessing Nocturnal Sleep/Wake Effects on Risk of Suicide Phase I Study from National Sleep Research Resource. Using demographic and sleep-wake disorder symptoms data as predictors and the occurrence of MDD measured base on the center for epidemiologic studies depression scale as an outcome, the following six machine learning predictive models were developed: eXtreme Gradient Boosting model (XGBoost), Light Gradient Boosting mode, AdaBoost, Gaussian Naïve Bayes, Complement Naïve Bayes, and multilayer perceptron. The models' performance was assessed using the AUC and other metrics, and the final model's predictor importance ranking was explained. RESULTS: XGBoost is the optimal predictive model in comprehensive performance with the AUC of 0.804 in the test set. All sleep-wake disorder symptoms were significantly positively correlated with the occurrence of adolescent MDD. The insomnia severity was the most important predictor compared with the other predictors in this study. CONCLUSION: This machine learning predictive model based on sleep-wake disorder symptoms can help to raise the awareness of risk of symptoms between sleep-wake disorders and MDD in adolescents and improve primary care and prevention.