AI 模拟天气与海洋的新方法:用球面坐标变换减少预测误差 | AGIDaily
AGI Daily
资讯日报词条论文人物新创

AI 模拟天气与海洋的新方法:用球面坐标变换减少预测误差

2026-08-31 / 共 2,632 字

8 月 27 日,arXiv 上线预印本《Dandelion: A Spherical Flower for Neural Simulation of Planetary Dynamics》。研究者 Till Muser、Giovanni Abati 与 Ivan Dokmanić 提出一种直接在球面上求解偏微分方程(PDE)的神经网络方法。它通过预测坐标变换(warp),把特征沿球面的大圆(测地线)搬运,以此交换不同位置的信息,不使用卷积、注意力或傅里叶层来完成这一步。论文同时发布七套球面 PDE 评测数据,作者报告 Dandelion 在每个数据集上都是最优或次优,并且随着分辨率提高,相对不采用坐标变换的对照方法,优势进一步扩大。

平面网格用于球面,会扭曲距离与面积

天气、海洋、行星大气这些过程天然定义在球面 S² 上,但主流科学机器学习工具箱默认的是平面欧氏网格。把欧氏架构硬套到经纬网格(lat–lon grid)上会引入几类系统性误差:

  • 经纬网格是等角而非等面积的:靠近两极的网格单元实际面积按 sin θ 缩小,笛卡尔卷积在高纬度被"压扁"。
  • 傅里叶神经算子(FNO)里的 2D FFT 假设平面双周期(环面),但经纬网格不是环面——最上一行不会绕回最下一行。
  • ViT 的笛卡尔位置编码把球面上的测地距离扭曲成平面距离。

论文作者认为,扩大模型并不能自动解决这些问题。模型可能记住随纬度变化的强迫模式,却没有学会在球面旋转后仍然成立的流体物理规律。

方法:用坐标变换代替卷积

Dandelion 是 Flower 架构的球面版。Flower 是作者团队今年早些时候提出的 warp-based 求解器,它通过多头坐标变换完成空间信息混合:每个头预测一个逐像素位移场,再从位移后的位置采样特征,计算成本随网格规模线性增长,不含傅里叶层、注意力或卷积。据 Dandelion 论文转述,17M 参数的 Flower 已在 The Well 与 PDEBench 的 16 个 2D/3D benchmark 上超过同规模的谱、卷积、注意力基线。

问题在于 Flower 的几个基本操作依赖平面几何:像素位移对应着随纬度变化的弧长;周期 padding 在经纬网格上不成立;步长 2×2 池化的方形感受野物理范围也随纬度变化。Dandelion 将这些受网格影响的操作改为直接适用于球面 S² 的形式。

第一处是切平面 warp。模型在查询点 p 的切平面 T_p S² 上预测位移(北向、东向两个分量),再用指数映射——在球面上可写为罗德里格斯公式——把切向量沿大圆旋转回球面。因为切向量的模等于沿测地线走过的弧长,warp 距离与纬度无关,感受野不再受参数化畸变影响。

第二处是球谐域池化。分层池化分两步:先把特征投影到带限子空间,即通过低通滤波保留较低频率的信息,再在粗网格上取值。球谐是 S² 上拉普拉斯–贝尔特拉米算子的本征基,每个 ℋ_ℓ 都是旋转群 SO(3) 的不可约表示,因此谱截断与旋转作用交换——分辨率变化按构造保持旋转对称。相比之下,经纬网格上的步长池化既不旋转等变,其池化模板在赤道与两极覆盖的物理面积也相差数量级。

这样得到的 U-Net 结构里没有卷积:空间混合完全由 warp 完成,唯一剩下的"卷积形状"算子是一层在每个位置线性混合通道的操作。球谐变换用 torch-harmonics 实现(因等角节点的离散球谐变换在 Nyquist 度处不精确,该投影实现上是近似的)。

七套模拟数据,供研究者快速比较不同模型

作者还发布了一套评测基准,用来补充现有球面数据集的空缺:一边是几个过于简化的问题(如无地形的受迫二维浅水方程),不足以区分方法;另一边是 ERA5 这类运行级再分析数据(经 WeatherBench 2),单次超参数搜索就可能耗掉数千 GPU-天,无法用于架构调试。新基准希望兼顾两者:覆盖足够多样的物理过程,以比较不同架构,同时控制数据规模,让模型能在几小时内从零完成训练。

