arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.09360physics.ao-phcs.AIcs.LGphysics.comp-ph

迈向物理一致的数据驱动天气预报:将数据同化与保持等变性的深度空间变换器相结合

Towards physically consistent data-driven weather forecasting: Integrating data assimilation with equivariance-preserving deep spatial transformers

Rice University(莱斯大学);Lawrence Berkeley National Laboratory(劳伦斯伯克利国家实验室);University of Maryland(马里兰大学)

Ashesh Chattopadhyay, Mustafa Mustafa, Pedram Hassanzadeh, Eviatar Bach, Karthik Kashinath

AI总结:

本文提出将保持等变性的深度空间变换器、数据同化和多时间步算法集成到数据驱动天气预报模型中,以提升物理一致性与短期预报精度,并通过ERA5的Z500数据验证其显著优于U-NET。

摘要:

人们对数据驱动天气预报(DDWP)的兴趣日益增长,例如使用在模型或再分析数据上训练的卷积神经网络(如U-NET)。在此,我们提出3个组件,与常用的DDWP模型集成,以提高其物理一致性和预报精度。这些组件是:1)一个添加到U-NET潜在空间中的深度空间变换器,用于保持一种称为等变性的性质,该性质与正确捕获时空数据中特征的旋转和缩放有关;2)一个数据同化(DA)算法,用于吸收含噪声观测并改善下一次预报的初始条件;3)一个多时间步算法,通过DA将不同时间步的DDWP模型预报相结合,提高短时间间隔预报的精度。为了展示每个组件的优势/可行性,我们使用ERA5再分析中的500 hPa位势高度(Z500),并检验DDWP框架特定设置的短期预报精度。结果表明,保持等变性的网络(U-STN)明显优于U-NET,例如将预报技巧提高了45%。使用sigma点集合卡尔曼(SPEnKF)算法进行DA并以U-STN作为前向模型,我们表明即使在高观测噪声下也能实现稳定、准确的DA循环。DDWP+DA框架从每个DA循环中用数据驱动前向模型廉价生成的大规模(O(1000))集合中获益显著。多时间步DDWP+DA框架也显示出前景,例如它将平均误差降低了2-3倍。

英文摘要:

There is growing interest in data-driven weather prediction (DDWP), for example using convolutional neural networks such as U-NETs that are trained on data from models or reanalysis. Here, we propose 3 components to integrate with commonly used DDWP models in order to improve their physical consistency and forecast accuracy. These components are 1) a deep spatial transformer added to the latent space of the U-NETs to preserve a property called equivariance, which is related to correctly capturing rotations and scalings of features in spatio-temporal data, 2) a data-assimilation (DA) algorithm to ingest noisy observations and improve the initial conditions for next forecasts, and 3) a multi-time-step algorithm, which combines forecasts from DDWP models with different time steps through DA, improving the accuracy of forecasts at short intervals. To show the benefit/feasibility of each component, we use geopotential height at 500~hPa (Z500) from ERA5 reanalysis and examine the short-term forecast accuracy of specific setups of the DDWP framework. Results show that the equivariance-preserving networks (U-STNs) clearly outperform the U-NETs, for example improving the forecast skill by $45\%$. Using a sigma-point ensemble Kalman (SPEnKF) algorithm for DA and U-STN as the forward model, we show that stable, accurate DA cycles are achieved even with high observation noise. The DDWP+DA framework substantially benefits from large ($O(1000)$) ensembles that are inexpensively generated with the data-driven forward model in each DA cycle. The multi-time-step DDWP+DA framework also shows promises, e.g., it reduces the average error by factors of 2-3.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