Making waves: Rethinking machine learning in wastewater effluent quality prediction through the overlooked roles of autocorrelation and baseline models.
Journal:
Water research
Published Date:
Feb 12, 2026
Abstract
Time-series machine learning (ML) approaches have been increasingly used to predict effluent quality in wastewater treatment plants (WWTPs), with a principal focus being on accuracy. However, as wastewater effluent quality is by nature autocorrelated, the interpretation of reported good ML performance could be overestimated by prediction target autocorrelation. Since standard performance metrics (e.g., R2) can yield high values even for simple persistence models, a re-evaluation using proper baselines and metrics is essential for model interpretations. This study addresses these gaps by highlighting the role of target autocorrelation and introducing the persistence model as a benchmark method. Autoregressive methods are evaluated using data from three published studies and two additional WWTP datasets. The aggregated SHAP analyses demonstrate that the importance of the historical target exceeds the second most influential parameter by 64 %-396 %, suggesting that a substantial portion of the reported "high accuracy" in short-term effluent quality prediction can be attributed to strong target autocorrelation rather than the effective learning of complex input-output relationships. Moreover, the persistence model frequently outperforms ML models in COD, TN, and TP predictions regardless of prediction horizon, especially in low-volatility scenarios. A novel index, PN-MAROC, is proposed to quantify data volatility and shows a strong correlation with performance of both persistence models (R2 = 0.93) and ML models (R2 = 0.75). This research highlights the need to consider autocorrelation and appropriate baselines in WWTP effluent quality predictions, and seeks to provide practical guidance for model interpretation and assessment.
Authors
Keywords
No keywords available for this article.