用新架构实现高效精准的区域气象概率预报
CRPS-LAM: Probabilistic Regional Weather Forecasting with Continuous Ranked Probability Score
- 设计混合CNN/GNN结构,提升区域气象预测效率与精度
- 基于CRPS损失训练,单次前向传播即可生成概率预报
- 速度比扩散模型快约39倍,适合实际业务部署
有限区域模型(LAM)可在区域尺度上以高于全球模型的分辨率进行气象预报。在高分辨率下,机器学习天气预测越来越多依赖集合方法生成概率预报。然而,现有机器学习LAM因依赖计算成本高的扩散模型或低效图神经网络,难以扩展。本文提出一种针对LAM气象预报问题定制的混合CNN/GNN架构,构建了DET-LAM确定性模型,其效率和精度均优于基于图的同类模型。进一步在此架构基础上构建生成模型CRPS-LAM,采用连续排名概率评分(CRPS)作为目标函数,实现单次前向传播下的高效训练与采样。相比扩散基线,推理速度提升约×39。在北欧区域的评估中,CRPS-LAM在多种大气变量上均生成了有技巧且校准良好的预报。
原文摘要 · Abstract (English)
Limited-Area Models (LAMs) enable weather forecasting over regional domains at higher resolutions than what is computationally feasible for global models. At such high resolutions, machine learning approaches for weather prediction increasingly rely on ensemble methods to produce probabilistic forecasts. However, existing machine learning LAMs are not scalable due to relying on computationally costly diffusion models or inefficient graph neural networks. We tackle this by introducing a new hybrid CNN/GNN architecture, tailored to the LAM weather forecasting problem. Using this architecture, we construct the DET-LAM deterministic model, producing LAM forecasts both more efficiently and accurately than its graph-based competitor. We then tackle the ensemble forecasting problem, by using this architecture as a backbone for the generative model CRPS-LAM. CRPS-LAM is trained using a Continuous Ranked Probability Score (CRPS) objective, enabling efficient training and sampling in a single forward pass. This yields a speedup of $\approx \times 39$ compared to diffusion-based baselines. We evaluate our approach on regional domains in northern Europe, demonstrating that CRPS-LAM produces skillful and well-calibrated forecasts across a range of atmospheric variables.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。