ML4WX

Data assimilation

Learning Optimal Filters Using Variational Inference

Eviatar Bach, Ricardo Baptista, Enoch Luk, Andrew Stuart

arXiv preprint · · California Institute of Technology (Caltech), University of Reading

PaperPDF

In one sentence. Learns the analysis map of filters such as the ensemble Kalman filter via variational inference, reducing bias in gain, inflation and localization estimates for nonlinear systems.

Abstract

Filtering - the task of estimating the conditional distribution for states of a dynamical system given partial and noisy observations - is important in many areas of science and engineering, including weather and climate prediction. However, the filtering distribution is generally intractable to obtain for high-dimensional, nonlinear systems. Filters used in practice, such as the ensemble Kalman filter (EnKF), provide biased probabilistic estimates for nonlinear systems and have numerous tuning parameters. Here, we present a framework for learning a parameterized analysis map - the transformation that takes samples from a forecast distribution, and combines with an observation, to update the approximate filtering distribution - using variational inference. In principle this can lead to a better approximation of the filtering distribution, and hence smaller bias. We show that this methodology can be used to learn the gain matrix, in an affine analysis map, for filtering linear and nonlinear dynamical systems; we also study the learning of inflation and localization parameters for an EnKF. The framework developed here can also be used to learn new filtering algorithms with more general forms for the analysis map.

Labels

Topic
Data assimilation
Type
New model or method
Method
Classical ML (trees, GP, linear, kernels)
Data
Other
Status
Research
Identifiers
arXiv:2406.18066 OpenAlex:W4400103783
Added
2026-10-22

Labels and summary are generated by Claude and reviewed weekly. Wrong? Edit this entry.