arXiv:2410.14137cs.LG2024-10被引 1

通过因果关联建模土壤与积雪过程,提升洪水预测精度与可解释性。

Hierarchical Conditional Multi-Task Learning for Streamflow Modeling

  • 构建分层多任务框架,利用任务嵌入连接水文模块。
  • 在数百个流域上实现更长时序的精准流速预测。
  • 适合水文建模、灾害预警与智能水利系统研究者。

径流对水资源管理至关重要,受复杂水文系统调控,其过程由气象力驱动的中间环节决定。尽管深度学习模型在径流预测中已达先进水平,但其端到端单任务学习难以捕捉系统内因果关系。为此,我们提出分层条件多任务学习(HCMTL),基于土壤水分与积雪过程与径流的因果联系,联合建模这两类过程。HCMTL利用任务嵌入连接网络模块,增强灵活性与表达能力,并捕捉土壤水和积雪之外的未观测过程。同时引入条件小批量策略,提升长时间序列建模性能。我们在全球数据集上对比五种基线模型,结果显示,HCMTL在数百个流域、长时间跨度下表现更优,证明将领域特定因果知识融入深度学习,可同时提升预测准确率与可解释性,有助于深化对复杂水文系统的理解,并支持高效水资源管理,以减轻干旱与洪涝等自然灾害风险。

原文摘要 · Abstract (English)

Streamflow, vital for water resource management, is governed by complex hydrological systems involving intermediate processes driven by meteorological forces. While deep learning models have achieved state-of-the-art results of streamflow prediction, their end-to-end single-task learning approach often fails to capture the causal relationships within these systems. To address this, we propose Hierarchical Conditional Multi-Task Learning (HCMTL), a hierarchical approach that jointly models soil water and snowpack processes based on their causal connections to streamflow. HCMTL utilizes task embeddings to connect network modules, enhancing flexibility and expressiveness while capturing unobserved processes beyond soil water and snowpack. It also incorporates the Conditional Mini-Batch strategy to improve long time series modeling. We compare HCMTL with five baselines on a global dataset. HCMTL's superior performance across hundreds of drainage basins over extended periods shows that integrating domain-specific causal knowledge into deep learning enhances both prediction accuracy and interpretability. This is essential for advancing our understanding of complex hydrological systems and supporting efficient water resource management to mitigate natural disasters like droughts and floods.

水文建模多任务学习因果推理

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