Saved in:
Bibliographic Details
Main Authors: Dobra, Alex, Pidstrigach, Jakiw, Reichelt, Tim, Fraccaro, Paolo, Jones, Anne, Jakubik, Johannes, de Witt, Christian Schroeder, Torr, Philip, Stier, Philip
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2511.00663
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917142200320000
author Dobra, Alex
Pidstrigach, Jakiw
Reichelt, Tim
Fraccaro, Paolo
Jones, Anne
Jakubik, Johannes
de Witt, Christian Schroeder
Torr, Philip
Stier, Philip
author_facet Dobra, Alex
Pidstrigach, Jakiw
Reichelt, Tim
Fraccaro, Paolo
Jones, Anne
Jakubik, Johannes
de Witt, Christian Schroeder
Torr, Philip
Stier, Philip
contents Sensitivity analysis is a cornerstone of climate science, essential for understanding phenomena ranging from storm intensity to long-term climate feedbacks. However, computing these sensitivities using traditional physical models is often prohibitively expensive in terms of both computation and development time. While modern AI-based generative models are orders of magnitude faster to evaluate, computing sensitivities with them remains a significant bottleneck. This work addresses this challenge by applying the adjoint state method for calculating gradients in generative flow models. We apply this method to the cBottle generative model, trained on ERA5 and ICON data, to perform sensitivity analysis of any atmospheric variable with respect to sea surface temperatures. We quantitatively validate the computed sensitivities against the model's own outputs. Our results provide initial evidence that this approach can produce reliable gradients, reducing the computational cost of sensitivity analysis from weeks on a supercomputer with a physical model to hours on a GPU, thereby simplifying a critical workflow in climate science. The code can be found at https://github.com/Kwartzl8/cbottle_adjoint_sensitivity.
format Preprint
id arxiv_https___arxiv_org_abs_2511_00663
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Sensitivity Analysis for Climate Science with Generative Flow Models
Dobra, Alex
Pidstrigach, Jakiw
Reichelt, Tim
Fraccaro, Paolo
Jones, Anne
Jakubik, Johannes
de Witt, Christian Schroeder
Torr, Philip
Stier, Philip
Machine Learning
Sensitivity analysis is a cornerstone of climate science, essential for understanding phenomena ranging from storm intensity to long-term climate feedbacks. However, computing these sensitivities using traditional physical models is often prohibitively expensive in terms of both computation and development time. While modern AI-based generative models are orders of magnitude faster to evaluate, computing sensitivities with them remains a significant bottleneck. This work addresses this challenge by applying the adjoint state method for calculating gradients in generative flow models. We apply this method to the cBottle generative model, trained on ERA5 and ICON data, to perform sensitivity analysis of any atmospheric variable with respect to sea surface temperatures. We quantitatively validate the computed sensitivities against the model's own outputs. Our results provide initial evidence that this approach can produce reliable gradients, reducing the computational cost of sensitivity analysis from weeks on a supercomputer with a physical model to hours on a GPU, thereby simplifying a critical workflow in climate science. The code can be found at https://github.com/Kwartzl8/cbottle_adjoint_sensitivity.
title Sensitivity Analysis for Climate Science with Generative Flow Models
topic Machine Learning
url https://arxiv.org/abs/2511.00663