摘要
数值天气预报 (NWP) 系统是现代社会的重要基础设施。然而,传统的 NWP 系统通过使用大型计算集群求解复杂的偏微分方程来解决问题,导致了大量的碳排放。探索高效且环保的 NWP 解决方案引起了人工智能 (AI) 和地球科学界的兴趣。为了缩小基于 AI 的方法与物理预测器之间的性能差距,本文提出了一种新的基于 Transformer 的 NWP 框架,称为 WeatherFormer,用于模拟复杂的时空大气动力学,并增强数据驱动 NWP 的能力。WeatherFormer 创新地引入了时空分解 Transformer 模块,以减少参数和内存消耗,其中提出了位置感知自适应傅里叶神经算子 (PAFNO) 用于位置敏感的令牌混合。此外,还采用了两种数据增强策略来提高性能并减少训练消耗。在 WeatherBench 数据集上的大量实验表明,WeatherFormer 在现有深度学习方法中取得了优异的性能,并进一步接近最先进的物理模型。