Ensemble Learning for Predictive Maintenance: A Comparative Analysis of Bagging, Boosting, and Stacking Strategies
Predictive maintenance (PdM) has become a crucial element of Industry 4.0, aiming to reduce equipment downtime and operational costs through data-driven failure prediction. Traditional machine learning models, while effective, face challenges such as imbalanced class distributions, sensor noise, and complex failure patterns, which limit their reliability. This study investigates the potential of ensemble learning techniques—XGBoost, Random Forest, and Logistic Regression—to enhance predictive accuracy and robustness in industrial PdM systems. Using the Kaggle Predictive Maintenance Dataset, we implement and compare three ensemble strategies: (1) Random Forest (bagging), (2) XGBoost (boosting), and (3) a stacked ensemble combining XGBoost, Random Forest, and Logistic Regression. Our experiments reveal that stacking achieves the highest performance (F1-score: 0.92), particularly in detecting rare failure classes. Additionally, we evaluate computational efficiency, interpretability trade-offs, and the impact of hyperparameter tuning on model generalizability. The results demonstrate that ensemble methods, especially hybrid approaches, significantly outperform traditional single-model frameworks, providing actionable insights for deploying scalable PdM solutions in industrial environments.
Authors
- Sonam Khattar (ORCID: https://orcid.org/0000-0002-5444-4358)
- Tushar Verma (ORCID: https://orcid.org/0000-0002-6696-4537)
- Sandip Bhattacharyya
- KrishanKant
Institutions
- Chandigarh University (IN)
Publication Details
- Published
- 2025-05-15
- DOI
- https://doi.org/10.1109/ic2e365635.2025.11167068
- Primary Topic
- Infrastructure Maintenance and Monitoring
- Type
- article
- Field-Weighted Citation Impact
- 0.00