Saved in:
Bibliographic Details
Main Authors: Mirowski, Piotr, Warde-Farley, David, Rosca, Mihaela, Grimes, Matthew Koichi, Hasson, Yana, Kim, Hyunjik, Rey, Mélanie, Osindero, Simon, Ravuri, Suman, Mohamed, Shakir
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2407.11666
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909258979737600
author Mirowski, Piotr
Warde-Farley, David
Rosca, Mihaela
Grimes, Matthew Koichi
Hasson, Yana
Kim, Hyunjik
Rey, Mélanie
Osindero, Simon
Ravuri, Suman
Mohamed, Shakir
author_facet Mirowski, Piotr
Warde-Farley, David
Rosca, Mihaela
Grimes, Matthew Koichi
Hasson, Yana
Kim, Hyunjik
Rey, Mélanie
Osindero, Simon
Ravuri, Suman
Mohamed, Shakir
contents Atmospheric states derived from reanalysis comprise a substantial portion of weather and climate simulation outputs. Many stakeholders -- such as researchers, policy makers, and insurers -- use this data to better understand the earth system and guide policy decisions. Atmospheric states have also received increased interest as machine learning approaches to weather prediction have shown promising results. A key issue for all audiences is that dense time series of these high-dimensional states comprise an enormous amount of data, precluding all but the most well resourced groups from accessing and using historical data and future projections. To address this problem, we propose a method for compressing atmospheric states using methods from the neural network literature, adapting spherical data to processing by conventional neural architectures through the use of the area-preserving HEALPix projection. We investigate two model classes for building neural compressors: the hyperprior model from the neural image compression literature and recent vector-quantised models. We show that both families of models satisfy the desiderata of small average error, a small number of high-error reconstructed pixels, faithful reproduction of extreme events such as hurricanes and heatwaves, preservation of the spectral power distribution across spatial scales. We demonstrate compression ratios in excess of 1000x, with compression and decompression at a rate of approximately one second per global atmospheric state.
format Preprint
id arxiv_https___arxiv_org_abs_2407_11666
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Neural Compression of Atmospheric States
Mirowski, Piotr
Warde-Farley, David
Rosca, Mihaela
Grimes, Matthew Koichi
Hasson, Yana
Kim, Hyunjik
Rey, Mélanie
Osindero, Simon
Ravuri, Suman
Mohamed, Shakir
Machine Learning
Computer Vision and Pattern Recognition
Atmospheric and Oceanic Physics
Atmospheric states derived from reanalysis comprise a substantial portion of weather and climate simulation outputs. Many stakeholders -- such as researchers, policy makers, and insurers -- use this data to better understand the earth system and guide policy decisions. Atmospheric states have also received increased interest as machine learning approaches to weather prediction have shown promising results. A key issue for all audiences is that dense time series of these high-dimensional states comprise an enormous amount of data, precluding all but the most well resourced groups from accessing and using historical data and future projections. To address this problem, we propose a method for compressing atmospheric states using methods from the neural network literature, adapting spherical data to processing by conventional neural architectures through the use of the area-preserving HEALPix projection. We investigate two model classes for building neural compressors: the hyperprior model from the neural image compression literature and recent vector-quantised models. We show that both families of models satisfy the desiderata of small average error, a small number of high-error reconstructed pixels, faithful reproduction of extreme events such as hurricanes and heatwaves, preservation of the spectral power distribution across spatial scales. We demonstrate compression ratios in excess of 1000x, with compression and decompression at a rate of approximately one second per global atmospheric state.
title Neural Compression of Atmospheric States
topic Machine Learning
Computer Vision and Pattern Recognition
Atmospheric and Oceanic Physics
url https://arxiv.org/abs/2407.11666