Employee Attrition: Analysis of Data Driven Models


  • Manju Nandal Noida Institute of Engineering and Technology
  • Veena Grover Noida Institute of Engineering and Technology
  • Divya Sahu Noida Institute of Engineering and Technology
  • Mahima Dogra Noida Institute of Engineering and Technology




Employee attrition, Ensemble Learning, Deep Learning, Machine Learning


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%.


Download data is not yet available.
<br data-mce-bogus="1"> <br data-mce-bogus="1">


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




How to Cite

M. Nandal, V. Grover, D. Sahu, and M. Dogra, “Employee Attrition: Analysis of Data Driven Models”, EAI Endorsed Trans IoT, vol. 10, Jan. 2024.