arXiv Machine Learning

Geospatial Diffusion-based Evolution Synthesis (GeoDES) for Storm-Centered Weather Augmentation

arXiv:2607. 19522v1 Announce Type: new Abstract: While machine learning-based weather models hold significant promise, they struggle to predict the detailed structure of large-scale weather systems such as cyclonic storms.

arXiv Machine Learning
Sep 4

SurgeGen: A Hybrid Generative Diffusion Framework for Storm Surge Scenario Synthesis

SurgeGen is a two‑stage generative diffusion framework that synthesizes storm surge scenarios conditioned on continuous storm parameters. The first stage produces a coarse baseline surge estimate, which then conditions a diffusion model that refines the output to capture realistic spatial patterns and variability. The method can generate diverse, realistic surge scenarios both within and beyond the training distribution.

By Shunan Zheng, John J. Hasenbein
arXiv Statistics ML
Sep 11

Stress-Testing Dynamical and Generative Downscaling Using Subseasonal Extreme Precipitation Forecasts

The study compares the Weather Research and Forecasting (WRF) dynamical model with an unpaired diffusion-based generative model for downscaling extreme precipitation events up to three weeks ahead. Both models outperform raw European Centre for Medium-Range Weather Forecasts forecasts when evaluated against Swiss rain gauge-radar observations, but their strengths differ by atmospheric regime: WRF excels in a multicell, non‑stationary event, while the diffusion model performs more consistently and better in a stationary supercell event.

By Mauricio Lima, Marika Koukoula, Romain Pilon, Monika Feldmann, Erwan Koch, Daniela I. V. Domeisen, Tom Beucler
arXiv Machine Learning
Jun 9

Can AI Weather Models Predict Beyond Two Weeks? A Quantitative Benchmark and Analysis of Long Rollouts

arXiv:2605. 30184v2 Announce Type: replace Abstract: While AI weather models excel at short-to-medium range forecasts (up to 15 days), they frequently suffer from ill-defined "instabilities" when rolled out over longer horizons.

By Fanny Lehmann, Firat Ozdemir, Yun Cheng, Torsten Hoefler, Sebastian Schemm, Benedikt Soja, Siddhartha Mishra
arXiv Machine Learning
Sep 15

Generative diffusion models for spatiotemporal influenza forecasting

The paper introduces Influpaint, a denoising diffusion probabilistic model adapted for forecasting influenza incidence. By representing influenza seasons as spatiotemporal images and training on a hybrid dataset of surveillance and simulated trajectories, the model learns a rich distribution of disease dynamics and performs forecasting as a conditional generation task. In retrospective and real‑time evaluations, Influpaint produces realistic, diverse epidemic trajectories and achieves forecast accuracy competitive with leading ensemble methods, especially when trained with 30% surveillance and 70% simulated data.

By Joseph Lemaitre, Justin Lessler
arXiv AI
Sep 2

Global Attention with Linear Complexity for Exascale Generative Data Assimilation in Earth System Prediction

The paper introduces STORM, a one‑stage generative AI framework that reformulates Earth system data assimilation as diffusion‑based Bayesian posterior sampling, replacing costly PDE ensemble forecasts with scalable AI inference. STORM employs a spatiotemporal transformer with a global‑attention algorithm that reduces computational complexity from quadratic to linear, enabling high‑resolution, long‑context modeling. The system scales to 74,400 GPUs on Frontier, achieving 96–99 % strong‑scaling efficiency and up to 6 ExaFLOPs sustained BF16 throughput, while supporting 32,768‑member ensembles for uncertainty quantification in just 34 seconds on 4,096 GPUs, and demonstrates improved hurricane tracking and climate reanalysis accuracy.

By Xiao Wang, Zezhong Zhang, Isaac Lyngaas, Hong-Jun Yoon, Jong-Youl Choi, Siming Liang, Janet Wang, Hristo G. Chipilski, Ashwin M. Aji, Feng Bao, Peter Jan van Leeuwen, Dan Lu, Guannan Zhang
arXiv Machine Learning
Jun 30

High-Resolution Climate Projections Using Diffusion-Based Downscaling of a Lightweight Climate Emulator

arXiv:2602. 13416v2 Announce Type: replace Abstract: The proliferation of data-driven models in weather and climate sciences has marked a significant paradigm shift, with advanced models demonstrating exceptional skill in medium-range forecasting.

By Haiwen Guan, Dibyajyoti Chakraborty, Moein Darman, Troy Arcomano, Ashesh Chattopadhyay, Romit Maulik