Employee Attrition: Analysis of Data Driven Models
DOI:
https://doi.org/10.4108/eetiot.4762Keywords:
Employee attrition, Ensemble Learning, Deep Learning, Machine LearningAbstract
Companies constantly strive to retain their professional employees to minimize the expenses associated with recruiting and training new staff members. Accurately anticipating whether a particular employee is likely to leave or remain with the company can empower the organization to take proactive measures. Unlike physical systems, human resource challenges cannot be encapsulated by precise scientific or analytical formulas. Consequently, machine learning techniques emerge as the most effective tools for addressing this objective. In this paper, we present a comprehensive approach for predicting employee attrition using machine learning, ensemble techniques, and deep learning, applied to the IBM Watson dataset. We employed a diverse set of classifiers, including Logistic regression classifier, K-nearest neighbour (KNN), Decision Tree, Naïve Bayes, Gradient boosting, AdaBoost, Random Forest, Stacking, XG Boost, “FNN (Feedforward Neural Network)”, and “CNN (Convolutional Neural Network)” on the dataset. Our most successful model, which harnesses a deep learning technique known as FNN, achieved superior predictive performance with highest Accuracy, recall and F1-score of 97.5%, 83.93% and 91.26%.
Downloads
References
Mohbey, K.: Employee’s Attrition Prediction Using Machine Learning Approaches. Presented at the January 1 (2020). https://doi.org/10.4018/978-1-7998-3095-5.ch005. DOI: https://doi.org/10.4018/978-1-7998-3095-5.ch005
Alduayj, S.S., Rajpoot, K.: Predicting Employee Attrition using Machine Learning. In: 2018 International Conference on Innovations in Information Technology (IIT). pp. 93–98 (2018). https://doi.org/10.1109/INNOVATIONS.2018.8605976. DOI: https://doi.org/10.1109/INNOVATIONS.2018.8605976
Yedida, R., Reddy, R., Vahi, R., Jana, R., Gv, A., Kulkarni, D.: Employee Attrition Prediction. (2018).
Mansor, N., S Sani, N., Aliff, M.: Machine Learning for Predicting Employee Attrition. Int. J. Adv. Comput. Sci. Appl. 12, 435–445 (2021). https://doi.org/10.14569/IJACSA.2021.0121149. DOI: https://doi.org/10.14569/IJACSA.2021.0121149
Abdulkareem, A.B., Sani, N., Sahran, S., Abdi, Z., Alyessari, A., Adam, A., Rahman, A.H.A., Abdulkarem, A.: Predicting COVID-19 Based on Environmental Factors WithMachine Learning. Presented at the (2021).
Pratt, M., Boudhane, M., Cakula, S.: Employee Attrition Estimation Using Random Forest Algorithm. Balt. J. Mod. Comput. 9, (2021). https://doi.org/10.22364/bjmc.2021.9.1.04. DOI: https://doi.org/10.22364/bjmc.2021.9.1.04
El-rayes, N., Smith, M., Taylor, S.M.: An Explicative and Predictive Study of Employee Attrition using Tree-based Models, https://papers.ssrn.com/abstract=3397445, (2019). https://doi.org/10.2139/ssrn.3397445. DOI: https://doi.org/10.2139/ssrn.3397445
Srivastava, Dr.P., Eachempati, P.: Intelligent Employee Retention System for Attrition Rate Analysis and Churn Prediction: An Ensemble Machine Learning and Multi- Criteria Decision-Making Approach. J. Glob. Inf. Manag. 29, 1– 29 (2021). https://doi.org/10.4018/JGIM.20211101.oa23. DOI: https://doi.org/10.4018/JGIM.20211101.oa23
Yadav, S., Jain, A., Singh, D.: Early Prediction of Employee Attrition using Data Mining Techniques. Presented at the December 1 (2018). https://doi.org/10.1109/IADCC.2018.8692137. DOI: https://doi.org/10.1109/IADCC.2018.8692137
Najafi, S., Shams Gharneh, N., Nezhad, A., Zolfani, S.: An Improved Machine Learning-Based Employees Attrition Prediction Framework with Emphasis on Feature Selection. (2021). https://doi.org/10.3390/MATH9111226. DOI: https://doi.org/10.3390/math9111226
Gao, X., Wen, J., Zhang, C.: An Improved Random Forest Algorithm for Predicting Employee Turnover. Math. Probl. Eng. 2019, 1–12 (2019). https://doi.org/10.1155/2019/4140707. DOI: https://doi.org/10.1155/2019/4140707
Bhatta, S., Zaman, I.U., Raisa, N., Fahim, S.I., Momen, S.: Machine Learning Approach to Predicting Attrition Among Employees at Work. In: Silhavy, R. (ed.) Artificial Intelligence Trends in Systems. pp. 285–294. Springer International Publishing, Cham (2022). https://doi.org/10.1007/978-3-031-09076-9_27. DOI: https://doi.org/10.1007/978-3-031-09076-9_27
Fallucchi, F., Coladangelo, M., Giuliano, R., De Luca, E.: Predicting Employee Attrition Using Machine Learning Techniques. Computers. 9, 86 (2020). https://doi.org/10.3390/computers9040086. DOI: https://doi.org/10.3390/computers9040086
Joseph, R., Udupa, S., Jangale, S., Kotkar, K., Pawar, P.: Employee Attrition Using Machine Learning And Depression Analysis. Presented at the May 6 (2021). https://doi.org/10.1109/ICICCS51141.2021.9432259. DOI: https://doi.org/10.1109/ICICCS51141.2021.9432259
Qutub, A., Al-Mehmadi, A., Al-Hssan, M., Aljohani, R., Alghamdi, H.: Prediction of Employee Attrition Using Machine Learning and Ensemble Methods. Int. J. Mach. Learn. Comput. 11, 110–114 (2021). https://doi.org/10.18178/ijmlc.2021.11.2.1022. DOI: https://doi.org/10.18178/ijmlc.2021.11.2.1022
Arqawi, S., Abu Rumman, M.A., Zitawi, E., Rabaya, A., Sadaqa, A., Abunasser, B., Abu-Naser, S.: PREDICTING EMPLOYEE ATTRITION AND PERFORMANCE USING DEEP LEARNING. 100, 6526–6536 (2022).
Kamath, R., Jamsandekar, S., Naik, P.: Machine Learning Approach for Employee Attrition Analysis. Int. J. Trend Sci.Res. Dev. Special Issue, 62–67 (2019). https://doi.org/10.31142/ijtsrd23065. DOI: https://doi.org/10.31142/ijtsrd23065
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 EAI Endorsed Transactions on Internet of Things
This work is licensed under a Creative Commons Attribution 3.0 Unported License.
This is an open-access article distributed under the terms of the Creative Commons Attribution CC BY 3.0 license, which permits unlimited use, distribution, and reproduction in any medium so long as the original work is properly cited.
