Abstract
BACKGROUND: In this study, machine learning was used to assess the prediction of the magnitude of depression changes in college students based on various psychological variable information. METHODS: A group of college students from a certain school completed two assessments in October 2021 and March 2022, respectively. We collected baseline levels of depression, demographic variables, parenting styles, college students' mental health information, personality information, coping styles, SCL-90, and social support information. We applied logistic regression, random forest, support vector machine, and k-nearest neighbor machine learning methods to predict the magnitude of depression changes in college students. We selected the best-performing model and outputted the importance of features collected at different time points. RESULTS: Whether it is predicting the magnitude of positive changes or negative changes in depression, support vector machines (SVM) had the best prediction performance (with an accuracy of 89.4% for predicting negative changes in depression and an accuracy of 91.9% for predicting positive changes in depression). The baseline level of depression, father's emotional expression, and mother's emotional expression were all important predictors for predicting the negative and positive changes in depression among college students. CONCLUSION: Machine learning models can predict the extent of depression changes in college students. The baseline level of depression, as well as the emotional state of both fathers and mothers, play a significant role in predicting the negative and positive changes associated with depression in college students. This provides new insights and methods for future psychological health research and practice.