arXiv:2506.09193cs.LG2025-06被引 7

LaDCast用潜在扩散模型实现高效中程天气集合预报,精度接近欧洲中心系统。

LaDCast: A Latent Diffusion Model for Medium-Range Ensemble Weather Forecasting

  • 在潜在空间生成小时级集合预报,无需显式扰动
  • 对极端天气如气旋轨迹预测更准,性能媲美欧洲中心系统
  • 降低存储与计算成本,适合实时高分辨率预报

准确的概率性天气预报需要高精度和高效的不确定性量化,这给集合数值天气预报(NWP)和近年机器学习方法带来了巨大负担。我们提出LaDCast,首个全球性的潜在扩散框架用于中程集合预报,完全在学习到的潜在空间中生成小时级集合预报。自编码器将高维ERA5再分析场压缩为紧凑表示,基于Transformer的扩散模型以任意小时为起点生成序列潜在更新。模型引入几何旋转位置嵌入(GeoRoPE)以考虑地球球面几何,采用双流注意力机制实现高效条件化,并使用正弦时间嵌入捕捉季节模式。LaDCast在确定性和概率性预测上均接近欧洲中期天气预报中心IFS-ENS的表现,且无需任何显式扰动。值得注意的是,其在追踪罕见极端事件(如气旋)方面表现更优,能更准确捕捉其路径。通过在潜在空间运行,LaDCast将存储与计算需求降低数个数量级,展示了实现实时千米级分辨率预报的可行路径。代码与模型已开源,训练与评估流程见:https://github.com/tonyzyl/ladcast。

原文摘要 · Abstract (English)

Accurate probabilistic weather forecasting demands both high accuracy and efficient uncertainty quantification, challenges that overburden both ensemble numerical weather prediction (NWP) and recent machine-learning methods. We introduce LaDCast, the first global latent-diffusion framework for medium-range ensemble forecasting, which generates hourly ensemble forecasts entirely in a learned latent space. An autoencoder compresses high-dimensional ERA5 reanalysis fields into a compact representation, and a transformer-based diffusion model produces sequential latent updates with arbitrary hour initialization. The model incorporates Geometric Rotary Position Embedding (GeoRoPE) to account for the Earth's spherical geometry, a dual-stream attention mechanism for efficient conditioning, and sinusoidal temporal embeddings to capture seasonal patterns. LaDCast achieves deterministic and probabilistic skill close to that of the European Centre for Medium-Range Forecast IFS-ENS, without any explicit perturbations. Notably, LaDCast demonstrates superior performance in tracking rare extreme events such as cyclones, capturing their trajectories more accurately than established models. By operating in latent space, LaDCast reduces storage and compute by orders of magnitude, demonstrating a practical path toward forecasting at kilometer-scale resolution in real time. We open-source our code and models and provide the training and evaluation pipelines at: https://github.com/tonyzyl/ladcast.

天气预报扩散模型潜空间集合预测

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