Dynamic Real-Time Forecasting Technique for Reclaimed Water Volumes in Urban River Environmental Management

In recent years, wastewater recycling has gained popularity as an alternative water source to address urban water shortages. Accurate wastewater quantity prediction is crucial for urban river water resource management. Decomposition ensemble models have shown excellent predictive capabilities but are challenged by boundary effects when decomposing the original data sequence. To address this, a rolling forecast decomposition ensemble scheme was developed. It involves using a machine learning (ML) model for prediction and progressively integrating prediction outcomes into the original sequence using complementary ensemble empirical mode decomposition with adaptive noise (CEEMDAN). Long short-term memory (LSTM) is then applied for sub-signal prediction and ensemble. The ML-CEEMDAN-LSTM model was introduced for wastewater quantity prediction, compared with non-decomposed ML models, CEEMDAN-based LSTM models, and ML-CEEMDAN-based LSTM models. Three ML algorithms—linear regression (LR), gradient boosting regression (GBR), and LSTM—were examined, using real-time prediction data and historical monitoring data, with historical data selected using the decision tree method. The study used daily water volumes data from two reclaimed water plants, CH and WQ, in Beijing. The results indicate that (1) ML models varied in their selection of real-time factors, with LR performing best among ML models during testing; (2) the ML-CEEMDAN-LSTM model consistently outperformed ML models; (3) the ML-CEEMDAN-LSTM hybrid model performed better than the CEEMDAN-LSTM model across different seasons. This study offers a reliable and accurate approach for reclaimed water volumes forecasting, critical for effective water environment management.