AGI Daily
广告
AGI Daily跟踪世界 AGI 进展AGI INTELLIGENCE · DAILY
全部算力算法数据人物日报主题
AGI Daily
全部算力算法数据人物日报主题
广告

Dandelion:只用坐标变换、无卷积的球面神经 PDE 求解器

今天 · 共 2,510 字

8 月 27 日,arXiv 上线预印本《Dandelion: A Spherical Flower for Neural Simulation of Planetary Dynamics》。研究者 Till Muser、Giovanni Abati 与 Ivan Dokmanić 提出一个"原生球面"的神经 PDE 求解器:它的空间信息混合不靠卷积、注意力或傅里叶层,只靠预测坐标变换(warp),把特征沿球面的大圆(测地线)搬运。论文同时发布七套球面 PDE benchmark,作者报告 Dandelion 在每个数据集上都是最优或次优,并且对非 warp 基线的领先随分辨率提高而扩大。

球面上的"错误坐标系"

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

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

论文作者认为这不是靠加大模型能自动补上的:高容量模型可以背下随纬度变化的强迫模式,却不学到应随旋转而普适的流体物理。

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

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

问题在于 Flower 的几个原语假设了平面几何:像素位移对应着随纬度变化的弧长;周期 padding 在经纬网格上不成立;步长 2×2 池化的方形感受野物理范围也随纬度变化。Dandelion 把这三处网格敏感的原语换成 S² 上的原生对应物。

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

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

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

一个填补空白的 benchmark

作者还发布了一套基准,理由是球面数据集两端失衡:一边是几个玩具问题(如无地形的受迫二维浅水方程),不足以区分方法;另一边是 ERA5 这类运行级再分析数据(经 WeatherBench 2),单次超参数搜索就可能耗掉数千 GPU-天,无法用于架构调试。新 benchmark 处在中间:物理上足够多样以压测架构选择,又小到几小时内可以从零训练。

七套数据集包括:

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

结果:领先随分辨率扩大

在 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 尺度、长 rollout 是否稳定,是判断它能否真正进入天气与海洋业务的两个可验证变量。

来源

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

评论

登录后即可参与评论。

加载评论中…

相关文章

  • 合成图:智能手机屏幕上显示 Cisco 标志,背景为人工智能相关符号

    Cisco 向 9 万员工铺开 MyAgent:受监督的自主执行如何在大企业中落地

    算法 · 2026-08-31

  • 腾讯混元 Hy4 preview 模型卡中的基准测试结果图表

    腾讯开源混元 Hy4:1M 上下文工程化落地,模型开始优化自身基础设施

    算法 · 2026-08-31

  • Claude 记忆设置中的 Topics 列表,展示 Claude 按主题保存的可编辑记忆条目

    Anthropic 合并 Claude 与 Cowork 记忆:默认开启,敏感类别默认不存

    算法 · 2026-08-28

  • Anthropic Model Hardware Standard(MHS)演示画面

    Anthropic 发布 MHS:给 AI agent 一套操作物理设备的互操作标准

    算法 · 2026-08-27

  • DeepSeek 新一轮融资配图:现有股东砺思资本、拾象与宁德时代将注入新资金

    DeepSeek 逼近 740 亿美元估值与科创板 IPO:两轮融资背后的算力账

    算法 · 2026-08-27

  • 上海世界人工智能大会上,人们走过宣传 Kimi K3 模型的月之暗面展位

    月之暗面与美国三大云谈 Kimi K3 分成,开源权重变现撞上地缘政治

    算法 · 2026-08-27

AGIDaily跟踪世界 AGI 进展© 2026 AGIDaily
关于邮件栏目主题定制投稿招聘广告SkillTermsContactRSS