

An Analytical Framework for Mental Health Prediction Using ML Models
Abstract
Mental health is a growing concern in modern society, especially in high-pressure work environments. With increasing awareness and the availability of large-scale mental health survey data, predictive modeling using machine learning (ML) techniques has emerged as a powerful tool for early diagnosis and support. This research presents a comprehensive pipeline that includes data preprocessing, feature engineering, model building, evaluation, and visualization to predict whether an individual will seek treatment for mental health issues. Multiple models including logistic regression, decision trees, random forests, k-nearest neighbors (KNN), bagging, boosting, stacking, and neural networks are implemented and evaluated. The stacking classifier showed the best performance with an accuracy of 81.7%. The results highlight the importance of factors like work interference, anonymity, and benefits in influencing mental health treatment-seeking behavior.
References
Chancellor, S., Lin, Z., Goodman, E. L., Zerwas, S., & De Choudhury, M. (2016). Quantifying and predicting mental illness severity in online pro-eating disorder communities. Proceedings of the 19th ACM Conference on Computer-Supported Cooperative Work & Social Computing, 1171–1184.
Coppersmith, G., Dredze, M., & Harman, C. (2014). Quantifying mental health signals in Twitter. Proceedings of the Workshop on Computational Linguistics and Clinical Psychology, 51–60.
Nguyen, T., Tran, T., Luo, W., & Venkatesh, S. (2014). Affective topic modeling for depression detection. Proceedings of the 5th ACM International Conference on Bioinformatics, Computational Biology, and Health Informatics, 593–600.
Shatte, A. B., Hutchinson, D. M., & Teague, S. J. (2019). Machine learning in mental health: A scoping review of methods and applications. Psychological Medicine, 49(9), 1426–1448.
Islam, M. R., Kabir, M. A., Ahmed, A., Kamal, A. R. M., Wang, H., & Ulhaq, A. (2018). Depression detection from social network data using machine learning techniques. Health Information Science and Systems, 6(1), 1–12.
Zhang, Y., Zhang, L., & Zhang, C. (2020). Comparative study of different machine learning techniques for mental health prediction. BMC Medical Informatics and Decision Making, 20(Suppl 10), 1–12.
Refbacks
- There are currently no refbacks.