Hybrid Explainable AI Framework for Predictive Maintenance of Aeration Systems in Wastewater Treatment Plants

Aeration systems are among the most energy-intensive components of wastewater treatment plants (WWTPs), consuming up to 75% of total electricity while being prone to performance degradation caused by diffuser fouling and pressure losses. Traditional maintenance strategies are largely reactive or preventive, leading to inefficient interventions, higher operational costs, and limited fault anticipation. This study addresses the need for an advanced predictive maintenance framework capable of early detection and differentiation of multiple aeration system faults. Using the Benchmark Simulation Model No. 2 (BSM2), two representative degradation scenarios—acute airflow pressure loss and chronic diffuser fouling—were simulated to generate a labeled dataset. A hybrid machine learning approach was developed, combining Random Forest-based feature selection with Long Short-Term Memory (LSTM) neural networks for temporal, multi-label fault classification. To enhance interpretability and operator trust, SHapley Additive exPlanations (SHAP) were applied to quantify feature contributions and provide transparent model predictions. The results show that the proposed framework achieves over 94% detection accuracy and provides early warnings compared to static threshold-based methods. The integration of explainable AI ensures actionable insights for maintenance planning. This approach supports more energy-efficient, reliable, and sustainable operation of WWTP aeration systems and offers a benchmark methodology for future predictive maintenance research.