Risk Prediction Model for Non-Suicidal Self-Injury in Chinese Adolescents with Major Depressive Disorder Based on Machine Learning.

BACKGROUND

Non-suicidal self-injury (NSSI) is a significant social issue, especially among adolescents with major depressive disorder (MDD). This study aimed to construct a risk prediction model using machine learning (ML) algorithms, such as XGBoost and random forest, to identify interventions for healthcare professionals working with adolescents with MDD.

METHODS

This study investigated 488 adolescents with MDD. Adolescents was randomly divided into 75% training set and 25% test set to testify the predictive value of risk prediction model. The prediction model was constructed using XGBoost and random forest algorithms. We evaluated the area under the receiver operating characteristic curve (AUC), sensitivity, specificity, accuracy, recall, F Score of the two models for comparing the performance of the two models.

RESULTS

There were 161 (33.00%) participants having NSSI. Compared without NSSI, there were statistically significant differences in gender (P=0.035), age (P=0.036), depressive symptoms (P=0.042), sleep quality (P=0.030), dysfunctional attitudes (P=0.048), childhood trauma (P=0.046), interpersonal problems (P=0.047), psychoticism (P) (P=0.049), neuroticism (N) (P=0.044), punishing and Severe (F2) (P=0.045) and Overly-intervening and Protecting (M2) (P=0.047) with NSSI. The AUC values for random forest and XGBoost were 0.780 and 0.807, respectively. The top five most important risk predictors identified by both machine learning methods were dysfunctional attitude, childhood trauma, depressive symptoms, F2 and M2.

CONCLUSION

The study demonstrates the suitability of prediction models for predicting NSSI behavior in Chinese adolescents with MDD based on ML. This model improves the assessment of NSSI in adolescents with MDD by health care professionals working. This provides a foundation for focused prevention and interventions by health care professionals working with these adolescents.