ML4WX

Topics

Data assimilation

ML for data assimilation and state estimation, learned observation operators, and end-to-end models that forecast directly from observations.

Search and filter this topic Plain list on GitHub

'18
'19
'20
'21
'22
'23
'24
77 papers per year

★ Highlights

★ FuXi Weather: A data-to-forecast machine learning system for global weather

Xiuyu Sun, Xiaohui Zhong, Xiaoze Xu et al. · Nature Communications ·

FuXi Weather is an ML system that assimilates multi-satellite observations and produces 10-day global forecasts, outperforming ECMWF HRES in observation-sparse regions like central Africa.

China Meteorological Administration (CMA)Fudan University (FuXi)Nanjing University of Information Science & Technology (NUIST)Tsinghua University (NowcastNet)University of California, Los Angeles (UCLA)

All papers (77)

★ FuXi Weather: A data-to-forecast machine learning system for global weather

Xiuyu Sun, Xiaohui Zhong, Xiaoze Xu et al. · Nature Communications ·

FuXi Weather is an ML system that assimilates multi-satellite observations and produces 10-day global forecasts, outperforming ECMWF HRES in observation-sparse regions like central Africa.

China Meteorological Administration (CMA)Fudan University (FuXi)Nanjing University of Information Science & Technology (NUIST)Tsinghua University (NowcastNet)University of California, Los Angeles (UCLA)

DiffDA: a Diffusion Model for Weather-scale Data Assimilation

Langwen Huang, Lukas Gianinazzi, Yuejiang Yu et al. · arXiv ·

DiffDA adapts the pretrained GraphCast network as a diffusion backbone to assimilate sparse observations at 0.25 degree, losing at most 24 hours of lead time versus state-of-the-art analyses.

3D-Var Data Assimilation using a Variational Autoencoder

Boštjan Melinc, Žiga Zaplotnik · arXiv ·

VAE-based 3D-Var assimilates simulated temperature observations in a learned latent space, yielding flow-dependent background covariances across tropical and extratropical regions.

Data assimilation in operator algebras

David Freeman, Dimitrios Giannakis, Brian Mintz et al. · Proceedings of the National Academy of Sciences ·

Introduces an operator-algebra framework for sequential data assimilation with kernel-based data-driven schemes, tested on Lorenz 96 and ENSO in a climate model.

A hybrid data assimilation system based on machine learning

Renze Dong, Hongze Leng, Chengwu Zhao et al. · Frontiers in Earth Science ·

Hybrid DA system uses a bilinear network as tangent-linear/adjoint and a CNN to adaptively blend 4DVar and EnKF analyses, tested on Lorenz-96 with lower cost.

Score-based Data Assimilation

François Rozet, Gilles Louppe · Advances in Neural Information Processing Systems 36 ·

Score-based data assimilation learns a generative prior over state trajectories and applies observations only at inference, enabling zero-shot observation scenarios for long dynamical-system trajectories.

Learning 4DVAR inversion directly from observations

Arthur Filoche, Julien Brajard, Anastase Charantonis et al. · arXiv ·

Hybrid network learns 4DVAR-style inversion directly from partial, noisy observations, using the variational mechanistic constraint for regularization and reduced computation.

Auto-differentiable Ensemble Kalman Filters

Yuming Chen, Daniel Sanz-Alonso, Rebecca Willett · arXiv ·

Auto-differentiable ensemble Kalman filters jointly learn surrogate dynamics and assimilate states, outperforming EM and particle-filter baselines on Lorenz-96.

Latent Space Data Assimilation by using Deep Learning

Mathis Peyron, Anthony Fillion, Selime Gürol et al. · Quarterly Journal of the Royal Meteorological Society ·

An autoencoder latent-space ensemble Kalman filter with a neural surrogate reduces data assimilation cost and beats ETKF-Q on an augmented Lorenz 96 system.

NVIDIA

Towards physically consistent data-driven weather forecasting: Integrating data assimilation with equivariance-preserving deep spatial transformers

