19th AIAI 2023, 14 - 17 June 2023, León, Spain

Deep Learning Based Employee Attrition Prediction

Kerem Gürler, Burcu Kuleli Pak, Vehbi Cagri Gungor

Abstract:

  Employee attrition is a critical issue for the business sectors as leaving employees cause various types of difficulties for the company. Some studies exist on examining the reasons for this phenomenon and predicting it with Machine Learning algorithms. In this paper, the causes for employee attrition is explored in three datasets, one of them being our own novel dataset and others obtained from Kaggle. Employee attrition was predicted with multiple Machine Learning and Deep Learning algorithms with feature selection and hyperparameter optimization and their performances are evaluated with multiple metrics. Deep Learning methods showed superior performances in all of the datasets we explored. SMOTE Tomek Links were utilized to oversample minority classes and effectively tackle the problem of class imbalance. Best performing methods were Deep Random Forest on HR Dataset from Kaggle and Neural Network for IBM and Adesso datasets with F1 scores of 0.972, 0.642 and 0.853, respectively.  

*** Title, author list and abstract as seen in the Camera-Ready version of the paper that was provided to Conference Committee. Small changes that may have occurred during processing by Springer may not appear in this window.