Pressure prediction of water supply network based on 1DCNN-GRU-multi-head attention

Abstract

ABSTRACT The water supply network is a crucial component of urban construction and plays a pivotal role in the normal operation of the city. The complexity and periodicity of pressure variations in the water supply networks pose significant challenges to traditional prediction models. In this study, we introduce a novel short-term pressure prediction model, termed the 1DCNN-GRU-multi-head attention (CGMA) model, which incorporates a multi-head attention mechanism alongside an integrated network consisting of one-dimensional convolutional neural networks (1DCNNs) and gated recurrent units (GRUs). Initially, the model employs a 1DCNN network to extract features from the pressure data. Subsequently, the extracted features are input into the GRU neural network, leveraging its long-term dependency capabilities to improve prediction accuracy. Then, the attention mechanism is incorporated into the GRU component to highlight key information, enabling the model to focus on more important data features, thereby improving the prediction performance. We implemented this model in a real urban water distribution system equipped with five pressure sensors. The model achieved a mean absolute error of 0.00197 and a root mean square error of 0.00262. When compared with alternative approaches, our method demonstrated superior predictive performance for pressure data, thereby confirming its efficacy in practical applications.