Predicting Employee Attrition: A Comparative Analysis of Machine Learning Models Using the IBM Human Resource Analytics Dataset

In employee turnover research and workforce management, addressing the impacts of suboptimal employee performance is crucial for organizations of all sizes and industries. Utilizing advanced machine learning classification models to predict potential employee resignations can enhance human resource departments’ intervention strategies, effectively mitigating attrition challenges. This research investigates the performance of various machine learning algorithms in classification tasks, focusing on their accuracy in predicting outcomes from a given dataset. Five models were evaluated: Random Forest, Support Vector Machine (SVM), Decision Tree, Gradient Boosting, and a Hybrid Model that integrates multiple algorithms. The goal was to identify which model yields the highest accuracy and to understand the strengths and weaknesses of each approach. Results showed that the Hybrid Model achieved the highest accuracy at 95.0%, suggesting that combining different algorithms effectively harnesses their strengths while mitigating individual weaknesses. The SVM accurately classified the instance with 88.6%, demonstrating its capability to manage complex decision boundaries in high-dimensional spaces. Both Random Forest and Gradient Boosting attained an accuracy of 87.3%, reflecting their ensemble techniques that enhance predictive performance by reducing overfitting and optimizing error reduction. In comparison, the Decision Tree classifier exhibited the least accuracy at 80.5%, highlighting its susceptibility to overfitting and limited generalizability. The superior performance of the Hybrid Model indicates a promising direction for future research, where integrating diverse algorithms could lead to more robust predictions. Overall, this study provides valuable insights for practitioners and researchers seeking to optimize model selection and improve predictive accuracy in their domains.

Authors

Institutions

Publication Details

Journal
Procedia Computer Science
Published
2025-01-01
DOI
https://doi.org/10.1016/j.procs.2025.04.659
Citations
4
Primary Topic
AI and HR Technologies
Type
article
Field-Weighted Citation Impact
11.05
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Predicting Employee Attrition: A Comparative Analysis of Machine Learning Models Using the IBM Human Resource Analytics Dataset

Rajkumar Govindarajan, N. Komal Kumar, Pavan Kumar G, S. E. et al.
4 citations
Procedia Computer Science
AI and HR Technologies
11.05
article

Predicting Employee Attrition: A Comparative Analysis of Machine Learning Models Using the IBM Human Resource Analytics Dataset

Rajkumar Govindarajan, N. Komal Kumar, Pavan Kumar G, S. E., ШЕВЧЕНКО О.П. ШЕВЧЕНКО О.П., Bhagavathula Siva Dhatri
article en
4 citations

Abstract

In employee turnover research and workforce management, addressing the impacts of suboptimal employee performance is crucial for organizations of all sizes and industries. Utilizing advanced machine learning classification models to predict potential employee resignations can enhance human resource departments’ intervention strategies, effectively mitigating attrition challenges. This research investigates the performance of various machine learning algorithms in classification tasks, focusing on their accuracy in predicting outcomes from a given dataset. Five models were evaluated: Random Forest, Support Vector Machine (SVM), Decision Tree, Gradient Boosting, and a Hybrid Model that integrates multiple algorithms. The goal was to identify which model yields the highest accuracy and to understand the strengths and weaknesses of each approach. Results showed that the Hybrid Model achieved the highest accuracy at 95.0%, suggesting that combining different algorithms effectively harnesses their strengths while mitigating individual weaknesses. The SVM accurately classified the instance with 88.6%, demonstrating its capability to manage complex decision boundaries in high-dimensional spaces. Both Random Forest and Gradient Boosting attained an accuracy of 87.3%, reflecting their ensemble techniques that enhance predictive performance by reducing overfitting and optimizing error reduction. In comparison, the Decision Tree classifier exhibited the least accuracy at 80.5%, highlighting its susceptibility to overfitting and limited generalizability. The superior performance of the Hybrid Model indicates a promising direction for future research, where integrating diverse algorithms could lead to more robust predictions. Overall, this study provides valuable insights for practitioners and researchers seeking to optimize model selection and improve predictive accuracy in their domains.

Procedia Computer ScienceVol. 258
La Maddalena (IT)
Openalex Percentile: Top 2%
AI and HR Technologies
11.05
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.