Ashesh Chattopadhyay, Mustafa Mustafa, Pedram Hassanzadeh et al. · Geoscientific Model Development, 15(5), 2022, 2221-2237 ·

Equivariance-preserving U-STN networks improve data-driven Z500 forecasts by 45% over U-Nets, and coupling them with ensemble Kalman data assimilation gives stable cycles under high observation noise.

Lawrence Berkeley National Laboratory / NERSC

Machine Learning for Model Error Inference and Correction

Massimo Bonavita, Patrick Laloyaux · Journal of Advances in Modeling Earth Systems ·

Neural networks reproduce weak-constraint 4D-Var model error corrections in the operational ECMWF IFS and could extend correction to the whole atmospheric column.

ECMWF

Data Assimilation Networks

Pierre Boudier, Anthony Fillion, Serge Gratton et al. · arXiv ·

Deep recurrent architecture learns data assimilation with approximate prior and posterior densities, matching EnKF performance on the Lorenz-95 toy system with Gaussian errors.

Combining data assimilation and machine learning to infer unresolved scale parametrisation

Julien Brajard, Alberto Carrassi, Marc Bocquet et al. · Philosophical Transactions of the Royal Society A Mathematical Physical and Engineering Sciences ·

Ensemble Kalman filter estimates model error from sparse noisy observations to train a neural parametrisation, yielding hybrid forecasts with better skill on Lorenz and a coupled ocean-atmosphere model.

Lawrence Berkeley National Laboratory / NERSCUniversity of Reading

Learning Variational Data Assimilation Models and Solvers

Ronan Fablet, Bertrand Chapron, Lucas. Drumetz et al. · arXiv ·

End-to-end neural variational data assimilation, with learned model and solver, outperforms classical gradient descent on Lorenz-63 and Lorenz-96 reconstructions.

Also relevant (24)

★ FengWu: Pushing the Skillful Global Medium-range Weather Forecast beyond 10 Days Lead

Kang Chen, Tao Han, Junchao Gong et al. · Commun. Earth Environ. 6, 518 (2025) ·

FengWu is a multi-modal transformer forecasting system that extends skillful forecasts to 10.75 days, beating GraphCast on 80% of 880 predictands.

Global medium-range NWPInstitute of Atmospheric Physics, Chinese Academy of SciencesNanjing University of Information Science & Technology (NUIST)Shanghai Artificial Intelligence Laboratory

Correcting Systematic and State‐Dependent Errors in the NOAA FV3‐GFS Using Neural Networks

Tse‐Chun Chen, Stephen Gregory Penny, Jeffrey S. Whitaker et al. · Journal of Advances in Modeling Earth Systems ·

Neural networks learn FV3-GFS systematic error from data assimilation analysis increments and apply online corrections, improving 10-day forecast skill at all lead times.

Hybrid physics-ML & parameterizationNOAA (incl. GFDL, PSL, GSL, NCEP/EMC)Columbia University (LEAP, Lamont-Doherty)

An Objective Quality Control of Surface Contamination Observations for ABI Water Vapor Radiance Assimilation

Zhenglong Li, Zheng Ma, Pei Wang et al. · Journal of Geophysical Research Atmospheres ·

Machine-learning quality control for ABI water vapor radiances blends seven bands to reject surface-contaminated pixels, improving Hurricane Harvey track forecasts.

Observations & remote sensingNOAA (incl. GFDL, PSL, GSL, NCEP/EMC)Institute of Atmospheric Physics, Chinese Academy of SciencesChina Meteorological Administration (CMA)

★ Earth System Modeling 2.0: A Blueprint for Models That Learn From Observations and Targeted High-Resolution Simulations

Tapio Schneider, Shiwei Lan, Andrew Stuart et al. · Geophysical Research Letters 2017 ·

A blueprint proposes Earth system models that learn parameterizations from observations and high-resolution simulations via data assimilation and machine learning, illustrated on a toy system.

Hybrid physics-ML & parameterizationCalifornia Institute of Technology (Caltech)NASA