Enhancing Retention Strategies through Deep Learning-Based Dropout Prediction
DOI:
https://doi.org/10.3991/itdaf.v4i1.59233Keywords:
Employee Dropout, Deep learning, SMOTE, GRU, PredictionAbstract
Employees are an organization’s most significant resource. Employee dropout may be costly for firms owing to the costs of recruiting, training, and lost productivity. By forecasting dropout, firms may take preventative actions such as developing retention programs, providing targeted assistance to at-risk employees, and addressing possible workplace concerns. The unpredictable dropout can assist in reducing dropout rates and saving money. Existing approaches to predicting employee dropout use machine learning (ML) techniques for employee dropout prediction, which do not present the correlation of various employee attributes that may have caused the dropout. Moreover, the imbalanced dataset affects the accuracy of prediction results. In this paper, Synthetic Minority Oversampling Technique (SMOTE) is applied to the dataset to solve the issue of imbalanced data. Following that, a deep learning technique, gated recurrent unit (GRU), is utilized to predict staff dropout effectively. It also aided in determining most of the relevant factors of employee results. For this purpose, the IBM employee dataset is utilized for training and assessing GRU using 10-fold test-train splitting. The ultimate objective is to effectively detect dropouts to assist any organization in improving various retention strategies. According to the results, the suggested technique achieves 95% accuracy, more significant than existing state-of-the-art approaches.
References
[1] Han H, Quan W, Al-Ansi A, Chung H, Ngah AH, Ariza-Montes A, Vega-Muñoz A. 2020. A theoretical framework development for hotel employee turnover: Linking trust in supports, emotional exhaustion, depersonalization, and reduced personal accomplishment at workplace. Sustainability 12(19):8065.
[2] Skelton AR, Nattress D, Dwyer RJ. 2020. Predicting manufacturing employee turnover intentions. J Econ Financ Adm Sci 25(49):101–117.
[3] Garg S, Sinha S, Kar AK, Mani M. 2022. A review of machine learning applications in human resource management. Int J Product Perform Manag 71(5):1590–1610.
[4] Sun L, Bunchapattanasakda C, others. 2019. Employee engagement: A literature review. Int J Hum Resour Stud 9(1):63–80.
[5] Degbey WY, Rodgers P, Kromah MD, Weber Y. 2021. The impact of psychological ownership on employee retention in mergers and acquisitions. Hum Resour Manag Rev 31(3):100745.
[6] Jain R, Nayyar A. 2018. Predicting employee attrition using xgboost machine learning approach. In: 2018 Int. Conf. Syst. Model. & Adv. Res. trends. pp 113–120.
[7] Srivastava PR, Eachempati P. 2021. 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(6):1–29.
[8] McCarthy D, Alexander P, Jung Y. 2022. Enhancing the organisational commitment of public sector accounting staff through the pursuit of CSR objectives. J Account & Organ Chang 18(2):304–324.
[9] Hmoud B, Laszlo V, others. 2019. Will artificial intelligence take over human resources recruitment and selection. Netw Intell Stud 7(13):21–30.
[10] Yasin R. 2021. Responsible leadership and employees’ turnover intention. Explore the mediating roles of ethical climate and corporate image. J Knowl Manag 25(7):1760–1781.
[11] Jain N, Tomar A, Jana PK. 2021. A novel scheme for employee churn problem using multi-attribute decision making approach and machine learning. J Intell Inf Syst 56:279–302.
[12] Jain PK, Jain M, Pamula R. 2020. Explaining and predicting employees’ attrition: a machine learning approach. SN Appl Sci 2:1–11.
[13] Zhang W, Li H, Tang L, Gu X, Wang L, Wang L. 2022. Displacement prediction of Jiuxianping landslide using gated recurrent unit (GRU) networks. Acta Geotech 17(4):1367–1382.
[14] Li C, Tang G, Xue X, Saeed A, Hu X. 2019. Short-term wind speed interval prediction based on ensemble GRU model. IEEE Trans Sustain energy 11(3):1370–1380.
[15] Kim T-Y, Cho S-B. 2019. Predicting residential energy consumption using CNN-LSTM neural networks. Energy 182:72–81.
[16] Kumar TS. 2020. Data mining based marketing decision support system using hybrid machine learning algorithm. J Artif Intell 2(03):185–193.
[17] Tahir S, Hafeez Y, Abbas MA, Nawaz A, Hamid B. 2022. Smart Learning Objects Retrieval for E-Learning with Contextual Recommendation based on Collaborative Filtering. Educ Inf Technol :1–38.
[18] Rasjid ZE, Setiawan R. 2017. Performance Comparison and Optimization of Text Document Classification using k-NN and Na{"i}ve Bayes Classification Techniques. Procedia Comput Sci 116:107–112.
[19] Zelaya CVG. 2019. Towards explaining the effects of data preprocessing on machine learning. In: 2019 IEEE 35th Int. Conf. data Eng. pp 2086–2090.
[20] Ali S, Hafeez Y, Humayun M, Jamail NSM, Aqib M, Nawaz A. 2021. Enabling recommendation system architecture in virtualized environment for e-learning. Egypt Informatics J. doi: 10.1016/j.eij.2021.05.003.
[21] Dutta S, Bandyopadhyay SK, Kumar Bandyopadhyay S. 2020. Employee attrition prediction using neural network cross validation method. Int J Commer Manag Res 6(3):80–85.
[22] Zhao Y, Hryniewicki MK, Cheng F, Fu B, Zhu X. 2019. Employee turnover prediction with machine learning: A reliable approach. In: Intell. Syst. Appl. Proc. 2018 Intell. Syst. Conf. Vol. 2. pp 737–758.
[23] Hang J, Dong Z, Zhao H, Song X, Wang P, Zhu H. 2022. Outside in: Market-aware heterogeneous graph neural network for employee turnover prediction. In: Proc. Fifteenth ACM Int. Conf. Web Search Data Min. pp 353–362.
[24] Al-Darraji S, Honi DG, Fallucchi F, Abdulsada AI, Giuliano R, Abdulmalik HA. 2021. Employee attrition prediction using deep neural networks. Computers 10(11):141.
[25] Pekel Ozmen E, Ozcan T. 2022. A novel deep learning model based on convolutional neural networks for employee churn prediction. J Forecast 41(3):539–550.
[26] Yahia N Ben, Hlel J, Colomo-Palacios R. 2021. From big data to deep data to support people analytics for employee attrition prediction. IEEE Access 9:60447–60458.
[27] Alsheref FK, Fattoh IE, M Ead W. 2022. Automated Prediction of Employee Attrition Using Ensemble Model Based on Machine Learning Algorithms. Comput. Intell. Neurosci. 2022:.
[28] Jadhav A, others. 2021. Churn prediction of employees using machine learning techniques. Teh Glas 15(1):51–59.
[29] Gao X, Wen J, Zhang C. 2019. An improved random forest algorithm for predicting employee turnover. Math. Probl. Eng. 2019:.
[30] Fallucchi F, Coladangelo M, Giuliano R, William De Luca E. 2020. Predicting employee attrition using machine learning techniques. Computers 9(4):86.
[31] Najafi-Zangeneh S, Shams-Gharneh N, Arjomandi-Nezhad A, Hashemkhani Zolfani S. 2021. An Improved Machine Learning-Based Employees Attrition Prediction Framework with Emphasis on Feature Selection. Mathematics 9(11):1226.
[32] Alamsyah A, Salma N. 2018. A comparative study of employee churn prediction model. In: 2018 4th Int. Conf. Sci. Technol. pp 1–4.
[33] Wang X, Zhi J. 2021. A machine learning-based analytical framework for employee turnover prediction. J Manag Anal 8(3):351–370.
[34] Mozaffari F, Rahimi M, Yazdani H, Sohrabi B. 2022. Employee attrition prediction in a pharmaceutical company using both machine learning approach and qualitative data. Benchmarking An Int. J. .
[35] Ali Shah SA, Uddin I, Aziz F, Ahmad S, Al-Khasawneh MA, Sharaf M. 2020. An enhanced deep neural network for predicting workplace absenteeism. Complexity 2020:1–12.
[36] Alduayj SS, Rajpoot K. 2018. Predicting employee attrition using machine learning. In: 2018 Int. Conf. Innov. Inf. Technol. pp 93–98.
[37] Chen J, Huang H, Cohn AG, Zhang D, Zhou M. 2022. Machine learning-based classification of rock discontinuity trace: SMOTE oversampling integrated with GBT ensemble learning. Int J Min Sci Technol 32(2):309–322.
[38] Fernández A, Garcia S, Herrera F, Chawla N V. 2018. SMOTE for learning from imbalanced data: progress and challenges, marking the 15-year anniversary. J Artif Intell Res 61:863–905.
[39] Richards JA, Richards JA. 2022. Feature reduction. Remote Sens Digit Image Anal :403–446.
[40] Abdulhammed R, Musafer H, Alessa A, Faezipour M, Abuzneid A. 2019. Features dimensionality reduction approaches for machine learning based network intrusion detection. Electronics 8(3):322.
[41] Alasadi SA, Bhaya WS. 2017. Review of data preprocessing techniques in data mining. J Eng Appl Sci 12(16):4102–4107.
[42] Kamiran F, Calders T. 2012. Data preprocessing techniques for classification without discrimination. Knowl Inf Syst 33(1):1–33.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2026 Ibrar Hussain, Sidra Tahir, Asif Nawaz, Kashif Mehmood, ahthasham Sajid, Sabitha Banu
This work is licensed under a Creative Commons Attribution 4.0 International License.
