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.
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.
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.
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.
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.
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.
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.
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.
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.
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.
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.
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.
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.
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.
Downloads
Published
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 4.0 license, which permits unlimited use, distribution, and reproduction in any medium so long as the original work is properly cited.
