Deep Learning Based Employee Attrition Prediction

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.