用状态空间模型提升洪水预报精度,实现多日流量轨迹的连贯预测。
HydroDiffusion: Diffusion-Based Probabilistic Streamflow Forecasting with a State Space Backbone
- 采用仅解码器的状态空间模型替代LSTM,一次去噪完成多日流量预测。
- 在531个流域测试中,全时程预测误差更小,且比DRUM模型更稳定。
- 适合需要长期、高精度水文预报的水利管理与灾害预警场景。
近期研究将扩散模型引入概率性径流预报,展现出良好的早期洪水预警能力。然而,现有方法依赖循环LSTM骨干网络和单步训练目标,难以捕捉长程依赖关系,且预测轨迹易累积误差。为此,我们提出HydroDiffusion,一种基于扩散模型、采用仅解码器状态空间模型骨干的预报框架。该框架通过单次前向过程联合去噪完整多日流量轨迹,保障时间一致性并减少自回归预测中的误差累积。我们在美国本土531个流域(来自CAMELS数据集)进行了评估,对比了两个带有LSTM骨干的扩散基线模型及近期提出的扩散径流模型(DRUM)。结果表明,HydroDiffusion在观测气象强迫驱动下具备优异的即时预报精度,并在整个模拟时域内保持一致性能;此外,在实际预报任务中,其确定性和概率性预报技能均优于DRUM。这些结果确立了HydroDiffusion作为中短期径流预报的稳健生成建模框架,为大陆尺度概率水文预测提供新基准与研究基础。
原文摘要 · Abstract (English)
Recent advances have introduced diffusion models for probabilistic streamflow forecasting, demonstrating strong early flood-warning skill. However, current implementations rely on recurrent Long Short-Term Memory (LSTM) backbones and single-step training objectives, which limit their ability to capture long-range dependencies and produce coherent forecast trajectories across lead times. To address these limitations, we developed HydroDiffusion, a diffusion-based probabilistic forecasting framework with a decoder-only state space model backbone. The proposed framework jointly denoises full multi-day trajectories in a single pass, ensuring temporal coherence and mitigating error accumulation common in autoregressive prediction. HydroDiffusion is evaluated across 531 watersheds in the contiguous United States (CONUS) in the CAMELS dataset. We benchmark HydroDiffusion against two diffusion baselines with LSTM backbones, as well as the recently proposed Diffusion-based Runoff Model (DRUM). Results show that HydroDiffusion achieves strong nowcast accuracy when driven by observed meteorological forcings, and maintains consistent performance across the full simulation horizon. Moreover, HydroDiffusion delivers stronger deterministic and probabilistic forecast skill than DRUM in operational forecasting. These results establish HydroDiffusion as a robust generative modeling framework for medium-range streamflow forecasting, providing both a new modeling benchmark and a foundation for future research on probabilistic hydrologic prediction at continental scales.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。