Neural ensemble Kalman filter: Data assimilation for compressible flows with shocks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhou, Xu-Hui, Beronilla, Lorenzo, Sleeman, Michael K., Hu, Hangchuan, Morzfeld, Matthias, Stuart, Andrew M., Zaki, Tamer A.
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908855242326016
author Zhou, Xu-Hui
Beronilla, Lorenzo
Sleeman, Michael K.
Hu, Hangchuan
Morzfeld, Matthias
Stuart, Andrew M.
Zaki, Tamer A.
author_facet Zhou, Xu-Hui
Beronilla, Lorenzo
Sleeman, Michael K.
Hu, Hangchuan
Morzfeld, Matthias
Stuart, Andrew M.
Zaki, Tamer A.
contents Data assimilation (DA) for compressible flows with shocks is challenging because many classical DA methods generate spurious oscillations and nonphysical features near uncertain shocks. We focus here on the ensemble Kalman filter (EnKF). We show that the poor performance of the standard EnKF may be attributed to the bimodal forecast distribution that can arise in the vicinity of an uncertain shock location; this violates the assumptions underpinning the EnKF, which assume a forecast which is close to Gaussian. To address this issue we introduce the new neural EnKF. The basic idea is to systematically embed neural function approximations within ensemble DA by mapping the forecast ensemble of shocked flows to the parameter space (weights and biases) of a deep neural network (NN) and to subsequently perform DA in that space. The nonlinear mapping encodes sharp and smooth flow features in an ensemble of NN parameters. Neural EnKF updates are therefore well-behaved only if the NN parameters vary smoothly within the neural representation of the forecast ensemble. We show that such a smooth variation of network parameters can be enforced via physics-informed transfer learning, and demonstrate that in so-doing the neural EnKF avoids the spurious oscillations and nonphysical features that plague the standard EnKF. The applicability of the neural EnKF is demonstrated through a series of systematic numerical experiments with an inviscid Burgers' equation, Sod's shock tube, and a two-dimensional blast wave.
format Preprint
id arxiv_https___arxiv_org_abs_2602_23461
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Neural ensemble Kalman filter: Data assimilation for compressible flows with shocks
Zhou, Xu-Hui
Beronilla, Lorenzo
Sleeman, Michael K.
Hu, Hangchuan
Morzfeld, Matthias
Stuart, Andrew M.
Zaki, Tamer A.
Fluid Dynamics
Machine Learning
Data assimilation (DA) for compressible flows with shocks is challenging because many classical DA methods generate spurious oscillations and nonphysical features near uncertain shocks. We focus here on the ensemble Kalman filter (EnKF). We show that the poor performance of the standard EnKF may be attributed to the bimodal forecast distribution that can arise in the vicinity of an uncertain shock location; this violates the assumptions underpinning the EnKF, which assume a forecast which is close to Gaussian. To address this issue we introduce the new neural EnKF. The basic idea is to systematically embed neural function approximations within ensemble DA by mapping the forecast ensemble of shocked flows to the parameter space (weights and biases) of a deep neural network (NN) and to subsequently perform DA in that space. The nonlinear mapping encodes sharp and smooth flow features in an ensemble of NN parameters. Neural EnKF updates are therefore well-behaved only if the NN parameters vary smoothly within the neural representation of the forecast ensemble. We show that such a smooth variation of network parameters can be enforced via physics-informed transfer learning, and demonstrate that in so-doing the neural EnKF avoids the spurious oscillations and nonphysical features that plague the standard EnKF. The applicability of the neural EnKF is demonstrated through a series of systematic numerical experiments with an inviscid Burgers' equation, Sod's shock tube, and a two-dimensional blast wave.
title Neural ensemble Kalman filter: Data assimilation for compressible flows with shocks
topic Fluid Dynamics
Machine Learning
url https://arxiv.org/abs/2602.23461