Machine Learning Approaches to Workplace Mental Health: Predicting Treatment-Seeking Behavior Using the OSMI Dataset
Main Article Content
Abstract
Employee mental health is increasingly recognized as essential for sustainable organizational performance, particularly in technology sectors where work intensity and psychological strain are prevalent. This study leverages machine learning to identify predictors of treatment-seeking behavior using the Open Sourcing Mental Illness (OSMI) dataset, which includes 1,387 anonymized responses from the 2014 OSMI survey. The survey examines employees’ experiences and perceptions of mental health in the global tech industry. Through data cleaning and encoding, key factors influencing help-seeking behavior were identified, including family history of mental illness and work interference due to psychological distress. Two machine learning models, Decision Tree and K-Nearest Neighbour (KNN), were employed for prediction. The Decision Tree model achieved an accuracy of 73%, while KNN attained 100%, suggesting high predictive power, albeit with potential overfitting risks. These findings align with recent studies promoting the integration of AI-driven analytics in workplace wellness programs to detect hidden behavioral trends and enable early interventions. The results demonstrate that machine learning models can offer valuable insights into employee well-being and preventative strategies. Future research should focus on incorporating larger, more diverse datasets and adopting explainable AI (XAI) techniques to enhance interpretability, fairness, and trust in predictive systems for mental health in the workplace.
Downloads
Article Details

This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
The Authors submitting a manuscript do so on the understanding that if accepted for publication, copyright of the article shall be assigned to journal Tech-E, Universitas Buddhi Dharma as publisher of the journal.
Copyright encompasses exclusive rights to reproduce and deliver the article in all form and media, including reprints, photographs, microfilms and any other similar reproductions, as well as translations. The reproduction of any part of this journal, its storage in databases and its transmission by any form or media, such as electronic, electrostatic and mechanical copies, photocopies, recordings, magnetic media, etc. , will be allowed only with a written permission from journal Tech-E.
journal Tech-E, the Editors and the Advisory Editorial Board make every effort to ensure that no wrong or misleading data, opinions or statements be published in the journal. In any way, the contents of the articles and advertisements published in the journal Tech-E, Universitas Buddhi Dharma are sole and exclusive responsibility of their respective authors and advertisers.
References
E. Monteiro and J. Joseph, “A Review on the Impact of Workplace Culture on Employee Mental Health and Well-Being,” Int. J. Case Stud. Business, IT, Educ., vol. 7, no. 2, pp. 291–317, 2023, doi: 10.47992/ijcsbe.2581.6942.0274.
A. Wu, E. C. Roemer, K. B. Kent, D. W. Ballard, and R. Z. Goetzel, “Organizational Best Practices Supporting Mental Health in the Workplace,” J. Occup. Environ. Med., vol. 63, no. 12, pp. e925–e931, Dec. 2021, doi: 10.1097/JOM.0000000000002407.
L. E. Søvold et al., “Prioritizing the Mental Health and Well-Being of Healthcare Workers: An Urgent Global Public Health Priority,” Front. Public Heal., vol. 9, May 2021, doi: 10.3389/fpubh.2021.679397.
WHO, “Mental health at work,” World Health Organization. [Online]. Available: https://www.who.int/news-room/fact-sheets/detail/mental-health-at-work
W. N. Burton, C.-Y. Chen, D. J. Conti, A. B. Schultz, G. Pransky, and D. W. Edington, “The Association of Health Risks With On-the-Job Productivity,” J. Occup. Environ. Med., vol. 47, no. 8, pp. 769–777, Aug. 2005, doi: 10.1097/01.jom.0000169088.03301.e4.
D. E. Bloom et al., “The global economic burden of Non-communicable Diseases,” World Cancer Rep. 2014, no. 87, pp. 1–8, 2014.
G. Giorgi et al., “COVID-19-Related Mental Health Effects in the Workplace: A Narrative Review,” Int. J. Environ. Res. Public Health, vol. 17, no. 21, p. 7857, Oct. 2020, doi: 10.3390/ijerph17217857.
J. M. Baker et al., “Quantification of Occupational and Community Risk Factors for SARS-CoV-2 Seropositivity Among Health Care Workers in a Large U.S. Health Care System,” Ann. Intern. Med., vol. 174, no. 5, pp. 649–654, May 2021, doi: 10.7326/M20-7145.
V. Juergensen et al., “The help-seeking process and predictors of mental health care use among individuals with depressive symptoms: a machine learning approach,” Front. Public Heal., vol. 12, Nov. 2024, doi: 10.3389/fpubh.2024.1504720.
M. Rashid and C. Wang, “Machine learning for mental health prediction: A comparative analysis,” BMC Med. Inform. Decis. Mak., vol. 22, p. 215, 2022.
B. W. Nelson, C. A. Low, N. C. Jacobson, P. A. Areán, and J. Torous, “Machine learning and digital mental health: Opportunities, challenges, and future directions,” Curr. Psychiatry Rep., vol. 24, no. 2, pp. 71–83, 2023.
K. S. Hammerschmidt, S. K. Lageman, and D. A. Colby, “Digital mental health interventions in the workplace: A systematic review,” J. Occup. Health Psychol., vol. 28, no. 1, pp. 1–15, 2023.
E. Carstoiu, Mental health definitions and impact. United Nations, 2020.
S. Uddin, A. Khan, M. E. Hossain, and M. A. Moni, “Comparing different supervised machine learning algorithms for disease prediction,” BMC Med. Inform. Decis. Mak., vol. 19, no. 1, p. 281, Dec. 2019, doi: 10.1186/s12911-019-1004-8.
J. R. Katzenbach and D. K. Smith, “The discipline of teams.,” Harv. Bus. Rev., vol. 71, no. 2, pp. 111–120, 1993.
D. Bzdok, N. Altman, and M. Krzywinski, “Statistics versus machine learning,” Nat. Methods, vol. 15, no. 4, pp. 233–234, Apr. 2018, doi: 10.1038/nmeth.4642.
S. R. Safavian and D. Landgrebe, “A survey of decision tree classifier methodology,” IEEE Trans. Syst. Man. Cybern., vol. 21, no. 3, pp. 660–674, 1991, doi: 10.1109/21.97458.
T. Cover and P. Hart, “Nearest neighbor pattern classification,” IEEE Trans. Inf. Theory, vol. 13, no. 1, pp. 21–27, Jan. 1967, doi: 10.1109/TIT.1967.1053964.
V. Podgorelec, P. Kokol, B. Stiglic, and I. Rozman, “Decision Trees: An Overview and Their Use in Medicine,” J. Med. Syst., vol. 26, no. 5, pp. 445–463, Oct. 2002, doi: 10.1023/A:1016409317640.
WHO, “Mental health in the workplace,” World Health Organization. [Online]. Available: ttps://www.who.int/news-room/fact-sheets/detail/mental-health-at-work
A. Thakkar, A. Gupta, and A. De Sousa, “Artificial intelligence in positive mental health: a narrative review,” Front. Digit. Heal., vol. 6, Mar. 2024, doi: 10.3389/fdgth.2024.1280235.
M. Bibi, T. G. Tan, and H. Yao, “Exploring the Impact of AI Capabilities on Employee Well-Being: A Mediated Moderation Analysis,” SAGE Open, vol. 15, no. 3, Jan. 2025, doi: 10.1177/21582440251361981.
M. Saarela and V. Podgorelec, “Recent Applications of Explainable AI (XAI): A Systematic Literature Review,” Appl. Sci., vol. 14, no. 19, p. 8884, Oct. 2024, doi: 10.3390/app14198884.
P. Biecek and T. Burzykowski, Explanatory Model Analysis. Chapman and Hall/CRC, 2021. doi: 10.1201/9780429027192.
H. Auf, P. Svedberg, J. Nygren, M. Nair, and L. E. Lundgren, “The Use of AI in Mental Health Services to Support Decision-Making: Scoping Review,” J. Med. Internet Res., vol. 27, p. e63548, Jan. 2025, doi: 10.2196/63548.
Abstract views: 305
/
PDF downloads: 90