Benchmarking Generative Models for Weather Data Assimilation on Real Station Observations
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Weather reanalysis products rely on computationally intensive numerical weather predictions followed by data assimilation that corrects the forecast toward observations. Recent advances in deep generative models offer a cheaper alternative that shifts much of this cost from inference to offline training. However, existing generative approaches have been evaluated on synthetic observations or under different datasets and evaluation schemes, making it unclear which design choices improve real-world data assimilation. We present the first controlled benchmark of generative data assimilation for single-time near-surface analysis from real weather station observations. Using 11,849 NOAA MADIS stations across the contiguous United States and four near-surface variables, we hold the dataset, observation operator, and deep learning architecture fixed, and measure spatial generalization at held-out stations. The benchmark compares the major design choices proposed for generative data assimilation, including diffusion versus flow matching, pixel versus latent-space formulations, and multiple inference-time conditioning strategies, against a classical 3D-Var baseline. The benchmark reveals three conclusions. First, the best generative methods outperform 3D-Var (35.7% vs. 33.3% RMSE improvement over ERA5), although 3D-Var receives the ERA5 field at the analysis time as its background and the generative methods receive none. Second, full-gradient guidance consistently outperforms stop-gradient and initial-noise optimization. Third, other choices provide little measurable benefit: diffusion and flow matching perform nearly identically under matched conditions, and latent-space variable mixing does not help. Both advantages widen when stations are sparse. Together, these results identify which components of generative data assimilation improve spatial generalization in near-surface analysis.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 9 of its 19 references that are in the map. First posted on arXiv 30 Sep 2026.
Builds on
9 of 19 references.
Closest ancestors · 3
Same direction, newest first
- 2025FlowDPS: Flow-Driven Posterior Sampling for Inverse Problems11 Mar
- 2024On conditional diffusion models for PDE simulations21 Oct
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep
Other references · 6
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,375
- 2024Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusioncited by 634
- 2024Deep Generative Data Assimilation in Multimodal Settingcited by 28
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Generative Data Assimilation of Sparse Weather Station Observations at Kilometer Scales
- 2026FlowDA: Accurate, Low-Latency Weather Data Assimilation via Flow Matching
- 2024DiffDA: a Diffusion Model for Weather-scale Data Assimilation
- 2026DAWIS: Data Assimilation with Windowed Inverse Sampling via Multitask Interpolants
- 2026Multimodal Spatiotemporal Atmospheric Data Assimilation with Latent Video Flow-matching
- 2026A Unified Framework for Bayesian Data Assimilation with Generative Models and Observation Interpolants
- 2023Precipitation nowcasting with generative diffusion models
- 2025Appa: Bending Weather Dynamics with Latent Diffusion Models for Global Data Assimilation