arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.14980cs.LGcs.AIphysics.ao-ph
收藏

用于降雨-径流模拟的深度状态空间模型

A Deep State Space Model for Rainfall-Runoff Simulations

University of Oklahoma(俄克拉荷马大学);Cornell University(康奈尔大学);Lawrence Berkeley National Laboratory(劳伦斯伯克利国家实验室)

Yihan Wang, Lujun Zhang, Annan Yu, N. Benjamin Erichson, Tiantian Yang

AI总结:

提出基于状态空间模型的S4D-FT用于降雨-径流模拟,在531个流域中超越LSTM等基准,挑战了LSTM在水文界的主导地位。

摘要:

研究水循环中降雨-径流过程的传统方法依赖于概念性或基于物理的水文模型。深度学习(DL)近期作为一种替代方案出现,并在水文界的降雨-径流模拟中蓬勃发展。然而,数十年历史的Long Short-Term Memory(LSTM)网络仍然是该任务的基准,表现优于Transformers等较新架构。在本研究中,我们提出了一种State Space Model(SSM),具体为Frequency Tuned Diagonal State Space Sequence(S4D-FT)模型,用于降雨-径流模拟。所提出的S4D-FT在美国本土(CONUS)的531个流域中,与成熟的LSTM以及基于物理的Sacramento Soil Moisture Accounting模型进行了基准对比。结果表明,S4D-FT能够在不同地区超越LSTM模型。我们开创性地将S4D-FT引入降雨-径流模拟,挑战了LSTM在水文界的主导地位,并扩展了可用于水文建模的DL工具库。

英文摘要:

The classical way of studying the rainfall-runoff processes in the water cycle relies on conceptual or physically-based hydrologic models. Deep learning (DL) has recently emerged as an alternative and blossomed in hydrology community for rainfall-runoff simulations. However, the decades-old Long Short-Term Memory (LSTM) network remains the benchmark for this task, outperforming newer architectures like Transformers. In this work, we propose a State Space Model (SSM), specifically the Frequency Tuned Diagonal State Space Sequence (S4D-FT) model, for rainfall-runoff simulations. The proposed S4D-FT is benchmarked against the established LSTM and a physically-based Sacramento Soil Moisture Accounting model across 531 watersheds in the contiguous United States (CONUS). Results show that S4D-FT is able to outperform the LSTM model across diverse regions. Our pioneering introduction of the S4D-FT for rainfall-runoff simulations challenges the dominance of LSTM in the hydrology community and expands the arsenal of DL tools available for hydrological modeling.

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

↑