Implementation of the K-Means Clustering Algorithm for Segmenting Employee Mental Health Profiles Based on Work Productivity Indicators

Maulia Rahman; Dedi Leman

doi:10.47709/cnahpc.v7i4.6974

Authors

Maulia Rahman Universitas Potensi Utama
Dedi Leman Institut Bisnis dan Komputer Indonesia, Indonesia

DOI:

https://doi.org/10.47709/cnahpc.v7i4.6974

Keywords:

K-Means clustering, employee mental health, work productivity, working from home, segmentation

Abstract

This study aims to identify mental health profile segmentation among employees based on work productivity indicators in the context of working from home (WFH) using the K-Means clustering algorithm. This study uniquely integrates mental health and productivity indicators into an unsupervised clustering framework. A cross-sectional method was conducted on 100 employee respondents with 10 main variables, analysed using K-Means with four optimal cluster evaluation methods. The results identified four distinct segments: Low WFH Adaptation (25%), High WFH Enthusiasts (30%), Mixed Preference (25%), and Office Preference (20%), with Silhouette Score validation of 0.623 and Davies-Bouldin Index of 0.967. The main findings reveal the paradox of High WFH Enthusiasts, who have the highest productivity (93%) but the highest mental health risk (1.90). This segmentation provides a practical framework for developing personalised mental health intervention strategies in employee management in the remote working era.

Downloads

Download data is not yet available.

References

Chen, Y., & Li, J. (2024). Computer-Aided Recognition Algorithm for Students’ Mental Health Problems using K-means Clustering. Computer-Aided Design and Applications, 21(S9), 19–37. https://doi.org/10.14733/cadaps.2024.S9.19-37

Elsamani, Y., Mejia, C., & Kajikawa, Y. (2023). Employee well-being and innovativeness: A multi-level conceptual framework based on citation network analysis and data mining techniques. In PLoS ONE (Vol. 18, Issue 1 January). https://doi.org/10.1371/journal.pone.0280005

Errichiello, L., & Pianese, T. (2021). The Role of Organizational Support in Effective Remote Work Implementation in the Post-COVID Era. March 2021, 221–242. https://doi.org/10.4018/978-1-7998-6754-8.ch013

Fadhil, Z. M. (2021). Hybrid of K-means clustering and naive Bayes classifier for predicting performance of an employee. Periodicals of Engineering and Natural Sciences, 9(2), 799–807. https://doi.org/10.21533/pen.v9i2.1898

Kerman, K., Korunka, C., & Tement, S. (2021). Work and home boundary violations during the COVID-19 pandemic: The role of segmentation preferences and unfinished tasks. In Applied Psychology (pp. 784–806). doi: 10.1111/apps.12335

Krishnan, C., Singh, S., & Baba, M. M. (2024). Effect of work from home and employee mental health through mediating role of workaholism and work-family balance. International Journal of Social Psychiatry, 70(1), 144–156. https://doi.org/10.1177/00207640231196741

Lei, J. (2022). An Analytical Model of College Students’ Mental Health Education Based on the Clustering Algorithm. Mathematical Problems in Engineering, 2022. https://doi.org/10.1155/2022/1880214

Liu, F., Yang, D., Liu, Y., Zhang, Q., Chen, S., Li, W., Ren, J., Tian, X., & Wang, X. (2022). Use of latent profile analysis and k-means clustering to identify student anxiety profiles. BMC Psychiatry, 22(1), 1–11. https://doi.org/10.1186/s12888-021-03648-7

Lunde, L. K., Fløvik, L., Christensen, J. O., Johannessen, H. A., Finne, L. B., Jørgensen, I. L., Mohr, B., & Vleeshouwers, J. (2022). The relationship between telework from home and employee health: a systematic review. BMC Public Health, 22(1), 1–14. https://doi.org/10.1186/s12889-021-12481-2

Muntyanu, A., Gabrielli, S., Donovan, J., Gooderham, M., Guenther, L., Hanna, S., Lynde, C., Prajapati, V. H., Wiseman, M., & Netchiporouk, E. (2023). The burden of alopecia areata: A scoping review focusing on quality of life, mental health and work productivity. Journal of the European Academy of Dermatology and Venereology, 37(8), 1490–1520. https://doi.org/10.1111/jdv.18926

Oakman, J., Kinsman, N., Stuckey, R., Graham, M., & Weale, V. (2020). A rapid review of mental and physical health effects of working at home: how do we optimise health? BMC Public Health, 20(1), 1–13. https://doi.org/10.1186/s12889-020-09875-z

Pettenuzzo, D., Sabbatucci, R., & Timmermann, A. (2023). Payout suspensions during the Covid-19 pandemic. Economics letters, 224, 111024.

Phadnis, S., Sengupta, S., & Chakraborty, A. (2021). Work from home, mental health and employee needs: A pilot study in selected information technology organizations in india. Asia Pacific Journal of Health Management, 16(3). https://doi.org/10.24083/apjhm.v16i3.977

Rosca, C. M., & Stancu, A. (2024). Fusing Machine Learning and AI to Create a Framework for Employee Well-Being in the Era of Industry 5.0. Applied Sciences (Switzerland), 14(23). https://doi.org/10.3390/app142310835

Shabir, G. (2024). Optimizing Workforce Efficiency: Leveraging Integrated Business Analytics and Machine Learning for Enhanced Performance Prediction. Research Gate, 23 March 2024, 2–10. https://doi.org/10.13140/RG.2.2.14125.50404

Van Zoonen, W., & Sivunen, A. E. (2022). The impact of remote work and mediated communication frequency on isolation and psychological distress. European Journal of Work and Organizational Psychology, 31(4), 610–621. https://doi.org/10.1080/1359432X.2021.2002299

Wells, J., Scheibein, F., Pais, L., Rebelo dos Santos, N., Dalluege, C. A., Czakert, J. P., & Berger, R. (2023). A Systematic Review of the Impact of Remote Working Referenced to the Concept of Work–Life Flow on Physical and Psychological Health. Workplace Health and Safety, 71(11), 507–521. https://doi.org/10.1177/21650799231176397

Yousefi, M., & Vanani, I. R. (2020). A Model for the Mental Health Evaluation of Skilled Workers in Technology-Based Companies.

Yuan, S., Kroon, B., & Kramer, A. (2024). Building prediction models with grouped data: A case study on the prediction of turnover intention. Human Resource Management Journal, 34(1), 20–38. https://doi.org/10.1111/1748-8583.12396

Zhang, X., Yang, W., Yang, W., Huang, B., Wang, Z., & Tian, S. (2025). Occupational Risk Prediction for Miners Based on Stacking Health Data Fusion. Applied Sciences (Switzerland), 15(6), 1–23. https://doi.org/10.3390/app15063129