一跳式模型实现高效天气概率预测,精度媲美多步生成方法。
Tyche: One Step Flow for Efficient Probabilistic Weather Forecasting

- 用单次函数计算直接映射噪声到未来天气状态,无需迭代去噪。
- 仅用1次求解步骤,在1.5°分辨率下达到与先进方法相当的预报精度和校准度。
- 适合需要快速生成大量可靠天气预测的气象业务场景。
概率天气预报不仅需要准确轨迹,还需对可能的大气未来状态给出校准良好的分布。近期数据驱动系统在确定性预报上表现优异,基于扩散的集合预报器显著提升了样本真实性和不确定性量化能力。但其推理成本随预报时长、集合规模和每步去噪次数增长,导致大规模运行代价高昂。为此,我们提出Tyche,一种用于高效概率天气预报的一步条件流模型。Tyche通过目标感知的平均速度流,将高斯噪声直接映射至未来天气状态,仅需一次函数评估(1-NFE)。为使该一步传输在高维地球物理场中可学习,我们设计了基于JVP正则化的修正目标,无需显式构造雅可比矩阵即可保证源与目标时间步间的时序自洽性。传输场由各向同性的Swin风格变压器参数化,既保留细粒度空间结构,又可在全球网格上高效扩展。为进一步提升自回归预报下的集合可靠性,我们引入基于滚动的微调阶段,并采用课程学习的CRPS校准监督。在1.5°、6小时分辨率的ERA5数据上实验表明,仅使用1次求解步骤的Tyche,在预报技能和校准度上均达到或超越现有先进多步生成基线及欧洲中期天气预报中心(ECMWF IFS)的运行集合。
原文摘要 · Abstract (English)
Probabilistic weather forecasting requires not only accurate trajectories, but calibrated distributions over plausible atmospheric futures. Recent data-driven systems have achieved remarkable deterministic skill, and diffusion-based ensemble forecasters have substantially improved sample realism and uncertainty quantification. However, their inference cost scales with forecast horizon, ensemble size, and the number of denoising steps required for each transition, making large operational ensembles expensive. To address this, we present Tyche, a one-step conditional flow model for efficient probabilistic weather forecasting. Tyche models the conditional forecast distribution with a destination-aware average-velocity flow that maps Gaussian noise directly to future weather states in a single function evaluation (1-NFE). To make this one-step transport learnable in high-dimensional geophysical fields, we derive a JVP-regularized rectification objective that enforces temporal self-consistency across source and destination flow timesteps without explicitly forming Jacobians. The transport field is parameterized by an isotropic Swin-style transformer that preserves fine-scale spatial structure while remaining scalable on global grids. To improve ensemble reliability under autoregressive forecasting, we further introduce a rollout-based finetuning stage with curriculum CRPS calibration supervision. Experiments on ERA5 at 1.5$^\circ$ and 6-hour resolution show that our Tyche, using merely a single NFE, matches or exceeds the forecast skill and calibration of state-of-the-art multi-step generative baselines and the operational ECMWF IFS ensemble.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。