七套数据集包括:

  • barotropic-jets:双半球 Galewsky 急流,随机旋转以迫使模型从演化中推断科里奥利力,而非背下固定网格;
  • active-turbulence:反常"反铁磁"涡旋链湍流,一种没有平面对应的闭流形效应;
  • cahn-hilliard:双组分相分离,考验四阶双调和算子和全局质量守恒;
  • shock-caps:浅水黎曼问题,多个测地圆帽产生的激波在球面上碰撞、聚焦;
  • dry-atmosphere-3D:Held–Suarez 干大气环流;
  • global-ocean-3D:100 年全球海洋模拟;
  • 加上 The Well 的 planetswe。

高分辨率下相对误差优势扩大,速度仍慢于 FNO

在 20M 参数对齐的条件下,与 FNO、SFNO、局部 R²/S² 邻域 transformer 及平面版 Flower2D 相比,Dandelion 在 7 个数据集的单步预测上 6 个最优,20 步连续预测(rollout)上 5 个最优,且处处都是最优或次优。球面 warp 相对平面 Flower2D 的增益在 rollout 上最清楚:planetswe 误差降到约 1/1.6,shock-caps 约 1/1.5,barotropic-jets 约 1/1.8,active-turbulence 约 1/4.5。

提高分辨率后,方法之间的差距进一步显现:在 256×512 分辨率下,单步预测与 rollout 的前两名都是 Dandelion 和 Flower2D,对非 warp 基线的领先随分辨率提高而扩大。计算吞吐量也影响这些方法能否实用。S² 邻域 transformer 在 128×256 及以上分辨率比 FNO 慢 30–40 倍,R² 邻域 transformer 慢 5–6 倍,而 Dandelion 约慢 2.2–2.6 倍。因此,在高分辨率任务中,除了比较参数量和预测误差,还需要考察模型的计算吞吐量。

尚未验证业务级分辨率与更长时间的预测稳定性

这是一篇未经同行评审的预印本,所有性能数字都是作者自己的实验结论,尚无第三方复现。论文自述的限制同样重要:分辨率上限 256×512(约 0.7°),而运行级天气模型已到 0.25°(720×1440);训练只在损失里展开最多两步自回归,20 步以上的稳定性和漂移没有测量;benchmark 本身是合成的,作者也承认在这些问题上表现好只是天气模型的"必要非充分条件"。

Dandelion 展示了仅用坐标变换在球面上混合信息的可行性,也提供了一套能在几小时内比较架构的评测数据。下一步需要验证,它能否扩展到 0.25° 的 ERA5 尺度,以及更长时间的连续预测能否保持稳定。这两项结果将决定它距离实际天气与海洋业务还有多远。

来源

  • Dandelion: A Spherical Flower for Neural Simulation of Planetary Dynamics
  • Flowers: A Warp Drive for Neural PDE Solvers

评论

登录后即可参与评论。

加载评论中…

相关文章

  • 蓝黄云层背景上的 WeatherNext 3 标题图

    Google 天气模型加入逐小时短期预报,补入卫星观测

    2026-09-03

  • Theo Conjecture 系统架构图,展示从数学对象注册表到 LLM Agent、精确计算测试和人类评审的循环流程

    AI 系统协助证明 Erdős–Staton 图论猜想,手稿仍待同行评审

    2026-07-30

  • 月球勘测轨道飞行器窄角相机拍摄的 Mons Rümker 火山地貌拼接图

    NASA 与 IBM 开源月球模型,近 200 万图块先解决数据对齐

    2026-09-11

AGI Daily跟踪世界 AI 进展
邮件栏目主题RSS
定制投稿招聘广告
关于MCPTermsContact
DexTouch-WM 论文示意图:人类佩戴触觉手套操作物体,触觉表示与动作被对齐到双臂灵巧手机器人

100 小时人类触觉提高了机器人的接触预测能力

2026-09-17

  • Dual-Flow 论文图 1:NanoGPT 设定下标准 Transformer 与 Dual-Flow 在多个数据预算上的验证损失训练轨迹,Dual-Flow 在每个预算都更低

    新 Transformer 架构分别调节提示词处理与答案生成的计算量

    2026-08-16

  • WeatherNext 2 使用的 Functional Generative Networks 架构示意图,展示从单一输入通过独立训练的神经网络注入函数空间噪声生成多个物理自洽的天气预报场景

    DeepMind 发布气旋预测模型 WeatherNext,预报精度优势约相当于提前一天

    2026-08-06