Feature-preserving Latent-EnKF for Data Assimilation of Flows with Shocks
Abstract
The ensemble Kalman filter (EnKF) is widely adopted for sequential data assimilation, but fails for solutions with discontinuities, such as shocks in compressible flows. For ensembles with uncertainty in discontinuity location, the EnKF analysis can lose physical consistency, which can be viewed through three lenses: (i) bimodal pointwise statistics in the forecast ensemble, (ii) mixed-sign analysis weights in the affine superposition of forecast members, and (iii) spatially distributed Kalman-gain profiles that modify the state around a discontinuity rather than translating it coherently. In order to preserve flow features in the analysis state and, as such, maintain physical consistency, a feature-preserving latent-EnKF is introduced. The proposed method performs the ensemble update in a low-dimensional latent space learned by a coordinate-conditioned auto-decoder. The latent representation retains near-Gaussian ensemble statistics and provides coordinates in which flow features vary coherently across the forecast ensemble. The EnKF updates performed on the latent codes are decoded into coherent changes in the represented flow features, translating shocks through coordinate morphing rather than superposing them at displaced locations. The corresponding Kalman-gain profiles generated by the latent-EnKF exhibit an approximately step-like variation across the discontinuities, allowing the fronts to translate while preserving sharp features and monotonicity. Sod shock-tube, Mach~2 shock-cylinder, and immersed-body shape-inference tests with sparse, noisy observations demonstrate accurate recovery of shocks and contact discontinuities without spurious oscillations.
Related
- Periodic dynamics of transonic shocks in Euler-Poisson flows
- Feature tracking in physics-informed neural networks via joint optimization of nonlinear deformation manifolds: application to shocks
- Preserving Structure in Variational Data Assimilation of Hyperbolic Conservation Laws via Regularization
- Anisotropic-Length-Preserving Weighted Anisotropic Curvature Flows
- Curvature measures and volume preserving flows in h-convex geometry
BibTeX
@misc{chandravamsi2026featurepreserving,
title = {{Feature-preserving Latent-EnKF for Data Assimilation of Flows with Shocks}},
author = {Hemanth Chandravamsi and Hangchuan Hu and Ponkrshnan Thiagarajan and Tamer A. Zaki},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2606.12559},
eprint = {2606.12559},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2606.12559},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - Feature-preserving Latent-EnKF for Data Assimilation of Flows with Shocks
AU - Hemanth Chandravamsi
AU - Hangchuan Hu
AU - Ponkrshnan Thiagarajan
AU - Tamer A. Zaki
PY - 2026
DA - 2026/10/14
PB - arXiv
DO - 10.48550/arxiv.2606.12559
UR - https://arxiv.org/abs/2606.12559
AB - The ensemble Kalman filter (EnKF) is widely adopted for sequential data assimilation, but fails for solutions with discontinuities, such as shocks in compressible flows. For ensembles with uncertainty in discontinuity location, the EnKF analysis can lose physical consistency, which can be viewed through three lenses: (i) bimodal pointwise statistics in the forecast ensemble, (ii) mixed-sign analysis weights in the affine superposition of forecast members, and (iii) spatially distributed Kalman-gain profiles that modify the state around a discontinuity rather than translating it coherently. In order to preserve flow features in the analysis state and, as such, maintain physical consistency, a feature-preserving latent-EnKF is introduced. The proposed method performs the ensemble update in a low-dimensional latent space learned by a coordinate-conditioned auto-decoder. The latent representation retains near-Gaussian ensemble statistics and provides coordinates in which flow features vary coherently across the forecast ensemble. The EnKF updates performed on the latent codes are decoded into coherent changes in the represented flow features, translating shocks through coordinate morphing rather than superposing them at displaced locations. The corresponding Kalman-gain profiles generated by the latent-EnKF exhibit an approximately step-like variation across the discontinuities, allowing the fronts to translate while preserving sharp features and monotonicity. Sod shock-tube, Mach~2 shock-cylinder, and immersed-body shape-inference tests with sparse, noisy observations demonstrate accurate recovery of shocks and contact discontinuities without spurious oscillations.
ER -