arXiv:2411.17937cs.LGcs.AI2024-11被引 9

用河流拓扑结构引导因果学习,提升洪水预测精度与效率

Spatio-temporal Causal Learning for Streamflow Forecasting

  • 基于河流网络构建先验因果图,指导时空图神经网络学习
  • 在德克萨斯州布拉索斯河流域测试中优于普通STGNN模型
  • 结合领域知识与深度学习,适合水文建模与灾害预警研究者

径流在国家水资源可持续规划与管理中至关重要。传统水文模型通过连接降雨、径流等物理过程模拟径流,这些数据具有内在的时空关联与因果关系,可提升预测性能。近年来,时空图神经网络(STGNNs)在交通、气象和疫情等领域表现优异,也展现出在径流管理中的潜力。然而,直接从海量观测数据中学习因果关系在理论上和计算上均具挑战。本研究采用河流流动图作为先验知识,辅助学习因果结构,并利用学习到的因果图预测目标站点的径流。提出的因果径流预测模型(CSF)在德克萨斯州布拉索斯河流域的真实场景中进行了验证。结果表明,该方法优于常规的时空图神经网络,在计算效率上也高于传统模拟方法。通过有效融合河流拓扑图与STGNN,本研究为径流预测提供了新范式,展示了将先进神经网络与领域知识结合在水文建模中提升性能的巨大潜力。

原文摘要 · Abstract (English)

Streamflow plays an essential role in the sustainable planning and management of national water resources. Traditional hydrologic modeling approaches simulate streamflow by establishing connections across multiple physical processes, such as rainfall and runoff. These data, inherently connected both spatially and temporally, possess intrinsic causal relations that can be leveraged for robust and accurate forecasting. Recently, spatio-temporal graph neural networks (STGNNs) have been adopted, excelling in various domains, such as urban traffic management, weather forecasting, and pandemic control, and they also promise advances in streamflow management. However, learning causal relationships directly from vast observational data is theoretically and computationally challenging. In this study, we employ a river flow graph as prior knowledge to facilitate the learning of the causal structure and then use the learned causal graph to predict streamflow at targeted sites. The proposed model, Causal Streamflow Forecasting (CSF) is tested in a real-world study in the Brazos River basin in Texas. Our results demonstrate that our method outperforms regular spatio-temporal graph neural networks and achieves higher computational efficiency compared to traditional simulation methods. By effectively integrating river flow graphs with STGNNs, this research offers a novel approach to streamflow prediction, showcasing the potential of combining advanced neural network techniques with domain-specific knowledge for enhanced performance in hydrologic modeling.

水文预测因果学习图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。