arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.02797cs.LG
收藏

用于配水管网泄漏检测与定位的算法启发图神经网络

Algorithm-Informed Graph Neural Networks for Leakage Detection and Localization in Water Distribution Networks

EPFL(洛桑联邦理工学院)

Zepeng Zhang, Olga Fink

AI总结:

针对配水管网泄漏检测中GNN分布外泛化不足的问题,本文提出AIGNN,通过预训练模拟Ford-Fulkerson最大流算法并迁移知识,结合压力重建与预测实现更优的泄漏检测定位。

摘要:

泄漏检测与定位是配水管网(WDN)高效、可持续管理面临的一项重大挑战。近期方法利用WDN固有的图结构,采用基于图的数据驱动方法。然而,这些方法往往会学到捷径,在分布内数据上表现良好,却无法泛化到分布外数据。为解决这一局限,并受经典算法完美泛化能力的启发,我们提出了一种算法启发图神经网络(AIGNN)。由于认识到WDN作为流网络发挥作用,纳入最大流信息有助于推断压力。在所提出的框架中,我们首先训练AIGNN模拟用于求解最大流问题的Ford-Fulkerson算法。随后,这种算法知识被迁移以解决WDN中的压力估计问题。框架部署了两个AIGNN:一个基于当前测量值重建压力,另一个基于先前测量值预测压力。通过比较重建器和预测器的输出,实现泄漏的检测与定位。通过预训练AIGNN使其像算法一样推理,预期它们能够提取更多与任务相关且可泛化的特征。实验结果表明,与未纳入算法知识的GNN相比,所提出的算法启发方法取得了更优结果,并具有更好的泛化能力。

英文摘要:

Detecting and localizing leakages is a significant challenge for the efficient and sustainable management of water distribution networks (WDN). Leveraging the inherent graph structure of WDNs, recent approaches have used graph-based data-driven methods. However, these methods often learn shortcuts that work well with in-distribution data but fail to generalize to out-of-distribution data. To address this limitation and inspired by the perfect generalization ability of classical algorithms, we propose an algorithm-informed graph neural network (AIGNN). Recognizing that WDNs function as flow networks, incorporating max-flow information can be beneficial for inferring pressures. In the proposed framework, we first train AIGNN to emulate the Ford-Fulkerson algorithm for solving max-flow problems. This algorithmic knowledge is then transferred to address the pressure estimation problem in WDNs. Two AIGNNs are deployed, one to reconstruct pressure based on the current measurements, and another to predict pressure based on previous measurements. Leakages are detected and localized by comparing the outputs of the reconstructor and the predictor. By pretraining AIGNNs to reason like algorithms, they are expected to extract more task-relevant and generalizable features. Experimental results demonstrate that the proposed algorithm-informed approach achieves superior results with better generalization ability compared to GNNs that do not incorporate algorithmic knowledge.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