提出可解释的模糊GNN,精准定位供水管网泄漏点并提供规则化解释。
Explainable Fuzzy GNNs for Leak Detection in Water Distribution Networks
- 融合互信息与模糊逻辑,实现节点分类的可解释性
- 检测F1达0.889,定位F1达0.814,接近传统GNN性能
- 适合需透明决策的水务工程师和运维团队使用
及时发现供水管网泄漏对节约资源和保障运行效率至关重要。尽管图神经网络(GNN)擅长捕捉传感器数据中的时空依赖关系,但其黑箱特性以及针对供水网络的可解释图模型研究有限,限制了实际应用。本文提出一种可解释GNN框架,通过互信息识别关键网络区域,并结合模糊逻辑为节点分类任务提供清晰的规则化解释。在多个GNN架构中,选择表现最优的广义图卷积网络(GENConv)并构建其模糊增强版本,以实现对泄漏位置的直观解释。所提出的模糊图神经网络(FGENConv)在检测任务上取得0.889的图F1分数,在定位任务上为0.814,略低于非模糊GENConv的0.938和0.858,但提供了空间局部化的模糊规则解释。该方法在精度与可解释性间取得平衡,使水利工程师能验证预测结果,节省人力,优化维护策略。代码已公开于github.com/pasqualedem/GNNLeakDetection。
原文摘要 · Abstract (English)
Timely leak detection in water distribution networks is critical for conserving resources and maintaining operational efficiency. Although Graph Neural Networks (GNNs) excel at capturing spatial-temporal dependencies in sensor data, their black-box nature and the limited work on graph-based explainable models for water networks hinder practical adoption. We propose an explainable GNN framework that integrates mutual information to identify critical network regions and fuzzy logic to provide clear, rule-based explanations for node classification tasks. After benchmarking several GNN architectures, we selected the generalized graph convolution network (GENConv) for its superior performance and developed a fuzzy-enhanced variant that offers intuitive explanations for classified leak locations. Our fuzzy graph neural network (FGENConv) achieved Graph F1 scores of 0.889 for detection and 0.814 for localization, slightly below the crisp GENConv 0.938 and 0.858, respectively. Yet it compensates by providing spatially localized, fuzzy rule-based explanations. By striking the right balance between precision and explainability, the proposed fuzzy network could enable hydraulic engineers to validate predicted leak locations, conserve human resources, and optimize maintenance strategies. The code is available at github.com/pasqualedem/GNNLeakDetection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。