观潮 · 模式预报与气象 AI 观察

SDECast 天气 AI 用连续时间随机微分方程代替自回归:一小时分辨率预报最长报到五天

2026-10-06

现有的机器学习天气预报模型基本都是固定时间分辨率上的自回归推进——学一个单步预测器,然后一步步滚起来。这种做法长时效预报很省算力,但用在更短的时间步上会严重累积误差,而且它没有显式编码大气动力学的局地性和时间连续性这两条微分方程天然带的性质。

这篇论文的做法是把 SDE Matching 改造成一个直接在物理量空间里学随机动力学的框架,训练时不需要反复积分随机微分方程。它管这叫 simulation-free(免仿真)。

---

为什么不是自回归

论文给的替代路线有两条,各有各的问题。神经常微分方程被探索过用于天气建模,但要么是确定性的,要么因为基于仿真的损失函数而训练昂贵。神经随机微分方程理论上更合适——通过漂移项参数化动力学、再用扩散项引入随机性,从而支持有原则的概率预报和高效集合生成——但传统上只能从充分细分的轨迹学,或者用需要训练中反复仿真的目标,这限制了它在高维系统上的应用。

SDE Matching 提供了一个免仿真的目标来从离散观测学隐变量 SDE。SDECast 的技术贡献是把它改造成直接在物理空间里学,并引入一个辅助的隐过程来构造训练目标,但要求这个隐过程在观测时刻与物理状态重合。这样学到的先验 SDE 就可以直接当成物理动力学的模型来解释。

它宣称的三项能力是:

数学上,后验 SDE 与先验共享同一个扩散系数,但漂移以观测轨迹为条件。论文给出两种理解:它是先验 SDE 加上观测似然后真实轨迹后验的变分近似;也可以看作先验 SDE 的受控版本,改动其动力学使之与观测轨迹一致。因为共享扩散系数,这个控制是隐式定义的。变分目标同时惩罚控制项的范数并重建观测。免仿真这一步靠把后验边缘直接参数化来实现——在相应假设下,可以得到后验漂移的解析表达式,于是损失只需采样并求值,不必积分后验 SDE。

SQG 系统上的连续时间预报表现
SQG 系统上的连续时间预报表现

图注:三联图,左为 RMSE、中为 CRPS、右为 spread-skill ratio(SSR),横轴为预报时效(天)。图例包含 Ours、ARCI-24-1h、AR-1h、DYfusion、ClimODE,以及 SSR = 1 的黑色虚线基准。

---

SQG 上:训练分辨率六小时,报一小时

SQG(地表准地转)是一个被广泛使用的湍流地学流模型。它相对简单,但抓住了地学湍流的关键性质——对初值敏感、跨多尺度相互作用,同时计算上仍然可处理。因为数据由已知的数值模拟器生成,这个设定允许评估的不只是预报质量,还有 SDECast 能否恢复出底层连续时间动力学。

实验设计里有一个关键设置:所有模型都用间隔六小时的快照训练,但在一小时分辨率上评估预报。基线是自回归的条件流匹配模型(CFM)。

三个结果:

学习到的漂移与真值漂移、六小时有限差分的对比
学习到的漂移与真值漂移、六小时有限差分的对比

图注:四联图,依次为 SQG 状态场、真值漂移、学习漂移、线性漂移。色标范围约 −0.5 到 0.5。学习漂移与真值漂移给出的空间结构一致,均呈涡旋状分布;相对参考的 RMSE 标为 0.101,而线性漂移为 0.263。

因为有数值模拟器在手,研究者能把学到的漂移和真值漂移直接对比,并同时给出一个六小时的有限差分估计。结果是:学习到的漂移与真值漂移高度相似,且比有限差分估计更好地描述了瞬时动力学——这说明 SDECast 能从粗采样观测中恢复连续时间行为。

但这里有一个诚实且重要的限定:学习到的漂移在最高空间频率上偏弱。 论文给出的理由是,在粗时间采样下最快的动力学只被弱分辨,SDECast 可能把部分高频变率归给随机性而非确定性漂移。由此推出一个反直觉但重要的结论:尽管底层 SQG 模拟器是确定性的,学到的 SDE 提供了对其未分辨混沌动力学的有效随机表示。相应地,在推理时去掉扩散项(论文称之为 SDECast deterministic)会显著降低预报性能,并给出过度平滑的、接近均值的轨迹。

