Optimizing IT Remote Workers Mental Health Prediction using Feature Engineering
DOI:
https://doi.org/10.62951/ijsme.v2i2.193Keywords:
Random Forest, Classification, Machine Learning, Mental HealthAbstract
Given the increasing prevalence of mental health challenges in digital work settings, especially among IT remote workers, early detection mechanisms have become critically important. This study aims to improve the prediction accuracy of mental health conditions among IT remote workers by integrating feature engineering techniques within machine learning models. Five algorithms consisting of Random Forest, Logistic Regression, K-Nearest Neighbors, Decision Tree, and Naive Bayes were evaluated. The Random Forest model achieved the best performance, with 83% accuracy, 83% precision, 100% recall, and a 90% F1-score, followed closely by Logistic Regression with 82% accuracy. Nevertheless, the results demonstrate the feasibility of applying machine learning to support the early detection of mental health risks, offering a strong foundation for future research in predictive analytics and the development of intelligent support systems within digital work environments.
References
Charalampous, M., Grant, C., Tramontano, C., & Michailidis, E. (2018). Systematically reviewing remote e-workers’ well-being at work: a multidimensional approach. European Journal of Work and Organizational Psychology, 28, 51–73. https://doi.org/10.1080/1359432X.2018.1541886
Cheng, J.-P., & Haw, S.-C. (2023). Mental Health Problems Prediction Using Machine Learning Techniques. International Journal on Robotics, Automation and Sciences, 5(2), 59–72. https://doi.org/10.33093/ijoras.2023.5.2.7
Glaz, L., Haralambous, Y., Kim-Dufor, D.-H., Lenca, P., Billot, R., Ryan, T., Marsh, J., Devylder, J., Walter, M., Berrouiguet, S., & Lemey, C. (2019). Machine Learning and Natural Language Processing in Mental Health: Systematic Review. Journal of Medical Internet Research, 23. https://doi.org/10.2196/15708
Idaiani, S., & Waris, L. (2022). Depression and Psychological Stress Among Health Workers in Remote Areas in Indonesia. Frontiers in Public Health, 10. https://doi.org/10.3389/fpubh.2022.743053
Kaushik, P., Jain, E., Gill, K. S., Upadhyay, D., & Devliyal, S. (2024). Optimizing Mental Health Prediction by Fine-Tuning Decision Classifier Parameters for Enhanced Accuracy. 2024 2nd International Conference on Sustainable Computing and Smart Systems (ICSCSS), 935–939. https://api.semanticscholar.org/CorpusID:271934379
Lyzwinski, L.-N. (2024). Organizational and occupational health issues with working remotely during the pandemic: a scoping review of remote work and health. Journal of Occupational Health, 66. https://doi.org/10.1093/joccuh/uiae005
Mallick, S., & Panda, M. (2024). Predictive Modeling of Mental Illness in Technical Workplace: A Feature Selection and Classification Approach. 2024 OPJU International Technology Conference (OTCON) on Smart Computing for Innovation and Advancement in Industry 4.0, 1–7. https://api.semanticscholar.org/CorpusID:272996374
McKinsey. (2021, February 18). The future of work after COVID-19. Https://Www.Mckinsey.Com/Featured-Insights/Future-of-Work/the-Future-of-Work-after-Covid-19. https://www.mckinsey.com/featured-insights/future-of-work/the-future-of-work-after-covid-19
Mohammad, B. N. S., & Siddiqui, K. (2021). Random Forest Regressor Machine Learning Model Developed for Mental Health Prediction Based on Mhi-5, Phq-9 and Bdi Scale. SSRN Electronic Journal. https://api.semanticscholar.org/CorpusID:237961112
Nash, C., Nair, R., & Naqvi, S. (2023). Machine Learning in ADHD and Depression Mental Health Diagnosis: A Survey. IEEE Access, 11, 86297–86317. https://doi.org/10.1109/ACCESS.2023.3304236
Nowrouzi-Kia, B., Haritos, A., Long, B.-Z. S., Atikian, C., Fiorini, L., Gohar, B., Howe, A., Li, Y., & Bani-Fatemi, A. (2024). Remote work transition amidst COVID-19: Impacts on presenteeism, absenteeism, and worker well-being—A scoping review. PLOS ONE, 19. https://doi.org/10.1371/journal.pone.0307087
Pritam, N., Gill, K. S., Kumar, M., Rawat, R., & Banerjee, D. (2024). Classification of Student Mental Health Analysis using Logistic Regression and other classification techniques through Machine Learning Methods. 2024 3rd International Conference for Innovation in Technology (INOCON), 1–5. https://doi.org/10.1109/INOCON60754.2024.10512216
Sarinastiti, N., Bimo, A., & Cole, J. (2022). RELATIONS OF REMOTE WORKING TO MENTAL HEALTH. ASPIRATION Journal. https://doi.org/10.56353/aspiration.v2i2.40
Shatte, A., Hutchinson, D., & Teague, S. (2019). Machine learning in mental health: a scoping review of methods and applications. Psychological Medicine, 49, 1426–1448. https://doi.org/10.1017/S0033291719000151
Sim, J., Yun, B., Park, H., Oh, J., & Yoon, J. (2024). Rising Mental Health Issues from Remote Work Challenges in South Korea Amid COVID-19. The European Journal of Public Health, 34. https://doi.org/10.1093/eurpub/ckae144.2208
Sofianti, T. D., Kurniawan, I., Pratama, A. T., & Florencia, J. (2023). Measuring Worker Perception on Remote Working Adoption During COVID-19 Pandemic: An Industrial Engineering Perspective. Engineering Science Letter. https://api.semanticscholar.org/CorpusID:258233375
Vaishnavi, K., Nikhitha Kamath, U., Ashwath Rao, B., & Subba Reddy, N. V. (2022). Predicting Mental Health Illness using Machine Learning Algorithms. Journal of Physics: Conference Series, 2161(1), 12021. https://doi.org/10.1088/1742-6596/2161/1/012021
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 International Journal of Science and Mathematics Education
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
