Predictive model of employee attrition based on stacking ensemble learning

•A data scientific approach makes it possible to predict employee attrition.•A stacking ensemble model shows higher performance than all existing models.•The ensemble model uses a combination of LR, RF, XGBoost, ANN.•Environment & relationship satisfaction are the most important prediction facto...

Full description

Saved in:
Bibliographic Details
Published in:Expert systems with applications 2023-04, Vol.215, p.119364, Article 119364
Main Authors: Chung, Doohee, Yun, Jinseop, Lee, Jeha, Jeon, Yeram
Format: Article
Language:English
Subjects:
Employee attrition
Ensemble
Machine learning
Predictive model
Citations: Items that this one cites
Items that cite this one
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:•A data scientific approach makes it possible to predict employee attrition.•A stacking ensemble model shows higher performance than all existing models.•The ensemble model uses a combination of LR, RF, XGBoost, ANN.•Environment & relationship satisfaction are the most important prediction factors. Since human resource is the most important resource of a company, employee attrition is an important agenda from the company's point of view. However, employee attrition occurs due to various reasons, and it is difficult for the HR manager or the leader of each department to know these signs in advance. Employee attrition results in considerable burdens and losses of the organization due to a variety of reasons such as interruption of ongoing tasks, cost of employee re-employment and retraining, and risk of leaking core technologies and know-hows. Therefore, in this study, we propose a model for predicting employee attrition so that we can take measures for talent management which in the past, has been carried out ex post. In this study, a predictive model was constructed based on 30 variables - that affect employee attrition - from the 'IBM HR Analytics Employee Attrition & Performance data', which consists of 1,470 records. To this end, a total of eight predictive models, including Logistic Regression, Random Forest, XGBoost, SVM, Artificial Neural Network model and ensemble model, were built and their performance was evaluated. In addition, when the impact of variables on employee attrition was analyzed, variables such as environmental satisfaction, overtime work, and relationship satisfaction were found to be the biggest contributors.
ISSN:0957-4174
1873-6793
DOI:10.1016/j.eswa.2022.119364