Benchmarking Generative Models for Near-Surface Data Assimilation on Real Station Observations
cs.LG, cs.AI
Submitted: 2026-09-30
Updated: 2026-10-07
Project page: https://earth-intelligence-lab.github.io/generative-weather-da
License: http://creativecommons.org/licenses/by/4.0/
The gist: Weather reanalysis products rely on computationally intensive numerical weather predictions followed by data assimilation that corrects the forecast toward observations.
Terminology
Abstract
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.
Sources
- FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operators
- LO-SDA: Latent Optimization for Score-based Atmospheric Data Assimilation
- PnP-DA: Towards Principled Plug-and-Play Integration of Variational Data Assimilation and Generative Models
- DiffDA: a Diffusion Model for Weather-scale Data Assimilation
- Appa: Bending Weather Dynamics with Latent Diffusion Models for Global Data Assimilation
- FlowDA: Accurate, Low-Latency Weather Data Assimilation via Flow Matching
- Generative assimilation and prediction for weather and climate
- Benchmarking AI-based data assimilation to advance data-driven global weather forecasting
- DAMBench: A Multi-Modal Benchmark for Deep Learning-based Atmospheric Data Assimilation
- Control-Augmented Autoregressive Diffusion for Data Assimilation
- Variational Control for Guidance in Diffusion Models
- Diffusion Model Based Posterior Sampling for Noisy Linear Inverse Problems
- D-Flow: Differentiating through Flows for Controlled Generation
- Partial recovery of meter-scale surface weather
- Fast Sampling of Diffusion Models with Exponential Integrator
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks