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.

Graph Convolutional Neural Network for Pressure Prediction in Water Distribution Network Sites

Daily water supply prediction method based on integrated learning and deep learning

Graph Neural Networks for Pressure Estimation in Water Distribution Systems

A Real-Time Method to Detect the Leakage Location in Urban Water Distribution Networks

Real-time water quality prediction in water distribution networks using graph neural networks with sparse monitoring data

W-WaveNet: A multi-site water quality prediction model incorporating adaptive graph convolution and CNN-LSTM

Hybrid Attention Networks for Flow and Pressure Forecasting in Water Distribution Systems

Anomaly Detection and Inlet Pressure Prediction in Water Distribution Systems Using Machine Learning

Pressure Prediction of a Water Distribution Network Based on SVM

Combined Wavelet Transform With Long Short-Term Memory Neural Network for Water Table Depth Prediction in Baoding City, North China Plain

Graph Laplace Regularization-based pressure sensor placement strategy for leak localization in the water distribution networks under joint hydraulic and topological feature spaces

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

Accurate prediction of water quality in urban drainage network with integrated EMD-LSTM model

A Convenient and Stable Graph-Based Pressure Estimation Methodology for Water Distribution Networks: Development and Field Validation.

The CEEMDAN-ps-gcGRU Model on Water Pressure Prediction with Strong Irregularity

A deep learning model with spatio-temporal graph convolutional networks for river water quality prediction

Pressure Sensor Placement in Water Supply Network Based on Graph Neural Network Clustering Method

A Transformer-Based Approach to Leakage Detection in Water Distribution Networks

Bridging Hydraulics and Graph Signal Processing: A New Perspective to Estimate Water Distribution Network Pressures.

Physics-Informed Graph Neural Networks for Water Distribution Systems

A weighting strategy to improve water demand forecasting performance based on spatial correlation between multiple sensors