ERA5 全球天气预报在 850 hPa 的技巧指标
ERA5 全球天气预报在 850 hPa 的技巧指标

图注:左图为 850 hPa 位温的单点预报示例,横轴为 00:00 UTC 起的 24 小时。橙线为集合平均,浅橙/深橙带为 ±1σ 与 ±2σ,黑线为 5 条样本轨迹,蓝叉为地面真值。

---

ERA5 上:全球尺度,小时分辨率

ERA5 上的集合预报与真变对照
ERA5 上的集合预报与真变对照

图注:左列七幅为真变(GT_t - GT_0),右列七幅为预报(PR_t - GT_0),自上而下时效递增,每行色标上限从 1.5 增大到 4.0。两列的空间型态逐行对应,说明预报保留了真变的大尺度结构。

论文的表述是:SDECast 扩展到全球天气预报的逐小时分辨率,能给出最长五天时效的有技巧概率预报。图像显示逐时变量的空间结构在多个时效上与真变对应,误差随时效增大而放大,这与图示的色标上限逐行增大一致。

逐时 U10 风场的真变与预报残差
逐时 U10 风场的真变与预报残差

图注:左列为 U10 的真变变化,右列为预报减真值的残差,自上而下时效递增。预报端残差色标范围与真变端一致,残差随时效从接近 ±2 增大到 ±4。

Z500 高度场的真变与预报残差
Z500 高度场的真变与预报残差

图注:与 U10 那张同构,改为 500 hPa 位势高度。格式为真变变化(左)与预报减真值(右)对照,共七行时效,色标上限由 0.10 增至 1.0。

补充一组 SQG 上的量:论文附了 RMSE、CRPS、SSR 三指标随预报时效的对比,包含 Persistence(持续性)、CFM、SDECast(线性)、SDECast(集合)、SDECast(确定性)几条线。其中 CFM 在 RMSE/CRPS 上明显最低,SDECast 确定性在 SSR 上退化最严重——这与正文"去掉扩散会过度平滑"的判断方向一致,但论文正文没有给出这组 SQG 数字的具体数值,只有图。

---

作者承认的限制

后验与扩散系数的参数化是简化的。 论文明确列在结论里:当前的限制包括简化的后验与扩散系数参数化,以及有限的实验范围。

漂移在最高空间频率上偏弱,这一点没有解决。 论文的解释是时间上粗采样导致最快动力学只被弱分辨,并推测 SDECast 把这部分变率归给了随机性。这是一个诚实的归因,但也意味着"确定性 vs 随机性"的划分在这个设定下并不是物理上唯一的。

免仿真训练依赖两个假设。 论文写明:需要相应量对参数可微、对状态变量可逆,才能得到后验漂移的解析表达式。这两条假设在数值上失效时的行为论文没有讨论。

全球实验只有一组数据、一个分辨率。 ERA5 是唯一的数据源,变量集合与分辨率都只有一套。论文没有给出多数据集、多分辨率的验证。

预报时效的上限是五天,且论文没有解释为什么停在这里。 正文只说"最长五天",没有给出这是算力限制、训练窗口长度限制还是技巧已经衰减的判断。

"扩展到全球"与"业务可用"是两件事。 论文宣称的是可扩展性和可技巧性,没有涉及运行成本、延迟或稳定性。

---

一句话判断

这篇论文最值得带走的是"训练分辨率不等于预报分辨率"这件事:模型用六小时间隔训练,却能在一小时分辨率上生成集合,而自回归基线被死死绑在训练分辨率上。 同时要接受它自己的判断——学到的漂移在最高频上偏弱,是因为那些变率被归进了扩散项。落到预报实践上,连续时间 SDE 的价值在于你可以自由选输出步长并拿到校准良好的集合,代价是必须接受高频确定性信息有偏。

在观潮中打开(含配图与交互)→