arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2405.03376cs.LGcs.CV

CRA5:通过高效变分 Transformer 对 ERA5 进行极端压缩,以支持便携的全球气候与天气研究

CRA5: Extreme Compression of ERA5 for Portable Global Climate and Weather Research via an Efficient Variational Transformer

Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);The Hong Kong University of Science and Technology(香港科技大学);The University of Sydney(悉尼大学)

Tao Han, Zhenghao Chen, Song Guo, Wanghan Xu, Lei Bai

AI总结:

提出低复杂度 VAEformer 神经编解码器,将 226 TB 的 ERA5 极端压缩为 0.7 TB 的 CRA5,压缩率超 300 倍,并保持科学分析及下游全球天气预报性能。

摘要:

数据驱动天气预报模型从数百 TB(太字节)再分析数据中学习,其出现显著提升了预报能力。然而,与数据存储和传输相关的高昂成本给数据提供者和用户带来了重大挑战,影响资源受限的研究人员,并限制他们参与基于 AI 的气象研究的可及性。为缓解这一问题,我们提出一种高效神经编解码器 Variational Autoencoder Transformer(VAEformer,变分自编码器 Transformer),用于气候数据的极端压缩,以显著降低数据存储成本,使基于 AI 的气象研究对研究人员而言更具便携性。与近期复杂的神经编解码器不同,我们的方法采用低复杂度 Auto-Encoder transformer。该编码器通过方差推理生成量化的潜在表示,并将潜在空间重参数化为高斯分布。该方法改进了用于交叉熵编码的分布估计。大量实验表明,在气候数据场景下,我们的 VAEformer 优于现有最先进压缩方法。通过应用 VAEformer,我们将最常用的 ERA5 气候数据集(226 TB)压缩为一个名为 CRA5 的新数据集(0.7 TB)。这意味着压缩率超过 300 倍,同时保留了该数据集用于精确科学分析的效用。此外,下游实验表明,在紧凑的 CRA5 数据集上训练的全球天气预报模型,可达到与在原始数据集上训练的模型相当的预报精度。代码、CRA5 数据集和预训练模型可在 https://github.com/taohan10200/CRA5 获取。

英文摘要:

The advent of data-driven weather forecasting models, which learn from hundreds of terabytes (TB) of reanalysis data, has significantly advanced forecasting capabilities. However, the substantial costs associated with data storage and transmission present a major challenge for data providers and users, affecting resource-constrained researchers and limiting their accessibility to participate in AI-based meteorological research. To mitigate this issue, we introduce an efficient neural codec, the Variational Autoencoder Transformer (VAEformer), for extreme compression of climate data to significantly reduce data storage cost, making AI-based meteorological research portable to researchers. Our approach diverges from recent complex neural codecs by utilizing a low-complexity Auto-Encoder transformer. This encoder produces a quantized latent representation through variance inference, which reparameterizes the latent space as a Gaussian distribution. This method improves the estimation of distributions for cross-entropy coding. Extensive experiments demonstrate that our VAEformer outperforms existing state-of-the-art compression methods in the context of climate data. By applying our VAEformer, we compressed the most popular ERA5 climate dataset (226 TB) into a new dataset, CRA5 (0.7 TB). This translates to a compression ratio of over 300 while retaining the dataset's utility for accurate scientific analysis. Further, downstream experiments show that global weather forecasting models trained on the compact CRA5 dataset achieve forecasting accuracy comparable to the model trained on the original dataset. Code, the CRA5 dataset, and the pre-trained model are available at https://github.com/taohan10200/CRA5.

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

↑