Probabilistic Emulation of a Global Climate Model with Spherical DYffusion

Salva Rühling Cachay, Brian Henn, Oliver Watt-Meyer, Christopher S. Bretherton, Rose Yu
6/21/2024
cs.LGcs.AIphysics.ao-ph

No official code found. Best implementation: Rose-STL-Lab/spherical-dyffusion (31 stars, Apache-2.0, PyTorch, last commit 2025-03-07).

Abstract

Data-driven deep learning models are transforming global weather forecasting. It is an open question if this success can extend to climate modeling, where the complexity of the data and long inference rollouts pose significant challenges. Here, we present the first conditional generative model that produces accurate and physically consistent global climate ensemble simulations by emulating a coarse version of the United States' primary operational global forecast model, FV3GFS. Our model integrates the dynamics-informed diffusion framework (DYffusion) with the Spherical Fourier Neural Operator (SFNO) architecture, enabling stable 100-year simulations at 6-hourly timesteps while maintaining low computational overhead compared to single-step deterministic baselines. The model achieves near gold-standard performance for climate model emulation, outperforming existing approaches and demonstrating promising ensemble skill. This work represents a significant advance towards efficient, data-driven climate simulations that can enhance our understanding of the climate system and inform adaptation strategies.

View on arXivView PDF

Code Implementations(1)

PyTorchApache-2.0

[NeurIPS 24] Probablistic Emulation of a Global Climate Model with Spherical DYffusion

313Shell, PythonOct 31, 20241 years agoApache-2.0
ai-for-scienceclimate-modelingdiffusiondyffusionlarge-ensembles+4 more

Similar Papers

Cite this paper

@article{cachay2024probabilistic,
  title  = {Probabilistic Emulation of a Global Climate Model with Spherical DYffusion},
  author = {Salva Rühling Cachay and Brian Henn and Oliver Watt-Meyer and Christopher S. Bretherton and Rose Yu},
  year   = {2024},
  eprint = {2406.14798},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.14798v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.14798.svg)](https://codeofpaper.com/paper/2406.14798)

Discussion