Comparison of Different Machine Learning Algorithms in the Mental Health Assessment of College Students
DOI:
https://doi.org/10.13052/jicts2245-800X.1243Keywords:
Machine learning, college student, mental health assessment, LightGBMAbstract
This paper assesses college students’ mental health based on the symptom checklist 90 (SCL-90). In view of the assessment data processing and analysis, the performance of different machine learning algorithms, including random forest (RF), LightGBM3, extreme gradient boosting (XGBoost), in the classification of college students’ mental health samples was compared. Moreover, the effect of different hyperparameter optimization methods (grid search, Bayesian optimization, and particle swarm optimization) was compared. The experiment on the SCL-90 assessment dataset found that the optimization effect of grid search was poor, and the highest F1 value and area under the curve (AUC) of the RF algorithm were 0.8914 and 0.9384, respectively, the highest F1 and AUC values of the XGBoost algorithm were 0.9166 and 0.9551, respectively. The LightGBM algorithm optimized by particle swarm optimization showed the best performance in the classification of mental health samples, with an F1 value of 0.9790 and an AUC of 0.9945. It also achieved optimal results when compared to machine learning algorithms such as naive Bayes and the support vector machines. The results prove the reliability and accuracy of the particle swarm optimization-improved LightGBM algorithm in the analysis of college students’ mental health assessment data. The algorithm can be applied in practice to provide an effective tool for the analysis of the mental health assessment data of college students.
Downloads
References
R. Patt, D. Meva, ‘Advancements in Machine Learning-Based Mental Health Prediction: A Comprehensive Review’, International Conference on Cognitive Computing and Cyber Physical Systems, pp. 497–507, 2024.
H. Liu, J. Xu, ‘Research on the Use of Neural Network for the Prediction of College Students’ Mental Health’, Sci. Program., 2022, pp. 1-8, 2022.
X. Wang, ‘Research on Mental Health Education for College Students’, Int. J. Soc. Sci. Educ. Res., 3(3), pp. 153–157, 2020.
X. L. Mao, H. M. Chen, ‘Investigation of contemporary college students’mental health status and construction of a risk prediction model’, World J. Psychiatr., 13(8), pp. 573–582, 2023.
Y. Tang, J. Su, ‘Prediction and analysis of College Students’ mental health based on BP neural network’, J. Intell. Fuzzy Syst., pp. 1–7, 2020.
S. Liao, Q. Zhang, R. Gan, ‘Construction of real-time mental health early warning system based on machine learning’, J. Phys.: Conf. Ser., 1812(1), pp. 1–7, 2021.
B. Purwandari, W. S. Wibawa, N. Fitriah, M. Christia, D. R. Bintari, ‘Internet Addiction and Mental Health Prediction Using Ensemble Learning Based on Web Browsing History’, ICSIM ’20: The 3rd International Conference on Software Engineering and Information Management, pp. 155–159, 2020.
V. Laijawala, A. Aachaliya, H. Jatta, V. Pinjarkar, ‘Classification Algorithms based Mental Health Prediction using Data Mining’, 2020 5th International Conference on Communication and Electronics Systems (ICCES), Coimbatore, India, pp. 1174–1178, 2020.
C. Du, C. Liu, P. Balamurugan, P. Selvaraj, ‘Deep Learning-based Mental Health Monitoring Scheme for College Students Using Convolutional Neural Network’, Int. J. Artif. Intell. T., (6/8), pp. 30, 2021.
M. Ali, A. Baqir, H. H. R. Sherazi, A. Hussain, A. H. Alshehri, M. A. Imran, ‘Machine Learning Based Psychotic Behaviors Prediction from Facebook Status Updates’, Comput. Mater. Con., (008), pp. 2411–2427, 2022.
W. Chen, R. Bokoch, ‘The Impact of Parental Solid Self, Treatment Involvement, Stress, and Parenting Styles on Children’s Mental Health Symptom Severity’, J. Child Fam. Stud., 33(9), pp. 2919–2932, 2024.
A. Nanda, S. Tuteja, S. Gupta, ‘Machine learning based analysis and prediction of college students’ mental health during COVID-19 in India’, Artif. Intell. Mach. Learn. Mental Health Pandem., pp. 167–187, 2022.
Shobhika, P. Kumar, S. Chandra, ‘Prediction and comparison of psychological health during COVID-19 among Indian population and Rajyoga meditators using machine learning algorithms’, Proc. Comput. Sci., 218, pp. 697–705, 2023.
V. Azzi, K. Iskandar, S. Obeid, S. Hallit, ‘Parental divorce and smoking dependence in Lebanese adolescents: the mediating effect of mental health problems’, BMC Pediatr., 22(1), pp. 1–9, 2022.
Y.T. Zhang, T. Huang, F. Zhou, A. D. Huang, X. Q. Ji, L. He, Q. Geng, J. Wang, C, Mei, Y. J. Xu, Z. L. Yang, J. B. Zhan, J. Cheng, ‘Correlation between Anxiety, Depression,and Sleep Quality in College Students’, Biomed. Environ. Sci., 35(7), pp. 648–651, 2022.
X. Li, W. Song, J. Y. Zhang, C. Lu, Y. X. Wang, Y. X. Zheng, W. N. Hao, ‘Factors associated with mental health of graduate nursing students in China’, Medicine, 100(3), pp. 1–6, 2021.
Z. S. Musihb, H. S. A. Hussein, A. M. A. Ali, ‘Disruption of sleep patterns among secondary school adolescents’, J. Integr. Nurs., 6(3), pp. 145–149, 2024.
J. Liu, Q. Y. Xu, W. S. Chen, ‘Classification of Bird and Drone Targets Based on Motion Characteristics and Random Forest Model Using Surveillance Radar Data’, IEEE Access, 9, pp. 160135–160144, 2021.
S. Smadi, O. Almomani, A. Saaidah, A. H. Mohammad, M. Alauthman, ‘VPN Encrypted Traffic classification using XGBoost’, Int. J. Adv. Trends Comput. Sci. Eng., 9(7), pp. 960, 2021.
R. S. Mohril, B. S. Solanki, M. S. Kulkarni, B. K. Lad, ‘XGBoost based residual life prediction in the presence of human error in maintenance,’ Neural Comput. Appl., 35(4), pp. 3025–3039, 2023.
V. A. Dev, M. R. Eden, ‘Formation lithology classification using scalable gradient boosted decision trees’, Comput. Chem. Eng., 128, pp. 392–404, 2019.
M. H. Abdurrahman, B. Irawan, C. Setianingsih, ‘A Review of Light Gradient Boosting Machine Method for Hate Speech Classification on Twitter’, 2020 2nd International Conference on Electrical, Control and Instrumentation Engineering (ICECIE), Kuala Lumpur, Malaysia, pp. 1–6, 2020.
D. Wang, X. Ban, L. Ji, X. Guan, K. Liu, X. Qian, ‘An Adaptive Shrinking Grid Search Chaotic Wolf Optimization Algorithm Using Standard Deviation Updating Amount’, Comput. Intel. Neurosc., 2020(11), pp. 1–15, 2020.
Y. Yang, C. Ji, K. Deng, ‘Rapid design of metamaterials via multitarget bayesian optimization’, Ann. Appl. Stat., 15(2), pp. 768–796, 2021.
P. B. Kumar, D. R. Parhi, C. Sahu, ‘An approach to optimize the path of humanoid robots using a hybridized regression-adaptive particle swarm optimization-adaptive ant colony optimization method’, Ind. Robot, 46(1), pp. 104–117, 2019.
X. F. Song, Y. Zhang, Y. N. Guo, X. Y. Sun, Y. L. Wang, ‘Variable-size Cooperative Coevolutionary Particle Swarm Optimization for Feature Selection on High-dimensional Data’, IEEE T. Evolut. Comput., 24(5), pp. 882–895, 2020.
X. W. Zhang, H. Liu, T. Zhang, Q. Wang, Y. Wang, L. Tu, ‘Terminal crossover and steering-based particle swarm optimization algorithm with disturbance’, Appl. Soft Comput., 85, pp. 1–21, 2019.
A. Ishaq, S. Sadiq, M. Umer, S. Ullah, S. Mirjalili, V. Rupapara, M. Nappi, ‘Improving the Prediction of Heart Failure Patients’ Survival Using SMOTE and Effective Data Mining Techniques’, IEEE Access, 9, pp. 39707–39716, 2021.
J. Y. Verbakel, E. W. Steyerberg, H. Uno, B. D. Cock, L. Wynants, G. S. Collins, B. V. Calster, ‘ROC curves for clinical prediction models part 1. ROC plots showed no added value above the AUC when evaluating the performance of clinical prediction models (vol 126, pg 207, 2020)’, J. Clin. Epidemiol., 130, pp. 171–173, 2021.
A. T. Nahini, M. D. Baranon, C. Ayena, A. M. Soglo, E. N. Sambieni, M. A. Sanni, ‘Multidimensional and Logistic Regression Analysis of Unmet Family Planning Needs among Women of Childbearing Age in Donga Department, Benin’, Open Access Library J., 11(9), pp. 1–18, 2024.
S. S. Prasetiyowati, Y. Sibaroni, ‘Unlocking the potential of Naive Bayes for spatio temporal classification: a novel approach to feature expansion,’ J. Big Data, 11(1), pp. 1–27, 2024.
A. Z. Khameneh, M. Ghaznavi, A. Kilicman, Z. Mahad, A. Mardani, ‘A maximal-clique-based clustering approach for multi-observer multi-view data by using k-nearest neighbor with S-pseudo-ultrametric induced by a fuzzy similarity’, Neural Comput. Appl., 36(16), pp. 9525–9550, 2024.
J. S. Bobby, B. S. C. Kapali, U. S. Kumar, M. A. Femina, ‘QCBO-WSVM: Quantum chaos butterfly optimization-based weighted support vector machine for neuropathic pain detection from EEG signal’, Int. J. Imag. Syst. Tech., 33(5), pp. 1606–1620, 2023.
B. Sindhusaranya, M. R. Geetha, ‘Retinal blood vessel segmentation using root Guided decision tree assisted enhanced Fuzzy C-mean clustering for disease identification’, Biomedical Signal Processing and Control, 82, pp. 1–10, 2023.
B. Bailey, D. R. Strunk, ‘Predicting anxiety and depression over 12 months of the COVID-19 pandemic: A machine learning study’, J. Clin. Psychol., 79(10), pp. 2388–2403, 2023.
