Machine Learning-Based Mental Health Classification System: Design, Implementation, and Evaluation

Mental disorders such as Bipolar Type-1, Bipolar Type-2, and Depression continue to affect millions of people worldwide, yet early and accurate diaganosis is challenging due to stigma, limited resources, and the subjectivity of self-reporting. Trying to bridge this gap, this project sought to develop a mental health diagnosis system that possesses the ability to classify individuals into Bipolar Type-1, Bipolar Type-2, Depression, or Normal states based on organized user input. Utilizing data acquired from an online repository, the system was designed with careful data cleaning, pre-processing, and class balancing using SMOTE for equal representation. Logistic Regression, Decision Tree, and Random Forest models were trained individually and then ensemble together using both hard and soft voting ensemble methods to obtain more stable predictions. The final ensemble model outperformed the individual models with accuracy up to 80%. This solution was deployed as a simple web app where users are able to answer a few guided questions and receive AI-generated feedback about their possible mental state instantly. The project demonstrates that the application of an ensemble of machine learning models will enhance early mental health screening and provide a supportive, accessible tool that will encourage individuals to seek professional help when needed.

Bipolar Disorder, Depression, Diagnosis System, Ensemble Learning

1. Alzheimer’s Association. (2023). 2023 Alzheimer’s disease facts and figures. Alzheimer’s & Dementia, 19(4), 1598-1695. https://doi.org/10.1002/alz.13016 [Google Scholar] [Crossref]

2. Arrieta, A. B., Díaz-Rodríguez, N., Del Ser, J., Bennetot, A., Tabik, S., Barbado, A., ... & Herrera, F. (2020). Explainable Artificial Intelligence (XAI): Concepts, taxonomies, opportunities and challenges toward responsible AI. Information Fusion, 58, 82-115. https://doi.org/10.1016/j.inffus.2019.12.012 [Google Scholar] [Crossref]

3. Bickman, L. (2020). Improving mental health services: A 50-year journey from randomized experiments to artificial intelligence and precision mental health. Administration and Policy in Mental Health and Mental Health Services Research, 47, 795-843. https://doi.org/10.1007/s10488-020-01078-5 [Google Scholar] [Crossref]

4. Chatterjee, M., Ghosh, S., & Chowdhury, S. R. (2021). Machine learning-based mental health prediction using survey data. Procedia Computer Science, 185, 293-302. https://doi.org/10.1016/j.procs.2021.05.031 [Google Scholar] [Crossref]

5. Farooq, F., Khalid, H., & Khan, M. S. (2022). AI-assisted early diagnosis of bipolar disorder. Journal of Affective Disorders, 307, 224-233. https://doi.org/10.1016/j.jad.2022.04.029 [Google Scholar] [Crossref]

6. Osborn, E., Wittkowski, A., Brooks, J., Briggs, P. E., & O’Brien, P. M. (2020). Women’s experiences of receiving a diagnosis of premenstrual dysphoric disorder: a qualitative investigation. BMC Women's Health, 20(1). doi:10.1186/s12905-020-01100-8 [Google Scholar] [Crossref]

7. Pim, C., Afzal, J., & Kamaldeep, B. (2023). The WHO World Mental Health Report: a call for action. The British Journal of Psychiatry, 222(6), 227-229. doi:10.1192/bjp.2023.9 [Google Scholar] [Crossref]

8. Sarker, I. H. (2021). Machine Learning: Algorithms, Real-World Applications and Research and Directions. Springer Link, 1-2. https://doi.org/10.1007/s42979-021-00592-x [Google Scholar] [Crossref]

9. World Health Organization. (2022). World mental health report: Transforming mental health for all. World Health Organization. Retrieved from https://www.who.int/publications/i/item/9789240049338 [Google Scholar] [Crossref]

10. Zaagsma, G., Stökl Ben Ezra, D., Rürup, M., Margolis, M., & Levi, A. S. (2022). Jewish Studies in the Digital Age. Berlin, Boston: De Gruyter Oldenbourg. doi:10.1515/9783110744828 [Google Scholar] [Crossref]

• What the Desert Fathers Teach Data Scientists: Ancient Ascetic Principles for Ethical Machine-Learning Practice
• Comparative Analysis of Some Machine Learning Algorithms for the Classification of Ransomware
• Comparative Performance Analysis of Some Priority Queue Variants in Dijkstra’s Algorithm
• Transfer Learning in Detecting E-Assessment Malpractice from a Proctored Video Recordings.
• Dual-Modal Detection of Parkinson’s Disease: A Clinical Framework and Deep Learning Approach Using NeuroParkNet