盘古天气:一种用于快速准确全球天气预报的 3D 高分辨率模型

Pangu-Weather: A 3D High-Resolution Model for Fast and Accurate Global Weather Forecast

谢凌曦 Lingxi Xie · Huawei · 2022-11-03 · arXiv:2211.02556 ↗ · 被引 273

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

本文提出盘古天气,一种基于深度学习的快速准确全球天气预报系统。为此,我们建立了数据驱动环境,从 ECMWF 第五代再分析(ERA5)数据中下载了 43 年的每小时全球天气数据,并训练了总参数量约 2.56 亿的若干深度神经网络。预报的空间分辨率为 0.25°×0.25°,与 ECMWF 综合预报系统(IFS)相当。更重要的是,AI 方法首次在所有要素(如位势、比湿、风速、温度等)和所有时间范围(从 1 小时到 1 周)的准确度(纬度加权 RMSE 和 ACC)上超越了最先进的数值天气预报(NWP)方法。提高预测精度的两个关键策略是:(i)设计了一种 3D 地球特定 Transformer(3DEST)架构,将高度(气压层)信息公式化为立方体数据;(ii)应用分层时间聚合算法以减轻累积预报误差。在确定性预报中,盘古天气在短期到中期预报(即预报时间范围从 1 小时到 1 周)中显示出巨大优势。盘古天气支持多种下游预报场景,包括极端天气预报(如热带气旋跟踪)和实时大集合预报。盘古天气不仅结束了关于 AI 方法能否超越传统 NWP 方法的争论,还揭示了改进深度学习天气预报系统的新方向。

In this paper, we present Pangu-Weather, a deep learning based system for fast and accurate global weather forecast. For this purpose, we establish a data-driven environment by downloading $43$ years of hourly global weather data from the 5th generation of ECMWF reanalysis (ERA5) data and train a few deep neural networks with about $256$ million parameters in total. The spatial resolution of forecast is $0.25^\circ\times0.25^\circ$, comparable to the ECMWF Integrated Forecast Systems (IFS). More importantly, for the first time, an AI-based method outperforms state-of-the-art numerical weather prediction (NWP) methods in terms of accuracy (latitude-weighted RMSE and ACC) of all factors (e.g., geopotential, specific humidity, wind speed, temperature, etc.) and in all time ranges (from one hour to one week). There are two key strategies to improve the prediction accuracy: (i) designing a 3D Earth Specific Transformer (3DEST) architecture that formulates the height (pressure level) information into cubic data, and (ii) applying a hierarchical temporal aggregation algorithm to alleviate cumulative forecast errors. In deterministic forecast, Pangu-Weather shows great advantages for short to medium-range forecast (i.e., forecast time ranges from one hour to one week). Pangu-Weather supports a wide range of downstream forecast scenarios, including extreme weather forecast (e.g., tropical cyclone tracking) and large-member ensemble forecast in real-time. Pangu-Weather not only ends the debate on whether AI-based methods can surpass conventional NWP methods, but also reveals novel directions for improving deep learning weather forecast systems.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 16)

阅读逐段中英对照全文 →