arXiv:2508.04888cs.LG2025-08KDD被引 2

用历史水文数据增强模型,提升大沼泽地水位预测精度。

Retrieval-Augmented Foundation Models for Water Level Prediction in the Everglades

  • 从历史数据中检索相似水文事件,补充模型输入上下文。
  • 长时序预测效果显著提升,极端天气下增益更明显。
  • 方法可推广至其他流域,适合环境科学与水利决策者。

准确预测大沼泽地水位对防洪、抗旱、水资源规划和生物多样性保护至关重要。尽管近期时间序列基础模型在通用任务上表现优异,但其在特定领域的应用效果仍不明确。本文构建了针对大沼泽地水位预测的专用数据集,发现现有先进模型性能仍有限。为此,我们引入基于检索的增强机制,从外部历史观测档案中检索相似的多变量水文事件,丰富预训练模型的输入上下文。研究了基于统计相似性与互信息的两种检索策略,分析其对预测性能的影响。大量实验表明,检索增强能持续提升长期水位预测能力,在极端事件中带来更大收益,对环境决策尤为关键。本研究为基于类比的检索提升预训练时间序列模型在环境科学中的应用提供了实证支持,揭示了其在大沼泽地水文预测中的优势、局限与失效模式。尽管聚焦大沼泽地,该框架具备通用性,只要有时间序列数据即可应用于其他水文系统。代码与数据已公开于 https://github.com/rahuul2992000/WaterRAF。

原文摘要 · Abstract (English)

Accurate water level forecasting in the Everglades is essential for flood mitigation, drought management, water resource planning, and biodiversity conservation. While recent time-series foundation models have shown strong performance on generic tasks (represented in their pre-training), their effectiveness in domain-specific applications remains insufficiently understood. In this work, we curate a domain-specific dataset for water-level forecasting in the Everglades and observe that the performance of current state-of-the-art models remains limited. To address this gap, we leverage a retrieval-augmented mechanism that retrieves analogous multivariate hydrological episodes from an external archive of historical observations to enrich the input context of those pre-trained models. We study two retrieval strategies, statistical similarity-based retrieval and mutual information-based retrieval, and analyze how incorporating retrieved historical contexts affects predictive performance. Extensive experiments show that retrieval augmentation consistently improves long-horizon water level forecasts and yields disproportionately larger gains during extreme events, which is particularly critical for environmental decision-making. Our study provides empirical evidence that analog-based retrieval can benefit pretrained time-series foundation models in environmental science, offering practical insights into their strengths, limitations, and failure modes when applied to hydrological forecasting in the Everglades. Although evaluated in the Everglades, the proposed framework is general and can be applied to other hydrological systems given time series data. The code and data have been made publicly available at https://github.com/rahuul2992000/WaterRAF.

水位预测检索增强时间序列环境建模

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