Accessibility navigation


Ensemble Kalman filter in latent space using a variational autoencoder pair

Pasmans, I. ORCID: https://orcid.org/0000-0001-5076-5421, Chen, Y. ORCID: https://orcid.org/0000-0002-2319-6937, Finn, T., Bocquet, M. and Carrassi, A. ORCID: https://orcid.org/0000-0003-0722-5600 (2025) Ensemble Kalman filter in latent space using a variational autoencoder pair. Quarterly Journal of the Royal Meteorological Society. ISSN 1477-870X (In Press)

[thumbnail of pasmans_vae_da_clean.pdf] Text - Accepted Version
· Restricted to Repository staff only
· The Copyright of this document has not been checked yet. This may affect its availability.
· Available under License Creative Commons Attribution.

3MB

It is advisable to refer to the publisher's version if you intend to cite from this work. See Guidance on citing.

Abstract/Summary

Popular (ensemble) Kalman filter data assimilation (DA) approaches assume that the errors in both the a priori estimate of the state and those in the observations are Gaussian. For constrained variables, e.g. sea ice concentration or stress, such an assumption does not hold. The variational autoencoder (VAE) is a machine learning (ML) technique that allows to map an arbitrary distribution to/from a latent space in which the distribution is supposedly closer to a Gaussian. We propose a novel hybrid DA-ML approach in which VAEs are incorporated in the DA procedure. Specifically, we introduce a variant of the popular ensemble transform Kalman filter (ETKF) in which the analysis is applied in the latent space of a single VAE or a pair of VAEs. In twin experiments with a simple circular model, whereby the circle represents an underlying submanifold to be respected, we find that the use of a VAE ensures that a posteriori ensemble members lie close to the manifold containing the truth. Furthermore, online updating of the VAE is necessary and achievable when this manifold varies in time, i.e. when it is non-stationary. We demonstrate that introducing an additional second latent space for the observational innovations improves robustness against detrimental effects of non-Gaussianity and bias in the observational errors but it slightly lessens the performance if observational errors are strictly Gaussian.

Item Type:Article
Refereed:Yes
Divisions:Science > School of Mathematical, Physical and Computational Sciences > National Centre for Earth Observation (NCEO)
Science > School of Mathematical, Physical and Computational Sciences > Department of Meteorology
ID Code:125425
Publisher:Royal Meteorological Society

University Staff: Request a correction | Centaur Editors: Update this record

Page navigation