Machine learning emulation of precipitation from km-scale regional climate simulations using a diffusion model

Henry Addison, Elizabeth Kendon, Suman Ravuri, Laurence Aitchison, Peter AG Watson
7/19/2024
physics.ao-phcs.LG

No official code found. Best implementation: henryaddison/mlde (22 stars, Apache-2.0, last commit 2026-02-11). 1 more below.

Abstract

High-resolution climate simulations are valuable for understanding climate change impacts. This has motivated use of regional convection-permitting climate models (CPMs), but these are very computationally expensive. We present a convection-permitting model generative emulator (CPMGEM), to skilfully emulate precipitation simulations by a 2.2km-resolution regional CPM at much lower cost. This utilises a generative machine learning approach, a diffusion model. It takes inputs at the 60km resolution of the driving global climate model and downscales these to 8.8km, with daily-mean time resolution, capturing the effect of convective processes represented in the CPM at these scales. The emulator is trained on simulations over England and Wales from the United Kingdom Climate Projections Local product, covering years between 1980 and 2080 following a high emissions scenario. The output precipitation has a similarly realistic spatial structure and intensity distribution to the CPM simulations. The emulator is stochastic, which improves the realism of samples. We show evidence that the emulator has skill for extreme events with ~100 year return times. It captures the main features of the simulated 21st century climate change, but exhibits some error in the magnitude. We demonstrate successful transfer from a "perfect model" training setting to application using GCM variable inputs. We also show that the method can be useful in situations with limited amounts of high-resolution data. Potential applications include producing high-resolution precipitation predictions for large-ensemble climate simulations and producing output based on different GCMs and climate change scenarios to better sample uncertainty.

View on arXivView PDF

Code Implementations(2)

henryaddison/mldeCommunity100%
Apache-2.0

Adaptation of PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations for emulating high resolution climate models

228C++, CudaJan 12, 20228 months agoApache-2.0
MIT

Evalution notebooks for "Machine learning emulation of precipitation from km-scale regional climate simulations using a diffusion model", a paper on ML approaches to emulate the Met Office's UK CPM precipitation output from UKCP Local, including introducing CPMGEM.

00Jupyter NotebookMar 27, 202511 months agoMIT

Similar Papers

Cite this paper

@article{addison2024machine,
  title  = {Machine learning emulation of precipitation from km-scale regional climate simulations using a diffusion model},
  author = {Henry Addison and Elizabeth Kendon and Suman Ravuri and Laurence Aitchison and Peter AG Watson},
  year   = {2024},
  eprint = {2407.14158},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2407.14158v2}
}

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/2407.14158.svg)](https://codeofpaper.com/paper/2407.14158)

Discussion