arXiv:2602.16579cs.LGcs.AI2026-02被引 1

用两阶段训练提升全球洪水预报精度,跨数据源表现更稳。

AIFL: A Global Daily Streamflow Forecasting Model Using a Deterministic LSTM Pre-trained on ERA5-Land and Fine-tuned on IFS

  • 先用40年再分析数据预训练,再用实时预报数据微调,应对数据偏差。
  • 独立测试集上中位数KGE'达0.66,优于现有主流模型。
  • 首个在Caravan生态中端到端训练的全球水文模型,适合业务化应用。

可靠的全球日尺度径流预报对防洪和水资源管理至关重要,但数据驱动模型在从历史再分析数据转向业务预报产品时常出现性能下降。本文提出AIFL(Artificial Intelligence for Floods),一种基于确定性LSTM的全球日径流预报模型。该模型在包含18,588个流域的Caravan数据集上训练,采用两阶段迁移学习策略以弥合再分析与预报之间的域偏移。首先在1980–2019年共40年的ERA5-Land再分析数据上预训练,捕捉稳健的水文过程;随后在2016–2019年操作型集成预报系统(IFS)预报数据上微调,适应数值天气预报特有的误差结构与偏差。消融实验表明,该两阶段方法优于仅使用IFS的基线及混合强迫的单阶段方案。据我们所知,这是首个在Caravan生态系统内端到端训练的全球模型。在独立的时间测试集(2021–2024)上,AIFL实现高预测能力,中位数修正版Kling-Gupta效率(KGE')为0.66,中位数纳什-萨特克利夫效率(NSE)为0.53。基准对比显示其精度可媲美当前最先进的全球系统。该模型为全球水文领域提供了一条简洁且业务鲁棒的基准线。

原文摘要 · Abstract (English)

Reliable global streamflow forecasting is essential for flood preparedness and water resource management, yet data-driven models often suffer from a performance gap when transitioning from historical reanalysis to operational forecast products. This paper introduces AIFL (Artificial Intelligence for Floods), a deterministic LSTM-based model designed for global daily streamflow forecasting. Trained on 18,588 basins curated from the Caravan dataset, AIFL utilises a two-stage transfer-learning strategy to bridge the reanalysis-to-forecast domain shift. The model is first pre-trained on 40 years of ERA5-Land reanalysis (1980-2019) to capture robust hydrological processes, then fine-tuned on operational Integrated Forecasting System (IFS) forecasts (2016-2019) to adapt to the specific error structures and biases of operational numerical weather prediction. Ablation experiments confirm that this two-stage approach outperforms both a naive IFS-only baseline and a mixed-forcing single-stage alternative. To our knowledge, this is the first global model trained end-to-end within the Caravan ecosystem. On an independent temporal test set (2021-2024), AIFL achieves high predictive skill with a median modified Kling-Gupta Efficiency (KGE') of 0.66 and a median Nash-Sutcliffe Efficiency (NSE) of 0.53. Benchmarking results show that AIFL achieves comparable accuracy to current state-of-the-art global systems. The model provides a streamlined and operationally robust baseline for the global hydrological community.

水文预报深度学习迁移学习全球模型

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