Abstract:Ensuring the safety and reliability of drinking water supply requires accurate prediction of water quality in water distribution networks (WDNs). However, existing hydraulic model-based approaches for system state prediction face challenges in model calibration with limited sensor data and intensive computing requirements, while current machine learning models are lack of capacity to predict the system states at sites that are not monitored or included in model training. To address these gaps, this study proposes a novel gated graph neural network (GGNN) model for real-time water quality prediction in WDNs. The GGNN model integrates hydraulic flow directions and water quality data to represent the topology and system dynamics, and employs a masking operation for training to enhance prediction accuracy. Evaluation results from a real-world WDN demonstrate that the GGNN model is capable to achieve accurate water quality prediction across the entire WDN. Despite being trained with water quality data from a limited number of sensor sites, the model can achieve high predictive accuracies (Mean Absolute Error = 0.07 mg L −1 and Mean Absolute Percentage Error = 10.0 %) across the entire network including those unmonitored sites. Furthermore, water quality-based sensor placement significantly improves predictive accuracy, emphasizing the importance of careful sensor location selection. This research advances water quality prediction in WDNs by offering a practical and effective machine learning solution to address challenges related to limited sensor data and network complexity. This study provides a first step towards developing machine learning models to replace hydraulic models in WDN modelling.

Graph Convolutional Recurrent Neural Networks for Water Demand Forecasting

Study of Artificial Neural Network Model for Forecasting Urban Water Demand

Hourly and Daily Urban Water Demand Predictions Using a Long Short-Term Memory Based Model

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

TransGlow: Attention-augmented Transduction model based on Graph Neural Networks for Water Flow Forecasting

An Ensemble Neural Network Model to Forecast Drinking Water Consumption

Medium-term water consumption forecasting based on deep neural networks

Water consumption time series forecasting in urban centers using deep neural networks

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

Water Supply Prediction Based on Initialized Attention Residual Network

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

A Hybrid Framework for Multivariate Time Series Forecasting of Daily Urban Water Demand Using Attention-Based Convolutional Neural Network and Long Short-Term Memory Network

Spatial-temporal graph neural networks for groundwater data

Short-term residential load forecasting using Graph Convolutional Recurrent Neural Networks

Wavelet-Inspired Multiscale Graph Convolutional Recurrent Network for Traffic Forecasting

Constructing Geographic and Long-term Temporal Graph for Traffic Forecasting

Spatial dynamic graph convolutional network for traffic flow forecasting

Study on water quantity prediction method based on bidirectional long and short time neural network

Short-Term Water Demand Forecast Based on Deep Learning Method

Attention-based Spatial-Temporal Graph Convolutional Recurrent Networks for Traffic Forecasting

Reliable multi-horizon water demand forecasting model: A temporal deep learning approach