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.

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

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

Artificial Neural Network Modelling of Concentrations of Nitrogen, Phosphorus and Dissolved Oxygen in a Non‐point Source Polluted River in Zhejiang Province, Southeast China

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

Optimal method for predicting the total nitrogen concentration of wastewater under the influence of multiple parameters

Prediction of effluent total nitrogen and energy consumption in wastewater treatment plants: Bayesian optimization machine learning methods

Self-organizing Modeling and Control of Activated Sludge Process Based on Fuzzy Neural Network

Optimized Deep Learning Models for Effluent Prediction in Wastewater Treatment Processes

Development and application of an intelligent nitrogen removal diagnosis and optimization framework for WWTPs: Low-carbon and stable operation

Prediction of biological nutrients removal in full-scale wastewater treatment plants using H 2 O automated machine learning and back propagation artificial neural network model: Optimization and comparison

Improved neural network with least square support vector machine for wastewater treatment process

Prediction of Anoxic Sulfide Biooxidation under Various Hrts Using Artificial Neural Networks

Novel Intelligent Control Framework for WWTP Optimization to Achieve Stable and Sustainable Operation

A novel model-based adaptive control strategy for step-feed SBRs dealing with influent fluctuation.

Real-time prediction of the chemical oxygen demand component parameters in activated sludge model using backpropagation neural network

Simulating and predicting the performance of a horizontal subsurface flow constructed wetland using a fully connected neural network

Prediction and optimization of wastewater treatment process effluent chemical oxygen demand and energy consumption based on typical ensemble learning models

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

Application of artificial neural network in process simulation and optimization of chemically enhanced primary treatment system

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

Intelligent Aeration Amount Prediction Control for Wastewater Treatment Process Based on Recurrent Neural Network