To Feel the Spatial: Graph Neural Network-Based Method for Leakage Risk Assessment in Water Distribution Networks

Abstract

As water distribution networks expand, evaluating pipeline network leakage risk has become increasingly crucial. Contrary to traditional evaluation methods, which are often hampered by subjective weight assignment, data scarcity, and high expenses, data-driven models provide advantages like autonomous weight learning, comprehensive coverage, and cost-efficiency. This study introduces a data-driven framework leveraging graph neural networks to assess leakage risk in water distribution networks. Employing geographic information system (GIS) data from a central Chinese city, encompassing pipeline network details and historical repair records, the model achieved superior performance compared to other data-driven approaches, evidenced by metrics such as precision, accuracy, recall, and the Matthews correlation coefficient. Further analysis of risk factors underscores the importance of factors like pipe age, material, prior failures, and length. This approach demonstrates robust predictive accuracy and offers significant reference value for leakage risk evaluation.