PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
March 14, 2026Computers2 citationsOpen Access

Employee Attrition Prediction: An Explanatory and Statistically Robust Ensemble Learning Model

View Full Paper
GNGhalia NassreddineJHJamil HammoudOAObada Al-Khatib

Key Points

  • The aim is to develop a predictive model for employee attrition that provides clear interpretability for HR decision-making.
  • Proposed an ensemble learning framework incorporating SHAP for feature selection and LIME for dual explainability.
  • Applied Optuna for hyperparameter optimization to enhance model performance.
  • Utilized random oversampling to address class imbalance for improved predictions.
  • Tested the model on two benchmark datasets (Kaggle HR Analytics and IBM HR datasets).
  • Achieved an accuracy of 98.72%, F1-score of 97.29%, and ROC–AUC of 0.994 on the Kaggle dataset.
  • Reached an accuracy of 97.72%, F1-score of 97.74%, and ROC–AUC of 0.995 on the IBM dataset.
  • Identified promotion history, tenure, job satisfaction, workload, average monthly hours, overtime, and financial incentives as key factors influencing attrition.
  • Demonstrated high computational efficiency making it suitable for real-world applications.

Abstract

Organizational productivity and workforce management are highly affected by employee attrition. Thus, an employee attrition prediction system may allow human resource management to enhance the workplace by minimizing attrition. This study proposes a new and interpretable ensemble learning framework for employee attrition prediction. The model integrates SHapley Additive exPlanations (SHAP)-based feature selection, Optuna hyperparameter optimization, and dual explainability using SHAP and Local Interpretable Model-agnostic Explanations (LIME). Random oversampling (ROS) is used to address class imbalance. The proposed framework allows for both global and local interpretability, enabling actionable insights into retention drivers. It was assessed using two benchmark datasets: the Kaggle HR Analytics dataset (14,999 records) and the IBM HR dataset (1470 records). The results revealed that the most impactful factors on employee attrition are promotion history, tenure, job satisfaction, workload, average monthly hours, overtime, and financial incentives. Furthermore, the proposed model achieved exceptional performance on both datasets. On the Kaggle dataset, it reached an accuracy of 98.72%, an F1-score of 97.29%, and an ROC–AUC of 0.994, while on the IBM dataset, it produced an accuracy of 97.72%, an F1-score of 97.74%, and an ROC–AUC of 0.995. Moreover, the proposed approach shows high computational efficiency, demonstrating that it is suitable for real-world deployment. These findings indicate that integrating explainable AI techniques, resampling tools, and automated hyperparameter tuning can achieve robust, accurate, and actionable employee attrition predictions, supporting HR managers’ decision-making.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Nassreddine et al. (2026) studied this question.

synapsesocial.com/papers/69b4fb9db39f7826a300be85https://doi.org/10.3390/computers15030185
Ask AI
Helpful
Bookmark
Share
View Full Paper