S3N 用有限元 LL–HP 映射与四螺旋状态空间扫描,把 10 天预报 nRMSE 降到 0.5177
相关研究与后续进展核心概要
该工作提出 S3N 球面螺旋扫描网络,用基于曲面有限元的 L2Proj 在经纬度网格与 HEALPix 等面积网格之间双向映射大气场,并用跨纬度注意力引导的四条极到极螺旋路径选择性状态空间扫描实现跨基面信息传播;在 ERA5/WeatherBench 2 上,S3N 在 4、7、10 天预报取得更低 nRMSE(10 天 0.5177,对比 OneForecast 的 0.6398),且误差增长更慢,但 6 小时与 1 天短期精度不及部分对比模型。
Figure 1: Comparison of various methods. (a) The average error of L2Proj is much lower than that of bilinear interpolation. (b) We treat HEALPix pixels from the twelve base faces as a sequence rather than stacking the twelve base faces as patches, thereby avoiding handcrafted boundary handling.
arXiv深度剖析
S3N 在长预报时效上取得更低误差:4、7、10 天 nRMSE 分别为 0.3171、0.4411、0.5177,10 天较次优的 OneForecast(0.6398)降低 19.1%。 相对以往以 LL 网格或分块 HEALPix 处理为主的 MLWP 方法,该框架把等面积球面表示与全局跨面信息传播统一起来,使优势随预报时效延长而更明显。 在 ERA5/WeatherBench 2 数据上,1979–2017 训练、2018–2019 验证、2020 测试,69 个大气场,6 小时时间分辨率,对比实验在相同条件下进行;表 2 给出各模型在各时效的 nRMSE。
L2Proj 把 LL 与 HP 数据表示为单位球面上的连续分片线性场,通过投影求解映射,双向映射平均 MSE 为 0.000252,低于双线性插值的 0.002153。 以往 LL–HP 映射多依赖逐点或学习式重采样,而两网格采样密度与分辨率不同;L2Proj 用统一有限元表述处理非匹配球面网格,且质量矩阵与跨网格耦合矩阵可离线预计算复用。 用 2017 与 2020 数据,将 69 个标准化大气场先由 LL 映射到 HP 再映回 LL,与原始 LL 场比较重建误差,表 1 报告全变量平均 MSE。
AQSS 块用跨纬度注意力引导沿四条极到极螺旋路径的选择性状态空间更新,使隐藏状态直接跨 HEALPix 基面边界传播,无需面间填充或移位窗口。 以往 HP 方法把像素组织为独立基面或局部窗口,跨面通信依赖手工边界处理、旋转或复制填充、移位窗口、掩码或特定布局索引;AQSS 以全局螺旋路径替代这些机制。 方法部分给出跨纬度八邻域采样、多头注意力、上下文对扫描输入与 Mamba 动态参数的调制,以及四条 RING 序螺旋路径的选择性扫描公式;消融显示去掉注意力引导的 SSM 参数调制后 10 天 nRMSE 由 0.5177 升至 2.3103。
消融表明加权路径融合主要改善短期精度,而注意力引导的 SSM 参数调制对抑制自回归误差累积最关键。 该分解把长程稳定性归因到具体组件,而非整体架构,为后续设计提供可检验的组件级依据。 表 4 在 6 小时至 10 天各时效比较完整模型与固定权重融合、去 SSM 引导、去注意力、去输入融合等变体,完整模型在所有时效 nRMSE 最低。
启示与展望
该结果面向使用 ERA5/WeatherBench 2 数据、6 小时时间分辨率、69 个大气场、HP 表示含 49152 个等面积像素的全球确定性中期预报设定;训练为单步预测加长程自回归微调,在单张 NVIDIA RTX 6000D(84 GB)上耗时少于 12 天。可直接复用的部分包括:L2Proj 的离线预计算映射矩阵(质量矩阵与跨网格耦合矩阵仅依赖节点位置与两网格分辨率,训练与推理时无需网格求交或数值积分),以及 AQSS 以 RING 序构造四条极到极螺旋路径、每条路径恰好访问每个 HP 像素一次的设计。对从事球面网格表示、等面积离散化或长程自回归稳定性的研究者,该框架提供了可迁移的组件;作者指出后续将改进两网格间映射分辨率与短期预报性能。
映射评估使用 2017 与 2020 数据,表 1 只报告全变量平均 MSE,未在文中展开逐变量分布;主结果与消融基于 2020 年测试与单一数据集,跨数据集、跨分辨率与概率化预报的表现仍是开放问题。消融中固定权重融合与去 SSM 引导的变体在短期时效误差明显更高,说明组件间耦合较强,替换其中任一环节时需重新校准。作者将改进映射分辨率与短期预报列为后续工作,因此当前版本在 6 小时与 1 天时效上的相对位置应视为该设定下的观察,而非普适结论。
