AI-Based Predictive Maintenance
AcademicDHBW Study Project · Oct 2024 – Jun 2025
Compared Random Forest, Causal Forest, and a PyTorch neural network to predict machine failure on the AI4I 2020 predictive-maintenance dataset.
- Random Forest classifier reached 0.97 ROC-AUC on held-out data.
- SHAP and feature importance pinpointed torque, rotational speed, and tool wear as the dominant failure signals.
- Causal Forest estimated how machine quality tiers causally shift failure risk, beyond simple correlation.