Sensitivity Analysis for Climate Science with Generative Flow Models (Papers Track)
Alex Dobra (University of Oxford); Jakiw Pidstrigach (University of Oxford); Tim Reichelt (Univeristy of Oxford); Paolo Fraccaro (IBM Research Europe); Johannes Jakubik (IBM Research Europe); Anne Jones (IBM Research Europe); Christian Schroeder de Witt (University of Oxford); Philip Torr (University of Oxford); Philip Stier (University of Oxford)
Abstract
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