Abstract
As corporations go through the Great Resignation, a post-pandemic economic trend in the surge of employee resignations, employee attrition has become one of the most significant problems for any organization. Employee attrition is defined as a reduction in the number of workers from various causes such as retirement, resignation, and termination. Because employees are important human resources (HR) of the organization and the subjects who own other valuable resources that the organization need, diverse opportunity costs occur when employee attrition takes place. To prevent such unwanted loss of valuable assets, various efforts have been made to predict and prevent employee attrition. In this study, three machine learning methods, Random Forest, XGBoost, and Artificial Neural Network, were used to predict employee attrition. Kaggle’s IBM HR Analytics Employee Attrition and Performance dataset which is composed of 1470 employee information was used as the data set. The variable to be predicted was whether or not employees leave the organization and a total of 35 variables such as academic background and environment satisfaction were considered. ‘Accuracy’, ‘Precision’, ‘Sensitivity’, and ‘F-1 Score’ were used as measures to calculate the prediction performance of the models. The result showed that XGBoost has the best performance in Accuracy while Random Forest showed the best performance in Precision. Artificial Neural Network showed the best performance in both Sensitivity and F1-Score.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Hopkins, J.C., Figaro, K.A.: The great resignation: an argument for hybrid leadership. Int. J. Bus. Manage. Res. 9(4), 393–400 (2021)
Klotz, A.C., Zimmerman, R.D.: On the turning away: an exploration of the employee resignation process. In: Research in Personnel and Human Resources Management. Emerald Group Publishing Limited (2015)
Watts, J.M.: Resignation. Fire Technol. 49(1), 1–2 (2013)
Allman, K.: Career matters: ‘The great resignation’ sweeping workplaces around the world. LSJ Law Soc. NSW J. 81, 46–47 (2021)
Klotz, A.C., Bolino, M.C.: Saying goodbye: the nature, causes, and consequences of employee resignation styles. J. Appl. Psychol. 101(10), 13986 (2016)
Cho, G.S.: The effects of commitment to organizational change on employee`s turnover intention: an empirical investigation for Korean bank employees under mergers and acquisitions. J. Hum. Resour. Manage. Res. 13(1), 167–182 (2006)
Chakraborty, R., Mridha, K., Shaw, R., Ghosh, A.: Study and prediction analysis of the employee turnover using machine learning approaches. In: 2021 IEEE 4th International Conference on Computing, Power and Communication Technologies (GUCON), pp. 1–6 (2021)
Sisodia, D.S., Vishwakarma, S., Pujahari, A.: Evaluation of machine learning models for employee churn prediction. In: 2017 International Conference on Inventive Computing and Informatics (ICICI), pp. 1016–1020 (2017)
Kwon K.: The Relationship between employee turnover and firm performance: an explorative study, 16(1), 1–26. Korea Labor Institute (2016)
PARK, Y., Lee, D.G.: Development of a resignation prediction model using HR data. In: Proceedings of the Korean Institute of Information and Communication Sciences Conference, pp. 100–103. The Korea Institute of Information and Communication Engineering (2021)
Zhao, Y., Hryniewicki, M.K., Cheng, F., Fu, B., Zhu, X.: Employee turnover prediction with machine learning: a reliable approach. Adv. Intell. Syst. Comput. 869 (2019)
Alao, D.A.B.A., Adeyemo, A.B.: Analyzing employee attrition using decision tree algorithms. Comput. Inf. Syst. Dev. Inf. Allied Res. J. 4(1), 17–28 (2013)
Tzeng, H.M., Hsieh, J.G., Lin, Y.L.: Predicting nurses’ intention to quit with a support vector machine. CIN Comput. Inf. Nurs. 22(4), 232–242 (2004)
Ajit, P.: Prediction of employee turnover in organizations using machine learning algorithms. Algorithms 4(5), C5 (2016)
Sexton, R.S., McMurtrey, S., Michalopoulos, J.O., Smith, A.M.: Employee turnover: a neural network solution. Comput. Oper. Res. 32, 2635–2651 (2005)
Belgiu, M., Drăguţ, L.: Random forest in remote sensing: a review of applications and future directions. ISPRS J. Photogram. Remote Sens. 114, 24–31 (2016)
Breiman, L.: Random forests. Mach. Learn. 45, 5–32 (2001)
Valle, M.A., Ruz, G.A.: Turnover prediction in a call center: behavioral evidence of loss aversion using random forest and naïve bayes algorithms. Appl. Artif. Intell. 29(9), 923–942 (2015)
Chen, T.: Xgboost: extreme gradient boosting. R Package Version. 0.4-2 1(4), 1–4 (2015)
Kim, Y., Choi, H., Kim, S.: A study on risk parity asset allocation model with XGBoost. J. Intell. Inf. Syst. 26(1), 135–149 (2020)
Jain, R., Nayyar, A.: Predicting employee attrition using XGBoost machine learning approach. In: 2018 International Conference On System Modeling and Advancement in Research Trends (SMART). IEEE (2018)
Gim, G.: Kaggle Data Strategy Practice Using SPSS, R, Python, 1st edn. Cheong-Ram Publication (2022)
Somers, M.J.: Application of two neural network paradigms to the study of voluntary employee turnover. J. Appl. Psychol. 84(2), 177 (1999)
Esmaieeli Sikaroudi, A.M., Ghousi, R., Sikaroudi, A.: A data mining approach to employee turnover. J. Ind. Syst. Eng. 8(4), 106–123 (2015)
Im, E.T., Gim, G.: Developing AI models with the AutoML platform Wise Prophet, R, Python, 1st edn. Cheong-Ram Publication (2022)
Umang, S.: A comparison study between ANN and ANFIS for the prediction of employee turnover in an organization. In: 2018 International Conference on Computing, Power and Communication Technologies (GUCON). IEEE (2018)
Jain, A.K., Mao, J., Mohiuddin, K.M.: Artificial neural networks: a tutorial. Computer 29(3), 31–44 (1996)
Kaggle, IBM HR analytics employe attrition & performance. https://www.kaggle.com/datasets/pavansubhasht/ibm-hr-analytics-attrition-dataset
Li, J., Cheng, K., Wang, S., Morstatter, F., Trevino, R.P., Tang, J., Liu, H.: Feature selection: a data perspective. ACM Comput. Surv. (CSUR) 50(6), 1–45 (2017)
Jović, A., Brkić, K., Bogunović, N.: A review of feature selection methods with applications. In: 2015 38th International Convention on Information and Communication Technology, Electronics and Microelectronics (MIPRO), pp. 1200–1205. IEEE (2015)
Lee, K., Hong C., Lee, E.H., Yang, W.H.: Comparison of artificial intelligence methods for prediction of mechanical properties. IOP Conf. Ser. Mater. Sci. Eng. 967(1), 012031. IOP Publishing (2020)
Bühlmann, P.: Bagging, boosting and ensemble methods. In: Handbook of Computational Statistics, pp. 985–1022. Springer, Berlin (2012)
Friedman, J.H.: Stochastic gradient boosting. Comput. Stat. Data Anal. 38, 367–378 (2002)
Wu, Y.-C., Feng, J.-W.: Development and application of artificial neural network. Wireless Pers. Commun. 102(2), 1645–1656 (2018)
Jenkins, B.K., Tanguay, A.R.: Handbook of Neural Computing and Neural Networks. MIT Press, Boston (1995)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this chapter
Cite this chapter
Gim, S., Im, E.T. (2023). A Study on Predicting Employee Attrition Using Machine Learning. In: Lee, R. (eds) Big Data, Cloud Computing, and Data Science Engineering. BCD 2022. Studies in Computational Intelligence, vol 1075. Springer, Cham. https://doi.org/10.1007/978-3-031-19608-9_5
Download citation
DOI: https://doi.org/10.1007/978-3-031-19608-9_5
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-19607-2
Online ISBN: 978-3-031-19608-9
eBook Packages: Intelligent Technologies and RoboticsIntelligent Technologies and Robotics (R0)