An End-to-End Framework for Predicting Employee Attrition with Explainable AI

Employee attrition is a persistent challenge for organizations, generating significant costs and impacting productivity. While machine learning can predict employee turnover, common approaches struggle with imbalanced data and lack interpretability for actionable insights. This paper presents a complete, end-toend framework for attrition prediction that is both accurate and transparent. Using the IBM HR Analytics dataset, the methodology features advanced preprocessing, engineered attributes, and class balancing via the Synthetic Minority Oversampling Technique (SMOTE). A Random Forest classifier is employed, withmodel evaluation based on Recall, F1-Score, Precision, and AUC-ROC — metrics aligned with the business goal of accurately identifying at-risk employees. To ensure decision transparency, Explainable AI tools SHAP and LIME are integrated, providing both global and local interpretability. Results reveal that the proposed model identifies 80% of actual attrition cases while highlighting key drivers such as overtime workload, compensation levels, and job satisfaction. This interpretability supports targeted retention strategies instead of generic interventions. The study demonstrates that combining machine learning withXAI techniques enables both reliable prediction and actionable insights. Future work includes integrating career progression timelines and deploying real-time attrition monitoring dashboards.

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2025-08-17
DOI
https://doi.org/10.5281/zenodo.16889363
Primary Topic
X-ray Diffraction in Crystallography
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

An End-to-End Framework for Predicting Employee Attrition with Explainable AI

Zenodo (CERN European Organization for Nuclear Research)
X-ray Diffraction in Crystallography
article

An End-to-End Framework for Predicting Employee Attrition with Explainable AI

article en

Abstract

Employee attrition is a persistent challenge for organizations, generating significant costs and impacting productivity. While machine learning can predict employee turnover, common approaches struggle with imbalanced data and lack interpretability for actionable insights. This paper presents a complete, end-toend framework for attrition prediction that is both accurate and transparent. Using the IBM HR Analytics dataset, the methodology features advanced preprocessing, engineered attributes, and class balancing via the Synthetic Minority Oversampling Technique (SMOTE). A Random Forest classifier is employed, withmodel evaluation based on Recall, F1-Score, Precision, and AUC-ROC — metrics aligned with the business goal of accurately identifying at-risk employees. To ensure decision transparency, Explainable AI tools SHAP and LIME are integrated, providing both global and local interpretability. Results reveal that the proposed model identifies 80% of actual attrition cases while highlighting key drivers such as overtime workload, compensation levels, and job satisfaction. This interpretability supports targeted retention strategies instead of generic interventions. The study demonstrates that combining machine learning withXAI techniques enables both reliable prediction and actionable insights. Future work includes integrating career progression timelines and deploying real-time attrition monitoring dashboards.

Zenodo (CERN European Organization for Nuclear Research)
Decent work and economic growth
Openalex Percentile: Top 59%
X-ray Diffraction in Crystallography
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.