AI Emulation of Stochastic Sudden Stratospheric Warming with Interpretable Latent Structure

Abstract

Rare weather regime transitions pose a challenge for data-driven modeling due to class imbalance. In this study, we develop a probabilistic deep learning emulator for a prototypical system with regime transitions, the stochastic Holton--Mass model of stratospheric variability, and analyze the structure of its learned latent space. The Holton--Mass model exhibits two metastable regimes, a strong and a weak polar vortex, maintained by nonlinear wave--mean flow interactions, with weak stochastic forcing intermittently triggering rare transitions between these regimes that qualitatively represent SSW events. We employ a ResNet-inspired Conditional Variational Autoencoder with six-layer encoder and decoder layers and explicit current-state conditioning to model the distribution of the system's state at the next time step (one day). The emulator accurately reproduces short-term dynamics, steady-state probability distributions, regime persistence statistics, rare transition rates, the transition committor function, and the transition expected lead time of the physical model. Beyond emulation fidelity, we interrogate the learned latent representation to understand how the model internalizes the underlying metastable structure of the dynamics. Principal Component Analysis of the 32-dimensional latent space reveals a clear and unsupervised separation into four physically interpretable clusters corresponding to strong versus weak vortex regimes and stable versus transition-prone configurations. Such emergent regime separation in latent space is hard to identify for deep generative models applied to high-dimensional stochastic systems. Our results show that carefully designed probabilistic emulators can uncover physically meaningful manifolds governing extreme-event dynamics, potentially aiding the development of improved operational advanced warning systems.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

More in Machine Learning

• arXiv preprint · 1 October 2026

  Varda-single-1.0: deterministic data-driven weather forecasting at 1 km resolution over Switzerland's complex topography

  Alberto Pennino, Francesco Zanetta, Michele Cattaneo and 28 others
• arXiv preprint · 2 October 2026

  S2S-JEPA: Predicting the Predictable at Subseasonal-to-Seasonal Timescales

  Chenyu Dong and Gianmarco Mengaldo
• arXiv preprint · 2 October 2026

  SDECast: Probabilistic Weather Forecasting in Continuous Time with Neural SDEs

  Maria Marchenko, Martin Andrae, Fredrik Lindsten and 1 other
• arXiv preprint · 30 September 2026

  A library for differentiable signal processing and machine learning on the sphere

  Thorsten Kurth, Max Rietmann, Mauro Bisson and 6 others
• arXiv preprint · 2 October 2026

  DAWIS: Data Assimilation with Windowed Inverse Sampling via Multitask Interpolants

  Erik Wikingsson, Martin Andrae, Tomas Landelius and 1 other

All new papers in Machine Learning