Abstract:Efficient management of wastewater treatment plants (WWTPs) necessitates accurate forecasting of influent water quality parameters (WQPs) and flow rate (Q) to reduce energy consumption and mitigate carbon emissions. The time series of WQPs and Q are highly non-linear and influenced by various factors such as temperature (T) and precipitation (Precip). Conventional models often struggle to account for long-term temporal patterns and overlook the complex interactions of parameters within the data, leading to inaccuracies in detecting WQPs and Q. This work introduced the Pre-training enhanced Spatio-Temporal Graph Neural Network (PT-STGNN), a novel methodology for accurately forecasting of influent COD, ammonia nitrogen (NH3-N), total phosphorus (TP), total nitrogen (TN), pH and Q in WWTPs. PT-STGNN utilizes influent data of the WWTP, air quality data and meteorological data from the service area as inputs to enhance prediction accuracy. The model employs unsupervised Transformer blocks for pre-training, with efficient masking strategies to effectively capture long-term historical patterns and contextual information, thereby significantly boosting forecasting accuracy. Furthermore, PT-STGNN integrates a unique graph structure learning mechanism to identify dependencies between parameters, further improving the model's forecasting accuracy and interpretability. Compared with the state-of-the-art models, PT-STGNN demonstrated superior predictive performance, particularly for a longer-term prediction (i.e., 12 h), with MAE, RMSE and MAPE at 12-h prediction horizon of 2.737 ± 0.040, 4.209 ± 0.060 and 13.648 ± 0.151 %, respectively, for the algebraic mean of each parameter. From the results of graph structure learning, it is observed that there are strong dependencies between NH3-N and TN, TP and Q, as well as Precip, etc. This study innovatively applies STGNN, not only offering a novel approach for predicting influent WQPs and Q in WWTPs, but also advances our understanding of the interrelationships among various parameters, significantly enhancing the model's interpretability.

Robust Neural Network Modeling with Small-Worldness for Effluent Total Phosphorus Prediction in Wastewater Treatment Process

Learning a neural network-based soft sensor with double-errors parallel optimization towards effluent variable prediction in wastewater treatment plants

An adaptive task-oriented RBF network for key water quality parameters prediction in wastewater treatment process

Using a Novel Algorithm Based on the Random Vector Functional Link Network and Multi-Verse Optimizer to Forecast Effluent Quality

Effluent Quality Soft Sensor for Wastewater Treatment Plant with Ensemble Sparse Learning-Based Online Next Generation Reservoir Computing

Real-time Water Quality Prediction of Wastewater Treatment Plants Using Advanced Deep Learning Networks

Enhancing Real-Time Prediction of Effluent Water Quality of Wastewater Treatment Plant Based on Improved Feedforward Neural Network Coupled with Optimization Algorithm

Soft-sensing of effluent total phosphorus using adaptive recurrent fuzzy neural network with Gustafson-Kessel clustering

A hybrid deep learning approach to improve real-time effluent quality prediction in wastewater treatment plant

Prediction of Effluent Quality in a Wastewater Treatment Plant by Dynamic Neural Network Modeling

Uncertainty and Reliability Analysis of Neural Networks Based Soft-sensors with Applications in Wastewater Treatment*

Development of a Novel Feedforward Neural Network Model Based on Controllable Parameters for Predicting Effluent Total Nitrogen

Hierarchical Neural Network Modeling Approach to Predict Sludge Volume Index of Wastewater Treatment Process.

A Data-Derived Soft-Sensor Method for Monitoring Effluent Total Phosphorus

Prediction of Wastewater Treatment Plant Effluent Water Quality Using Recurrent Neural Network (RNN) Models

Pre-training enhanced spatio-temporal graph neural network for predicting influent water quality and flow rate of wastewater treatment plant: Improvement of forecast accuracy and analysis of related factors

Neural Networks for Approximating Numerical Water Quality Models: Applicability and Deceptive Effects of Insensitive Parameters

Effluent Quality Prediction of Wastewater Treatment Plant Based on Wavelet Neural Networks

A Fuzzy Neural Network-Based Soft Sensor for Modeling Nutrient Removal Mechanism in a Full-Scale Wastewater Treatment System

Multiple Neural Network-Based Model to Predict Ammonia Nitrogen Content in Wastewater

Optimized Deep Learning Models for Effluent Prediction in Wastewater Treatment Processes