Variational Parameter Calibration with Physics-Aware Latent-Space Surrogates

arXiv cs.LG Papers

Summary

This paper introduces a physics-aware autoencoder-based latent-space framework for reduced-order forward modeling and variational parameter estimation in parametric dynamical systems, demonstrated on computational fluid dynamics benchmarks. The method enables differentiable surrogate-based inverse modeling and shows improved calibration robustness under realistic noisy or partial observations.

arXiv:2608.11435v1 Announce Type: new Abstract: Forward and inverse modeling of parametric dynamical systems requires surrogate models that are not only accurate for state prediction, but also informative for parameter calibration. However, a systematic end-to-end differentiable formulation for coupling deep-learning-based reduced-order surrogates with variational parameter estimation remains underdeveloped. In this work, we introduce a physics-aware neural-network-based latent-space framework for reduced-order forward modeling and variational parameter estimation. The proposed autoencoder-based approach yields a differentiable surrogate that maps physical parameters to predicted flow fields through a latent representation. The observable supervision is used during offline training to encourage the latent variables to retain information correlated with system parameters, while the online inverse problem is solved in the parameter space through the surrogate-induced observation operator. The method is evaluated on two computational-fluid-dynamics benchmarks. The results show that reconstruction accuracy alone is insufficient for inverse modeling, owing to the lack of end-to-end differentiability or physics awareness for variational parameter calibration. Quantitative latent-space analysis further shows that observable supervision improves case-level separability and temporal organization of latent representations. Experiments with realistic measurement settings, including noisy, low-resolution, randomly masked, and block-wise partial observations, demonstrate the robustness of the proposed framework and show that it generally reduces calibration error and variability compared with the standard surrogate models.
Original Article
View Cached Full Text

Cached at: 08/13/26, 03:36 PM

# Variational Parameter Calibration with Physics-Aware Latent-Space Surrogates
Source: [https://arxiv.org/html/2608.11435](https://arxiv.org/html/2608.11435)
Qiyao Zhou1, Xujia Zhu2, Pierre Joli1, Yu Cong1, Sibo Cheng3,∗Address:1LMEE, UnivEvry, Université Paris\-Saclay, Evry, France 2L2S, Université Paris\-Saclay, CNRS, CentraleSupéléc, Gif\-sur\-Yvette, France 3CEREA, ENPC, EDF R&D, Institut Polytechnique de Paris, Île\-de\-France, France ∗corresponding: sibo\.cheng@enpc\.fr

###### Abstract

Forward and inverse modeling of parametric dynamical systems requires surrogate models that are not only accurate for state prediction, but also informative for parameter calibration\. However, a systematic end\-to\-end differentiable formulation for coupling deep\-learning\-based reduced\-order surrogates with variational parameter estimation remains underdeveloped\. In this work, we introduce a physics\-aware neural\-network\-based latent\-space framework for reduced\-order forward modeling and variational parameter estimation\. The proposed autoencoder\-based approach yields a differentiable surrogate that maps physical parameters to predicted flow fields through a latent representation\. The observable supervision is used during offline training to encourage the latent variables to retain information correlated with system parameters, while the online inverse problem is solved in the parameter space through the surrogate\-induced observation operator\. The method is evaluated on two computational\-fluid\-dynamics benchmarks\. The results show that reconstruction accuracy alone is insufficient for inverse modeling, owing to the lack of end\-to\-end differentiability or physics awareness for variational parameter calibration\. Quantitative latent\-space analysis further shows that observable supervision improves case\-level separability and temporal organization of latent representations\. Experiments with realistic measurement settings, including noisy, low\-resolution, randomly masked, and block\-wise partial observations, demonstrate the robustness of the proposed framework and show that it generally reduces calibration error and variability compared with the standard surrogate models\.

###### Keywords:

Deep learning , Data assimilation , Reduced\-order modeling , Parameter calibration , Computational fluid dynamics

## Main Notations

𝒳\\mathcal\{X\}Snapshot matrix𝐱t\\mathbf\{x\}\_\{t\}Flattened state vector at timett𝐂𝒳\\mathbf\{C\}\_\{\\mathcal\{X\}\}Empirical covariance matrix of snapshotsΨj\\Psi\_\{j\}jj\-th POD basisqqDimension of latent space𝐋𝒳,q\\mathbf\{L\}\_\{\\mathcal\{X\},q\}POD projection matrix withqqretained modes𝐱^t\\hat\{\\mathbf\{x\}\}\_\{t\}POD coefficient vector at timettkj,tk\_\{j,t\}jj\-th POD coefficient at timett𝜽\\boldsymbol\{\\theta\}Parameter vectorΩ𝜽\\Omega\_\{\\boldsymbol\{\\theta\}\}Parameter spacekj​\(𝜽\)k\_\{j\}\(\\boldsymbol\{\\theta\}\)jj\-th POD coefficient with parameter𝜽\\boldsymbol\{\\theta\}𝐳\\mathbf\{z\}Latent vector in CAE𝐳r\\mathbf\{z\}^\{r\}Latent vector in OACAEℱe,ℱd\\mathcal\{F\}\_\{e\},\\mathcal\{F\}\_\{d\}Encoder and decoder of CAEℱer,ℱdr\\mathcal\{F\}\_\{e\}^\{r\},\\mathcal\{F\}\_\{d\}^\{r\}Encoder and decoder of OACAEℰa\\mathcal\{E\}\_\{a\}Observable regression MLP branchℰb\\mathcal\{E\}\_\{b\}Parameter\-to\-latent regressor MLP in OACAE\-MLPℰc\\mathcal\{E\}\_\{c\}Parameter\-to\-latent regressor MLP in CAE\-MLPϕe,ϕd,ϕa,ϕb,ϕc\\phi\_\{e\},\\phi\_\{d\},\\phi\_\{a\},\\phi\_\{b\},\\phi\_\{c\}Trainable weights of encoders/decoders/regressorsℒ1,ℒ2\\mathcal\{L\}\_\{1\},\\mathcal\{L\}\_\{2\}CAE\-MLP training loss functions \(reconstruction/latent regression\)ℒ1r,ℒ2r\\mathcal\{L\}\_\{1\}^\{r\},\\mathcal\{L\}\_\{2\}^\{r\}OACAE\-MLP training loss functions \(reconstruction/latent regression\)α,β\\alpha,\\betaWeighting factors balancing loss termsNeN\_\{e\}Ensemble size in EnKF𝜽b,𝜽a\\boldsymbol\{\\theta\}^\{b\},\\boldsymbol\{\\theta\}^\{a\}Background and analysis parameter vectors𝐁\\mathbf\{B\}Background error covariance matrix𝐑\\mathbf\{R\}Observation error covariance matrix𝐊∗\\mathbf\{K\}^\{\*\}Kalman gain in EnKFℋPOD\-GPR\\mathcal\{H\}\_\{\\text\{POD\-GPR\}\}Observation operator with POD\-GPR surrogateℋCAE\-MLP\\mathcal\{H\}\_\{\\text\{CAE\-MLP\}\}Observation operator with CAE\-MLP surrogateJ3D\-Var,J4D\-VarJ\_\{\\textbf\{3D\-Var\}\},J\_\{\\textbf\{4D\-Var\}\}3D\-Var/4D\-Var cost functions
## 1Introduction

### 1\.1Parametric systems

Many physical and engineering problems can be described by Partial Differential Equations \(PDEs\) whose solutions depend on a set of parameters, such as material properties, boundary or initial conditions, and geometrical configurations\. Such parametric systems provide a flexible and realistic representation of physical processes, allowing researchers and engineers to explore system behavior across a wide range of configurations\. At the same time, the resulting solution manifolds are often high\-dimensional and may exhibit strong nonlinear variability with respect to the parameters, making their analysis computationally demanding\.

A solution field of physical quantities, such as temperature, displacement, velocity, or pressure, is typically expressed by

𝐱:𝒯×Ω×𝒫→ℝm,\\mathbf\{x\}:\\mathcal\{T\}\\times\\Omega\\times\\mathcal\{P\}\\;\\to\\;\\mathbb\{R\}^\{m\},where they depend on timet∈𝒯⊂ℝt\\in\\mathcal\{T\}\\subset\\mathbb\{R\}, spatial coordinatess∈Ω⊂ℝds\\in\\Omega\\subset\\mathbb\{R\}^\{d\}, and system parameters𝐩∈𝒫⊂ℝd𝐩\\mathbf\{p\}\\in\\mathcal\{P\}\\subset\\mathbb\{R\}^\{d\_\{\\mathbf\{p\}\}\}\. The latter may encode initial or boundary conditions, material properties, or other system characteristics, and the solution field is constrained by the governing PDE\.

The study of parametric systems is motivated by a wide range of applications\. In materials engineering, parametric descriptions of microstructure or elastic coefficients are essential for predicting macroscopic responses under load\[[3](https://arxiv.org/html/2608.11435#bib.bib1)\]\. In wildfire propagation models, wind conditions and vegetation parameters strongly affect the fire front dynamics\[[8](https://arxiv.org/html/2608.11435#bib.bib4)\]\. In fluid dynamics, different Reynolds numbers or inflow profiles generate distinct vortex shedding and turbulence structures\[[23](https://arxiv.org/html/2608.11435#bib.bib2),[55](https://arxiv.org/html/2608.11435#bib.bib3)\]\. These examples illustrate the complexity and sensitivity of parametric PDE solutions, motivating the development of efficient modeling and analysis tools\.

Among parametric PDE systems, Computational Fluid Dynamics \(CFD\) represents one of the most prominent and challenging application areas, with a wide range of applications in engineering and geophysical sciences\. The governing equations of fluid dynamics are the Navier\-Stokes equations, which express conservation of mass and momentum\. For a Newtonian fluid with time\-independent density, and neglecting body forces, a simplified form of these equations can be written as

∇⋅\(ρ​𝐮\)\\displaystyle\\nabla\\cdot\(\\rho\\mathbf\{u\}\)=0,\\displaystyle=0,\(1\)∂∂t​\(ρ​𝐮\)\+∇⋅\(ρ​𝐮𝐮\)\\displaystyle\\frac\{\\partial\}\{\\partial t\}\(\\rho\\mathbf\{u\}\)\+\\nabla\\cdot\(\\rho\\mathbf\{u\}\\mathbf\{u\}\)=−∇p\+∇⋅μ⁡\[∇𝐮\+\(∇𝐮\)⊤\],\\displaystyle=\-\\nabla p\+\\nabla\\cdot\\mu\\big\[\\nabla\\mathbf\{u\}\+\(\\nabla\\mathbf\{u\}\)^\{\\top\}\\big\],whereρ\\rhodenotes the fluid density,𝐮=\(u,v\)⊤\\mathbf\{u\}=\(u,v\)^\{\\top\}the velocity field,ppthe pressure, andμ\\muthe viscosity\. Under incompressibility and Newtonian assumptions, these equations form the classical foundation of fluid mechanics\[[4](https://arxiv.org/html/2608.11435#bib.bib5),[33](https://arxiv.org/html/2608.11435#bib.bib6)\]\. Boundary conditions, material properties, and geometrical configurations define a high\-dimensional parameter space that strongly shapes the resulting flow dynamics, leading to transitions between laminar and turbulent regimes, vortex shedding, and highly unsteady behavior\. The numerical solution of the Navier\-Stokes equations relies on high\-fidelity discretization techniques such as finite difference, finite volume, finite element, or spectral methods\. While these approaches can deliver accurate solutions, they are computationally expensive, particularly for complex geometries, long\-time integrations, or high\-dimensional parameter studies\.

### 1\.2Reduced order modeling

Compared with full\-order modeling approaches\[[23](https://arxiv.org/html/2608.11435#bib.bib2)\]for high\-fidelity computational fluid dynamics simulations, reduced\-order modeling aims to reduce the computational cost of repeated simulations while preserving the dominant structures of the underlying dynamics\. Most reduced\-order models are constructed from a set of high\-fidelity solutions, usually referred to as snapshots, and seek a low\-dimensional representation of the solution manifold\.

Projection\-based methods constitute a classical class of reduced\-order modeling techniques\. Proper orthogonal decomposition \(POD\)\[[16](https://arxiv.org/html/2608.11435#bib.bib12)\]and dynamic mode decomposition \(DMD\)\[[53](https://arxiv.org/html/2608.11435#bib.bib13),[38](https://arxiv.org/html/2608.11435#bib.bib45)\]are representative examples that construct linear reduced spaces from snapshot data\. These methods are efficient, interpretable, and often effective when the dominant flow variability can be captured by a limited number of linear modes\. However, their reduced coordinates are mainly optimized according to reconstruction criteria\. As a result, they are not necessarily organized with respect to the physical parameters that would typically be inferred in an inverse problem\. To enable non\-intrusive prediction for new parameter values, projection\-based models are often combined with regression techniques, e\.g\., Gaussian process regression \(GPR\) and kernel ridge regression \(KRR\)\[[56](https://arxiv.org/html/2608.11435#bib.bib14)\], to learn mappings from the parameter space to reduced coordinates\. This would result in hybrid reduced\-order surrogates\. Typicaly examples include POD\-GPR\[[40](https://arxiv.org/html/2608.11435#bib.bib29)\]and POD\-LSTM\[[46](https://arxiv.org/html/2608.11435#bib.bib21)\]\. These approaches can provide efficient parameter\-to\-state prediction without repeatedly solving the full\-order model\. Nevertheless, when the reduced representation is inherited from a reconstruction\-oriented projection, the learned surrogate may still fail to preserve parameter\-sensitive directions that are important for calibration\.

Deep\-learning\-based reduced\-order models provide a nonlinear alternative to projection\-based approaches\. Convolutional autoencoders, multilayer perceptrons, and recurrent neural\-network architectures have been widely used to construct nonlinear latent representations and parametric surrogates for time\-dependent physical systems\[[47](https://arxiv.org/html/2608.11435#bib.bib18),[7](https://arxiv.org/html/2608.11435#bib.bib19),[58](https://arxiv.org/html/2608.11435#bib.bib20)\]\. In particular, autoencoder \(AE\)\-based models can approximate nonlinear solution manifolds more flexibly than linear reduced bases\. However, a standard autoencoder is usually trained only by minimizing the reconstruction error\. Consequently, same as projection\-based approaches, the latent variables are not explicitly constrained to be correlated with physical parameters, time, or other informations\. This can lead to latent representations that are accurate for reconstruction but less informative for parameter identification\.

Recent developments have explored the incorporation of physical information or observable constraints into latent representations\[[24](https://arxiv.org/html/2608.11435#bib.bib23),[59](https://arxiv.org/html/2608.11435#bib.bib24),[25](https://arxiv.org/html/2608.11435#bib.bib25)\]\. By embedding physical parameters, or other system information into the training process, physics\-aware autoencoder architectures can produce more structured latent spaces\. This idea is particularly relevant for inverse modeling, because parameter calibration requires a surrogate that is not only accurate in the physical space, but also informative to the parameters being estimated\. This motivates the observable\-augmented autoencoder framework introduced in this work\.

### 1\.3Inverse problem: Parameter calibration using data assimilation

While reduced\-order models are often developed for forward prediction, many practical applications require solving the inverse problem of estimating unknown system parameters from observations of the system dynamics\. This parameter\-calibration task is essential for improving model fidelity and predictive capability, but it is generally ill\-posed because different parameter values may generate similar observed responses\[[5](https://arxiv.org/html/2608.11435#bib.bib54),[1](https://arxiv.org/html/2608.11435#bib.bib55)\], especially when the available observations are noisy, sparse, or incomplete\. Data assimilation provides a principled framework for addressing this problem by combining prior parameter information with observational data in a statistically consistent manner\.

Among existing data\-assimilation methodologies, sequential filtering approaches, such as the ensemble Kalman filter, and variational formulations \(including three\- and four\-dimensional variational assimilation\), have been widely used for parameter estimation while accounting for uncertainties in the background and observations\[[14](https://arxiv.org/html/2608.11435#bib.bib41),[6](https://arxiv.org/html/2608.11435#bib.bib40),[2](https://arxiv.org/html/2608.11435#bib.bib39),[9](https://arxiv.org/html/2608.11435#bib.bib38),[18](https://arxiv.org/html/2608.11435#bib.bib42),[37](https://arxiv.org/html/2608.11435#bib.bib46)\]\. In surrogate\-based inverse modeling, a reduced\-order model is embedded into the assimilation procedure as a fast forward or observation operator\[[10](https://arxiv.org/html/2608.11435#bib.bib60),[15](https://arxiv.org/html/2608.11435#bib.bib61)\]\. This strategy avoids repeated full\-order simulations and can substantially reduce the online computational cost of parameter calibration\. Existing reduced\-order data\-assimilation frameworks can be broadly grouped by the surrogate type and the assimilation strategy\. Projection\-based models, such as POD–Galerkin models, have been coupled with variational assimilation for parameter estimation in fluid and heat\-transfer problems\[[16](https://arxiv.org/html/2608.11435#bib.bib12),[39](https://arxiv.org/html/2608.11435#bib.bib49)\]\. However, their reduced coordinates are usually obtained from projection and reconstruction criteria and are not explicitly organized by the physical information\. Regression\-based non\-intrusive surrogates, such as POD\-GPR, have also been used with ensemble\-based assimilation for large\-scale flow and pollutant\-dispersion problems\[[40](https://arxiv.org/html/2608.11435#bib.bib29),[41](https://arxiv.org/html/2608.11435#bib.bib28)\]\. Although such surrogates are compatible with ensemble filtering through repeated forward evaluations, their online cost can grow with the ensemble size and observation window, and their reduced representations are not necessarily parameter\-identifiable and physics\-consistent\. Ensemble\-based assimilation can rely on repeated surrogate evaluations, whereas variational assimilation requires gradients of the cost function with respect to the control parameters\. Therefore, the present framework uses an end\-to\-end differentiable parameter\-to\-observation surrogate, enabling efficient optimization of the three\- and four\-dimensional variational objectives by automatic differentiation\.

Therefore, two requirements must be considered simultaneously for deep\-learning\-based surrogate data assimilation\. First, the surrogate should be end\-to\-end differentiable with respect to the physical control parameters, allowing efficient gradient\-based optimization in variational assimilation\. Second, its latent representation should preserve information that is relevant to the parameters being calibrated, rather than being optimized only for field reconstruction\. Existing projection\-based, regression\-based, and standard autoencoder\-based reduced\-order data\-assimilation frameworks do not fully address these two requirements at the same time\. This motivates the physics\-aware neural\-network based latent\-space variational framework developed in this work, in which observable supervision is used to organize the latent representation and a differentiable parameter\-to\-latent\-to\-prediction\-to\-observation surrogate is used as the observation operator for parameter calibration\.

### 1\.4Main contributions

Motivated by the above limitations, this work develops a physics\-aware latent\-space framework for reduced\-order forward modeling and variational parameter calibration\. The main contributions of this study are summarized as follows:

- 1\.We formulate an end\-to\-end differentiable reduced\-order data\-assimilation framework for inverse modeling, in which a deep\-learning\-based surrogate serves as the observation operator for three\- and four\-dimensional variational parameter estimation\.
- 2\.We introduce a physics\-aware latent\-space surrogate into variational data assimilation by leveraging observable\-augmented autoencoding, which improves online calibration accuracy and reduces variability in the physical control\-parameter space\.
- 3\.We validate the proposed framework on CFD benchmarks by assessing its robustness under degraded\-observation settings and by conducting systematic sensitivity analyses of the forward\-modeling and inverse\-calibration settings, in comparison with standard AE\-variational DA and POD\-GPR\-ensemble DA baselines\.

## 2Methodology

### 2\.1Problem setup

In this work, we consider parametrized, time\-dependent flow systems whose dynamics are represented by high\-dimensional state variables evolving on a spatial grid\. The primary objective of this work is to construct efficient reduced\-order surrogate models capable of approximating the full\-order system dynamics and to embed these surrogates into DA frameworks for parameter calibration\.

Let𝐱t∈ℝNc×Nx×Ny\\mathbf\{x\}\_\{t\}\\in\\mathbb\{R\}^\{N\_\{c\}\\times N\_\{x\}\\times N\_\{y\}\}denote the discrete solution fields of the system at timett, whereNcN\_\{c\}is the number of physical variables \(two in this work, corresponding to the horizontal and vertical velocity components\) and\(Nx,Ny\)\(N\_\{x\},N\_\{y\}\)define the spatial grid dimensions\. The datasets considered in this work consist of multiple simulation cases, each corresponding to a distinct configuration of physical parameters and boundary conditions\. For each case, a sequence of time\-dependent snapshots\{𝐱t\}t=0nstate−1\\\{\\mathbf\{x\}\_\{t\}\\\}\_\{t=0\}^\{n\_\{\\text\{state\}\}\-1\}is collected together with the associated parameter values and temporal indices, wherenstaten\_\{\\text\{state\}\}denotes the number of snapshots\. Each snapshot can therefore be viewed as a data pair

\(𝐱t,𝜽\),\\left\(\\mathbf\{x\}\_\{t\},\\,\\boldsymbol\{\\theta\}\\right\),where𝜽∈ℝdθ\\boldsymbol\{\\theta\}\\in\\mathbb\{R\}^\{d\_\{\\theta\}\}gathers the system parameters and the temporal index\. From these data, we consider three closely related modeling tasks\. First, in the forward modeling and reconstruction setting, the surrogate model takes high\-dimensional states𝐱t\\mathbf\{x\}\_\{t\}as input and outputs their reconstructed counterparts, with the objective of learning a compact latent representation that preserves the original flow fields\. Second, in the forward\-prediction setting, the surrogate model aims to approximate the parametric mapping from system parameters to flow states, taking𝜽\\boldsymbol\{\\theta\}as input and producing the corresponding predicted states𝐱t\\mathbf\{x\}\_\{t\}\. Finally, in the inverse problem addressed by DA, the objective is to infer calibrated system parameters𝜽\\boldsymbol\{\\theta\}from observed flow states𝐱t\\mathbf\{x\}\_\{t\}, which may be available at a single time instant or at multiple time instants\. In this setting, the forward surrogate models efficiently provide the state predictions required in the assimilation process\.

### 2\.2Reduced\-order modeling approaches

#### 2\.2\.1POD\-GPR

Due to the high\-dimensional nature of𝐱t\\mathbf\{x\}\_\{t\}, constructing Gaussian process surrogate models directly in the physical space is hindered by the*curse of dimensionality*, stemming from the non\-parametric nature of this method\. As a result, it is typically combined with dimensionality reduction techniques, such as POD\[[40](https://arxiv.org/html/2608.11435#bib.bib29)\]\. Following this idea, we first employ POD to construct a reduced linear subspace from a collection of snapshots\. We assemble the snapshot matrix

𝒳=\[𝐱0,𝐱1,…,𝐱nstate−1\]∈ℝ\(Nc×Nx×Ny\)×nstate,\\mathcal\{X\}=\\big\[\\,\\mathbf\{x\}\_\{0\},\\mathbf\{x\}\_\{1\},\\ldots,\\mathbf\{x\}\_\{n\_\{\\text\{state\}\}\-1\}\\,\\big\]\\;\\in\\;\\mathbb\{R\}^\{\(N\_\{c\}\\times N\_\{x\}\\times N\_\{y\}\)\\times n\_\{\\text\{state\}\}\},\(2\)where each column represents a flattened state vector at a given time step\. The empirical covariance matrix is then given by

𝐂𝒳=1nstate−1​𝒳​𝒳T,\\mathbf\{C\}\_\{\\mathcal\{X\}\}=\\frac\{1\}\{n\_\{\\text\{state\}\}\-1\}\\,\\mathcal\{X\}\\,\\mathcal\{X\}^\{T\},\(3\)which admits the eigen\-decomposition

𝐂𝒳=𝐋𝒳​𝐃𝒳​𝐋𝒳T,\\mathbf\{C\}\_\{\\mathcal\{X\}\}=\\mathbf\{L\}\_\{\\mathcal\{X\}\}\\,\\mathbf\{D\}\_\{\\mathcal\{X\}\}\\,\\mathbf\{L\}\_\{\\mathcal\{X\}\}^\{T\},\(4\)where the columns of𝐋𝒳=\[Ψ0,…,Ψnstate−1\]∈ℝ\(Nc×Nx×Ny\)×ns​t​a​t​e\\mathbf\{L\}\_\{\\mathcal\{X\}\}=\[\\,\\Psi\_\{0\},\\ldots,\\Psi\_\{n\_\{\\text\{state\}\}\-1\}\\,\]\\in\\mathbb\{R\}^\{\(N\_\{c\}\\times N\_\{x\}\\times N\_\{y\}\)\\times n\_\{state\}\}are the POD modes, and𝐃𝒳=diag⁡\(λ0,…,λnstate−1\)\\mathbf\{D\}\_\{\\mathcal\{X\}\}=\\mathrm\{diag\}\(\\lambda\_\{0\},\\ldots,\\lambda\_\{n\_\{\\text\{state\}\}\-1\}\)contains the eigenvalues in decreasing order\.

To obtain a reduced subspace of dimensionqq\(q∈ℕ\+,q≪nstateq\\in\\mathbb\{N\}^\{\+\},q\\ll n\_\{\\text\{state\}\}\), we retain the firstqqPOD modes, forming the projection matrix

𝐋𝒳,q=\[Ψ0,…,Ψq−1\]∈ℝ\(Nc×Nx×Ny\)×q\.\\mathbf\{L\}\_\{\\mathcal\{X\},q\}=\[\\,\\Psi\_\{0\},\\ldots,\\Psi\_\{q\-1\}\\,\]\\;\\in\\;\\mathbb\{R\}^\{\(N\_\{c\}\\times N\_\{x\}\\times N\_\{y\}\)\\times q\}\.For a given state vector𝐱t\\mathbf\{x\}\_\{t\}, the reduced latent representation \(POD coefficients\) is

𝐳t=𝐋𝒳,qT​𝐱t=\[k0,t,k1,t,…,kq−1,t\]⊤∈ℝq,\\mathbf\{z\}\_\{t\}=\\mathbf\{L\}\_\{\\mathcal\{X\},q\}^\{T\}\\mathbf\{x\}\_\{t\}=\\big\[\\,k\_\{0,t\},\\,k\_\{1,t\},\\,\\ldots,\\,k\_\{q\-1,t\}\\,\\big\]^\{\\top\}\\in\\mathbb\{R\}^\{q\},\(5\)wherekj,tk\_\{j,t\}denotes thejj\-th POD coefficient at timettand the reconstruction in the original space is

𝐱tr=𝐋𝒳,q​𝐳t\.\\mathbf\{x\}\_\{t\}^\{r\}=\\mathbf\{L\}\_\{\\mathcal\{X\},q\}\\,\\mathbf\{z\}\_\{t\}\.\(6\)
To enable forward prediction in the reduced space, the reduced coordinates must be approximated by an additional model\. In this work, the regression task is performed using Gaussian process regression \(GPR\) applied to the POD coefficients\[[40](https://arxiv.org/html/2608.11435#bib.bib29)\]\. For completeness, the regression formulation of the POD\-GPR surrogate is provided in[A](https://arxiv.org/html/2608.11435#A1)\.

#### 2\.2\.2OACAE\-MLP

While the POD\-GPR framework provides an effective surrogate modeling strategy for reduced\-order representations, it relies on the assumption that the system dynamics can be captured by a linear subspace and on the regression of reduced POD coefficients that are not explicitly constrained by physical structures\. In complex nonlinear flow regimes, however, these assumptions may become restrictive, leading to limited expressiveness and reduced robustness when extrapolating across parameter ranges or over long temporal horizons\. To overcome these limitations, we present a deep learning based reduced\-order model that combines Convolutional AutoEncoder \(CAE\) with Multilayer Perceptron \(MLP\), further enhanced by the application of Observable\-Augmented Convolutional AutoEncoder \(OACAE\)\[[25](https://arxiv.org/html/2608.11435#bib.bib25)\]\.

A CAE is a neural network architecture commonly employed for nonlinear dimensionality reduction and reconstruction of high\-dimensional data\. As illustrated in Figure[1](https://arxiv.org/html/2608.11435#S2.F1), a CAE model typically consists of an encoderℱe\\mathcal\{F\}\_\{e\}that maps the input field to a low\-dimensional latent representation, and a decoderℱd\\mathcal\{F\}\_\{d\}that reconstructs the field from the latent variables\. Formally, we write

𝐳=ℱe​\(𝐱,ϕe\),𝐱^=ℱd​\(𝐳,ϕd\),\\mathbf\{z\}\\;=\\;\\mathcal\{F\}\_\{e\}\(\\mathbf\{x\};\\phi\_\{e\}\),\\qquad\\hat\{\\mathbf\{x\}\}\\;=\\;\\mathcal\{F\}\_\{d\}\(\\mathbf\{z\};\\phi\_\{d\}\),\(7\)where𝐳∈ℝq\\mathbf\{z\}\\in\\mathbb\{R\}^\{q\}denotes the latent variables andϕ⋅\\phi\_\{\\cdot\}denotes the weights of the corresponding neural network\. The encoder is composed of successive convolutional layers that extract local spatial features and progressively reduce dimensionality\. The resulting feature maps are flattened and passed through an MLP to obtain the latent representation\. The decoder follows a symmetric architecture based on transposed convolutions to reconstruct the high\-dimensional field\. By leveraging the locality and translation invariance of convolutional kernels, CAEs are well\-suited to capture multiscale flow structures and nonlinear manifolds beyond the representational capability of linear techniques such as POD\.

Figure 1:Convolutional autoencoder \(CAE\) for two\-channel velocity fields\. The module highlighted in light yellow is activated when embedding parameters into the latent space, yielding the Observable\-Augmented CAE \(OACAE\)\.A standard CAE is purely data\-driven: it learns a latent representation by minimizing only the reconstruction error\. As a consequence, the latent variables are not explicitly constrained to be organized with respect to physical parameters or the time index\. Nearby parameter configurations may therefore correspond to weakly correlated latent codes, and different parameter\-induced dynamical regimes may partially overlap in the latent space\. Such a representation can be sufficient for reconstruction, but it is not necessarily favorable for inverse modeling, where the surrogate must preserve information that is sensitive to the parameters to be calibrated\.

To address this issue, we augment the CAE by incorporating physical parameters into the training process, thereby obtaining a more informative encoder–decoder architecture and guiding the latent space toward physically consistent structures\. The key idea is to supplement the CAE with an additional MLP branch that links the latent variables to measurable physical quantities, such as fluid density, viscosity, and time index\. This observable\-augmented structure is illustrated by the highlighted branch in light yellow in Figure[1](https://arxiv.org/html/2608.11435#S2.F1)\.

Formally, in the OACAE architecture \(notations with superscriptrrare used to distinguish from the standard CAE\), in addition to the usual encoding–decoding relations

𝐳r=ℱer​\(𝐱,ϕer\),𝐱^r=ℱdr​\(𝐳r,ϕdr\),\\mathbf\{z\}^\{r\}\\;=\\;\\mathcal\{F\}\_\{e\}^\{r\}\(\\mathbf\{x\};\\phi\_\{e\}^\{r\}\),\\qquad\\hat\{\\mathbf\{x\}\}^\{r\}\\;=\\;\\mathcal\{F\}\_\{d\}^\{r\}\(\\mathbf\{z\}^\{r\};\\phi\_\{d\}^\{r\}\),\(8\)we introduce a regression taskℰa\\mathcal\{E\}\_\{a\}from the latent space to the parameter space:

𝜽^=ℰa​\(𝐳r,ϕa\),\\hat\{\\boldsymbol\{\\mathbf\{\\theta\}\}\}\\;=\\;\\mathcal\{E\}\_\{a\}\(\\mathbf\{z\}^\{r\};\\phi\_\{a\}\),\(9\)where𝜽\\boldsymbol\{\\mathbf\{\\theta\}\}denotes the vector of physical observables, andϕa\\phi\_\{a\}are the trainable parameters of the observable regression network\. The overall training loss combines the field reconstruction error with the regression error of the observables:

ℒ1r=‖𝐱−𝐱^r‖22\+α​‖𝜽−𝜽^‖22,\\mathcal\{L\}\_\{1\}^\{r\}\\;=\\;\\\|\\mathbf\{x\}\-\\hat\{\\mathbf\{x\}\}^\{r\}\\\|\_\{2\}^\{2\}\\;\+\\;\\alpha\\,\\\|\\boldsymbol\{\\mathbf\{\\theta\}\}\-\\hat\{\\boldsymbol\{\\mathbf\{\\theta\}\}\}\\\|\_\{2\}^\{2\},\(10\)whereα\>0\\alpha\>0balances the contribution of observable supervision against the reconstruction objective\.

\(a\)Offline training phase 1\(b\)Offline training phase 2\(c\)Online prediction
Figure 2:Workflow of the OACAE\-MLP surrogate model\. \(a\) Offline training of the OACAE with an auxiliary MLP branchℰa\\mathcal\{E\}\_\{a\}to incorporate physical parameters\. \(b\) Offline training of the parameter\-to\-latent regressorℰb\\mathcal\{E\}\_\{b\}using the frozen OACAE obtained in phase \(a\)\. \(c\) Online prediction stage, where a new parameter vector𝜽\\boldsymbol\{\\theta\}is mapped to the latent space byℰb\\mathcal\{E\}\_\{b\}and decoded by the OACAE decoder to produce the corresponding physical field\.The observable\-supervision term inℒ1r\\mathcal\{L\}\_\{1\}^\{r\}encourages the latent variables to retain the information in the physical fields that is most correlated with the prescribed physical observables, including the system parameters and the time index\. More precisely, different combinations of physical parameters guide the autoencoder to learn distinguishable latent representations for different dynamical scenarios\. Furthermore, for a fixed set of physical parameters, the additional guidance from the time index encourages the latent variables to evolve in a temporally consistent and smoother manner\. This temporal regularity is particularly important for multi\-time\-step data assimilation, such as 4D\-Var, where the optimization relies on the consistency of the predicted states over a sequence of observation times\. The resulting latent\-space organization is quantitatively examined in[D](https://arxiv.org/html/2608.11435#A4), where case\-level separability and temporal smoothness metrics are reported for both CAE and OACAE\.

Once the OACAE model is available, its encoderℱer​\(⋅,ϕer\)\\mathcal\{F\}\_\{e\}^\{r\}\(\\cdot;\\phi\_\{e\}^\{r\}\)and decoderℱdr​\(⋅,ϕdr\)\\mathcal\{F\}\_\{d\}^\{r\}\(\\cdot;\\phi\_\{d\}^\{r\}\)are frozen and used as fixed functions\. To construct the forward reduced\-order model, we introduce an MLP, denoted byℰb\\mathcal\{E\}\_\{b\}, which maps the system parameters to the observable\-augmented latent space \(see Figure[2\(b\)](https://arxiv.org/html/2608.11435#S2.F2.sf2)\)\. The predicted latent vector is then decoded by the OACAE decoder to recover the physical field:

𝐳r,∗=ℰb​\(𝜽,ϕbr\),𝐱^r,∗=ℱdr​\(𝐳r,∗,ϕdr\),\\mathbf\{z\}^\{r,\*\}\\;=\\;\\mathcal\{E\}\_\{b\}\(\\boldsymbol\{\\theta\};\\phi\_\{b\}^\{r\}\),\\qquad\\hat\{\\mathbf\{x\}\}^\{r,\*\}\\;=\\;\\mathcal\{F\}\_\{d\}^\{r\}\(\\mathbf\{z\}^\{r,\*\};\\phi\_\{d\}^\{r\}\),\(11\)where𝜽∈ℝdθ\\boldsymbol\{\\theta\}\\in\\mathbb\{R\}^\{d\_\{\\theta\}\}denotes the normalized input system parameters,ϕbr\\phi\_\{b\}^\{r\}is the trainable weights of the regressor,𝐳r,∗\\mathbf\{z\}^\{r,\*\}is the latent prediction, and𝐱^r,∗\\hat\{\\mathbf\{x\}\}^\{r,\*\}is the final predicted field\.

The training ofℰb\\mathcal\{E\}\_\{b\}is performed in a supervised fashion using pairs\{\(𝜽i,𝐱i\)\}i=1N\\\{\(\\boldsymbol\{\\theta\}\_\{i\},\\mathbf\{x\}\_\{i\}\)\\\}\_\{i=1\}^\{N\}, as illustrated in Figure[2\(b\)](https://arxiv.org/html/2608.11435#S2.F2.sf2)\. For each snapshot, the corresponding reference latent code is precomputed via the frozen encoder,𝐳ir=ℱer​\(𝐱i,ϕer\)\\mathbf\{z\}^\{r\}\_\{i\}=\\mathcal\{F\}\_\{e\}^\{r\}\(\\mathbf\{x\}\_\{i\};\\phi\_\{e\}^\{r\}\)\. The objective loss function forℰb\\mathcal\{E\}\_\{b\}combines a prediction error in the physical space with a regression error in the latent space:

ℒ2r=‖𝐱−𝐱^r,∗‖22\+β​‖𝐳r−𝐳r,∗‖22,\\mathcal\{L\}\_\{2\}^\{r\}\\;=\\;\\\|\\mathbf\{x\}\-\\hat\{\\mathbf\{x\}\}^\{r,\*\}\\\|\_\{2\}^\{2\}\+\\beta\\,\\\|\\mathbf\{z\}^\{r\}\-\\mathbf\{z\}^\{r,\*\}\\\|\_\{2\}^\{2\},\(12\)whereβ\>0\\beta\>0balances the contribution of latent supervision against field prediction\. During this training stage, the OACAE encoder and decoder remain fixed, so that only the parametersϕbr\\phi\_\{b\}^\{r\}of the MLPℰb\\mathcal\{E\}\_\{b\}are updated\. This workflow provides a non\-intrusive parametric ROM that directly maps physical parameters to physical fields via the augmented latent representation, while leveraging the structure learned by the OACAE\. In the online stage \(see Figure[2\(c\)](https://arxiv.org/html/2608.11435#S2.F2.sf3)\), this allows fast and repeated evaluations: once trained, the model can efficiently generate large numbers of flow\-field predictions for arbitrary parameter inputs without resorting to expensive numerical simulations\.

The OACAE and its observable branch are implemented using a standard convolutional encoder\-decoder architecture with MaxPooling layers and fully connected layers, complemented by additional MLP components\. The detailed layer\-by\-layer configurations of the OACAE encoder/decoder, the observable regression branch MLP \(ℰa\\mathcal\{E\}\_\{a\}\), and the parameter\-to\-latent regressor MLP \(ℰb\\mathcal\{E\}\_\{b\}\) are provided in[B](https://arxiv.org/html/2608.11435#A2)\.

As illustrated in Figure[2](https://arxiv.org/html/2608.11435#S2.F2), the training procedure of the OACAE\-MLP ROM consists of two successive phases\. In the first phase \(Figure[2\(a\)](https://arxiv.org/html/2608.11435#S2.F2.sf1)\), the parameters of the encoder, decoder, and observable branch \(ϕer,ϕdr,ϕar\\phi\_\{e\}^\{r\},\\phi\_\{d\}^\{r\},\\phi\_\{a\}^\{r\}\) are optimized by minimizing the lossℒ1r\\mathcal\{L\}\_\{1\}^\{r\}defined in Equation \([13](https://arxiv.org/html/2608.11435#S2.E13)\)\. In the second phase \(Figure[2\(b\)](https://arxiv.org/html/2608.11435#S2.F2.sf2)\), the encoder and decoder are kept frozen, and the parameter\-to\-latent regressorℰb\\mathcal\{E\}\_\{b\}with parametersϕbr\\phi\_\{b\}^\{r\}is trained by minimizingℒ2r\\mathcal\{L\}\_\{2\}^\{r\}in Equation \([14](https://arxiv.org/html/2608.11435#S2.E14)\)\. For clarity, we rewrite the two objective functions below to explicitly highlight their dependence on the trainable parameters:

ℒ1r​\(ϕar,ϕer,ϕdr\)=‖𝐱−ℱdr​\(ℱer​\(𝐱,ϕer\),ϕdr\)‖22\+α​‖𝜽−ℰa​\(ℱer​\(𝐱,ϕer\),ϕar\)‖22,\\mathcal\{L\}\_\{1\}^\{r\}\(\\phi\_\{a\}^\{r\},\\phi\_\{e\}^\{r\},\\phi\_\{d\}^\{r\}\)\\;=\\;\\\|\\mathbf\{x\}\-\\mathcal\{F\}\_\{d\}^\{r\}\(\\mathcal\{F\}\_\{e\}^\{r\}\(\\mathbf\{x\};\\phi\_\{e\}^\{r\}\);\\phi\_\{d\}^\{r\}\)\\\|\_\{2\}^\{2\}\\;\+\\;\\alpha\\,\\\|\\boldsymbol\{\\theta\}\-\\mathcal\{E\}\_\{a\}\(\\mathcal\{F\}\_\{e\}^\{r\}\(\\mathbf\{x\};\\phi\_\{e\}^\{r\}\);\\phi\_\{a\}^\{r\}\)\\\|\_\{2\}^\{2\},\(13\)ℒ2r​\(ϕbr\)=‖𝐱−ℱdr​\(ℰb​\(𝜽,ϕbr\),ϕdr\)‖22\+β​‖ℱer​\(𝐱,ϕer\)−ℰb​\(𝜽,ϕbr\)‖22,\\mathcal\{L\}\_\{2\}^\{r\}\(\\phi\_\{b\}^\{r\}\)\\;=\\;\\\|\\mathbf\{x\}\-\\mathcal\{F\}\_\{d\}^\{r\}\(\\mathcal\{E\}\_\{b\}\(\\boldsymbol\{\\theta\};\\phi\_\{b\}^\{r\}\);\\phi\_\{d\}^\{r\}\)\\\|\_\{2\}^\{2\}\+\\beta\\,\\\|\\mathcal\{F\}\_\{e\}^\{r\}\(\\mathbf\{x\};\\phi\_\{e\}^\{r\}\)\-\\mathcal\{E\}\_\{b\}\(\\boldsymbol\{\\theta\};\\phi\_\{b\}^\{r\}\)\\\|\_\{2\}^\{2\},\(14\)with the corresponding optimization problems solved by the Adam optimizer:

\(ϕar,ϕer,ϕdr\)∗=arg⁡min\(ϕar,ϕer,ϕdr\)​ℒ1r​\(ϕar,ϕer,ϕdr\),\(\\phi\_\{a\}^\{r\},\\phi\_\{e\}^\{r\},\\phi\_\{d\}^\{r\}\)^\{\*\}\\;=\\;\\arg\\min\_\{\(\\phi\_\{a\}^\{r\},\\phi\_\{e\}^\{r\},\\phi\_\{d\}^\{r\}\)\}\\,\\mathcal\{L\}\_\{1\}^\{r\}\(\\phi\_\{a\}^\{r\},\\phi\_\{e\}^\{r\},\\phi\_\{d\}^\{r\}\),\(15\)\(ϕbr\)∗=arg⁡min\(ϕbr\)​ℒ2r​\(ϕbr\)\.\(\\phi\_\{b\}^\{r\}\)^\{\*\}\\;=\\;\\arg\\min\_\{\(\\phi\_\{b\}^\{r\}\)\}\\,\\mathcal\{L\}\_\{2\}^\{r\}\(\\phi\_\{b\}^\{r\}\)\.\(16\)
In this study, the weighting coefficients are set toα=0\.05\\alpha=0\.05andβ=1\\beta=1based on an empirical assessment of the trained model performance \(see[E\.1](https://arxiv.org/html/2608.11435#A5.SS1)for a detailed analysis of the choice of these coefficients\)\.

In addition, we denote by CAE\-MLP the baseline model obtained by replacing the OACAE in the OACAE\-MLP framework with a standard CAE, that is, by settingα=0\\alpha=0and removing the observable regression branch\. Accordingly, the CAE\-MLP can be regarded as a special case of the proposed OACAE\-MLP model, and is introduced to assess the impact of physics\-aware latent supervision\. In this case, the model consists of a standard CAE with an encoderℱe\\mathcal\{F\}\_\{e\}and a decoderℱd\\mathcal\{F\}\_\{d\}together with a parameter\-to\-latent regressor MLP, denoted byℰc\\mathcal\{E\}\_\{c\}\.

Formally, the encoder\-decoder relations of the standard CAE read

𝐳=ℱe​\(𝐱,ϕe\),𝐱^=ℱd​\(𝐳,ϕd\),\\mathbf\{z\}\\;=\\;\\mathcal\{F\}\_\{e\}\(\\mathbf\{x\};\\phi\_\{e\}\),\\qquad\\hat\{\\mathbf\{x\}\}\\;=\\;\\mathcal\{F\}\_\{d\}\(\\mathbf\{z\};\\phi\_\{d\}\),\(17\)where𝐳\\mathbf\{z\}is the latent code, andϕe,ϕd\\phi\_\{e\},\\phi\_\{d\}are the trainable weights of the encoder and decoder\.

The training loss for the CAE stage is thus given by

ℒ1​\(ϕe,ϕd\)=‖𝐱−ℱd​\(ℱe​\(𝐱,ϕe\),ϕd\)‖22\.\\mathcal\{L\}\_\{1\}\(\\phi\_\{e\},\\phi\_\{d\}\)\\;=\\;\\\|\\mathbf\{x\}\-\\mathcal\{F\}\_\{d\}\(\\mathcal\{F\}\_\{e\}\(\\mathbf\{x\};\\phi\_\{e\}\);\\phi\_\{d\}\)\\\|\_\{2\}^\{2\}\.\(18\)
In the second stage, a parameter\-to\-latent regressorℰc\\mathcal\{E\}\_\{c\}with weightsϕc\\phi\_\{c\}is trained to map input parameters𝜽\\boldsymbol\{\\theta\}to the latent space, while the decoderℱd\\mathcal\{F\}\_\{d\}remains frozen\. The corresponding loss function is

ℒ2​\(ϕc\)=‖𝐱−ℱd​\(ℰc​\(𝜽,ϕc\),ϕd\)‖22\+β​‖ℱe​\(𝐱,ϕe\)−ℰc​\(𝜽,ϕc\)‖22,\\mathcal\{L\}\_\{2\}\(\\phi\_\{c\}\)\\;=\\;\\\|\\mathbf\{x\}\-\\mathcal\{F\}\_\{d\}\(\\mathcal\{E\}\_\{c\}\(\\boldsymbol\{\\theta\};\\phi\_\{c\}\);\\phi\_\{d\}\)\\\|\_\{2\}^\{2\}\+\\beta\\,\\\|\\mathcal\{F\}\_\{e\}\(\\mathbf\{x\};\\phi\_\{e\}\)\-\\mathcal\{E\}\_\{c\}\(\\boldsymbol\{\\theta\};\\phi\_\{c\}\)\\\|\_\{2\}^\{2\},\(19\)with optimization problems

\(ϕe,ϕd\)∗=arg⁡min\(ϕe,ϕd\)​ℒ1​\(ϕe,ϕd\),\(ϕc\)∗=arg⁡minϕc​ℒ2​\(ϕc\)\.\(\\phi\_\{e\},\\phi\_\{d\}\)^\{\*\}\\;=\\;\\arg\\min\_\{\(\\phi\_\{e\},\\phi\_\{d\}\)\}\\mathcal\{L\}\_\{1\}\(\\phi\_\{e\},\\phi\_\{d\}\),\\qquad\(\\phi\_\{c\}\)^\{\*\}\\;=\\;\\arg\\min\_\{\\phi\_\{c\}\}\\mathcal\{L\}\_\{2\}\(\\phi\_\{c\}\)\.\(20\)

### 2\.3Data assimilation frameworks for parameter calibration

In this section, we present three complementary DA frameworks that leverage the previously introduced ROMs\. First, we discuss the use of the Ensemble Kalman Filter \(EnKF\) combined with the POD\-GPR surrogate for parameter estimations\. Second, we formulate variational assimilation strategies \(3D\-Var/4D\-Var\) combined with CAE\-MLP and OACAE\-MLP, allowing for optimization strategies of parameter estimates\.

#### 2\.3\.1EnKF combined with POD\-GPR

Among the available data assimilation techniques, the EnKF is adopted in combination with the POD\-GPR surrogate due to its derivative\-free and ensemble\-based formulation\. In contrast to variational approaches, which require gradients of the reduced\-order model with respect to the parameters, the POD\-GPR surrogate does not provide analytic derivatives and is therefore not directly compatible with gradient\-based optimization methods\. The EnKF addresses the calibration problem within a statistical framework by propagating an ensemble of parameter realizations, making it particularly well suited to handle the variability and nonlinearity inherent in POD\-GPR predictions\. Moreover, the use of ensembles naturally represents uncertainty while maintaining computational efficiency, making EnKF a natural benchmark for parameter calibration with POD\-GPR\.

The Ensemble Kalman Filter \(EnKF\) is a sequential Monte Carlo method that extends the classical Kalman filter to high\-dimensional nonlinear problems\. It represents the uncertainty of the control vector by an ensemble of realizations, which are propagated forward by the model and used to compute the Kalman gain in a statistical manner\. Given a background ensemble𝐄b=\(𝜽1b,…,𝜽Neb\)⊤∈ℝd×Ne\\mathbf\{E\}^\{b\}=\(\\boldsymbol\{\\theta\}\_\{1\}^\{b\},\\ldots,\\boldsymbol\{\\theta\}\_\{N\_\{e\}\}^\{b\}\)^\{\\top\}\\in\\mathbb\{R\}^\{d\\times N\_\{e\}\}sampled from a prior distribution𝒩⁡\(𝜽¯b,𝐁\)\\mathcal\{N\}\(\\bar\{\\boldsymbol\{\\theta\}\}^\{b\},\\mathbf\{B\}\), where𝜽¯b\\bar\{\\boldsymbol\{\\theta\}\}^\{b\}denotes the background mean parameter vector \(typically chosen as the empirical mean of the training dataset\) and𝐁\\mathbf\{B\}represents the background error covariance matrix characterizing the prior uncertainty in the parameters, estimated from the empirical covariance of the training parameter ensemble, computed as the sample covariance of the parameter anomalies with respect to the ensemble mean\. The EnKF proceeds as follows\[[41](https://arxiv.org/html/2608.11435#bib.bib28)\]:

*\(i\) Forecast step\.*Each ensemble member is directly evaluated through the observation operatorℋ\\mathcal\{H\}, yielding

𝐱if=ℋ\(𝜽ib\),i=1,…,Ne\.\\mathbf\{x\}\_\{i\}^\{f\}=\\mathcal\{H\}\(\\boldsymbol\{\\theta\}\_\{i\}^\{b\}\),\\qquad i=1,\\ldots,N\_\{e\}\.\(21\)
*\(ii\) Analysis step\.*The forecast ensemble\{𝐱if\}\\\{\\mathbf\{x\}\_\{i\}^\{f\}\\\}is compared with the observations𝐲\\mathbf\{y\}\. The error cross\-covariances are estimated from the ensemble as

𝐁𝐇⊤≈1Ne−1​∑i=1Ne\(𝜽ib−𝜽¯b\)​\(𝐱if−𝐱f¯\)⊤,\\mathbf\{B\}\\mathbf\{H\}^\{\\top\}\\;\\approx\\;\\frac\{1\}\{N\_\{e\}\-1\}\\sum\_\{i=1\}^\{N\_\{e\}\}\(\\boldsymbol\{\\theta\}\_\{i\}^\{b\}\-\\bar\{\\boldsymbol\{\\theta\}\}^\{b\}\)\\big\(\\mathbf\{x\}\_\{i\}^\{f\}\-\\overline\{\\mathbf\{x\}^\{f\}\}\\big\)^\{\\top\},\(22\)𝐇𝐁𝐇⊤≈1Ne−1​∑i=1Ne\(𝐱if−𝐱f¯\)​\(𝐱if−𝐱f¯\)⊤\.\\mathbf\{H\}\\mathbf\{B\}\\mathbf\{H\}^\{\\top\}\\;\\approx\\;\\frac\{1\}\{N\_\{e\}\-1\}\\sum\_\{i=1\}^\{N\_\{e\}\}\\big\(\\mathbf\{x\}\_\{i\}^\{f\}\-\\overline\{\\mathbf\{x\}^\{f\}\}\\big\)\\big\(\\mathbf\{x\}\_\{i\}^\{f\}\-\\overline\{\\mathbf\{x\}^\{f\}\}\\big\)^\{\\top\}\.\(23\)The Kalman gain is then

𝐊∗=𝐁𝐇⊤​\(𝐇𝐁𝐇⊤\+𝐑\)−1,\\mathbf\{K\}^\{\*\}\\;=\\;\\mathbf\{B\}\\mathbf\{H\}^\{\\top\}\\big\(\\mathbf\{H\}\\mathbf\{B\}\\mathbf\{H\}^\{\\top\}\+\\mathbf\{R\}\\big\)^\{\-1\},\(24\)where𝐑\\mathbf\{R\}denotes the observation error covariance matrix that quantifies measurement uncertainties\.

Finally, each parameter ensemble member is updated according to

𝜽ia=𝜽ib\+𝐊∗\(𝐲i−𝐱if\),i=1,…,Ne,\\boldsymbol\{\\theta\}\_\{i\}^\{a\}=\\boldsymbol\{\\theta\}\_\{i\}^\{b\}\+\\mathbf\{K\}^\{\*\}\\big\(\\mathbf\{y\}\_\{i\}\-\\mathbf\{x\}\_\{i\}^\{f\}\\big\),\\qquad i=1,\\ldots,N\_\{e\},\(25\)where𝐲i\\mathbf\{y\}\_\{i\}are perturbed observations sampled from𝒩⁡\(𝐲,𝐑\)\\mathcal\{N\}\(\\mathbf\{y\},\\mathbf\{R\}\)\.

The optimal parameter estimate is obtained as the ensemble mean

𝜽¯a=1Ne​∑i=1Ne𝜽ia\.\\bar\{\\boldsymbol\{\\theta\}\}^\{a\}=\\frac\{1\}\{N\_\{e\}\}\\sum\_\{i=1\}^\{N\_\{e\}\}\\boldsymbol\{\\theta\}\_\{i\}^\{a\}\.\(26\)
In our framework, the EnKF is applied directly to the system parameters𝜽\\boldsymbol\{\\theta\}\. The complete parameter vector must be provided as input to the POD\-GPR surrogate, which serves as the observation operator in the data assimilation process\. This vector includes both the physical parameters of the system and the time index associated with each snapshot\. Accordingly, a single𝜽\\boldsymbol\{\\theta\}corresponds to one snapshot, whereas the combination of fixed physical parameters with a sequence of time indices describes the entire flow evolution\. For parameter calibration, however, we restrict the update to the most sensitive physical parameters, denoted by𝜽control\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}, while the remaining physical parameters and the snapshot time index \(e\.g\.,tkt\_\{k\}\) are considered as known, denoted by𝜽tkfixed\\boldsymbol\{\\theta\}^\{\\mathrm\{fixed\}\}\_\{t\_\{k\}\}\. Thus, each parameter vector at timetkt\_\{k\}can be represented as

𝜽tk=\(𝜽control,𝜽tkfixed\)\.\\boldsymbol\{\\theta\}\_\{t\_\{k\}\}=\\big\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\},\\;\\boldsymbol\{\\theta\}^\{\\mathrm\{fixed\}\}\_\{t\_\{k\}\}\\big\)\.\(27\)The observation operatorℋ\\mathcal\{H\}is identified with the POD\-GPR surrogate model, denotedℋPOD\-GPR\\mathcal\{H\}\_\{\\text\{POD\-GPR\}\}, which maps a complete parameter vector𝜽tk\\boldsymbol\{\\theta\}\_\{t\_\{k\}\}to a prediction of the physical field\. The corresponding observation𝐲\\mathbf\{y\}is taken from the true physical field \(e\.g\., flattened snapshots of high\-fidelity simulations\)\. This leads to the following assimilation stages at timetkt\_\{k\}:

1. 1\.*Prediction step:*each ensemble member𝜽icontrol,b\\boldsymbol\{\\theta\}\_\{i\}^\{\\text\{control\},b\}is combined with the fixed parameters and the time indextkt\_\{k\}to form 𝜽tk,i=\(θicontrol,b,θtkfixed\),\\boldsymbol\{\\theta\}\_\{t\_\{k\},i\}=\\big\(\\theta^\{\\mathrm\{control\},b\}\_\{i\},\\;\\theta^\{\\mathrm\{fixed\}\}\_\{t\_\{k\}\}\\big\),\(28\)and propagated through the surrogate to obtain the forecast field𝐱if=ℋPOD\-GPR​\(𝜽tk,i\)\\mathbf\{x\}\_\{i\}^\{f\}=\\mathcal\{H\}\_\{\\text\{POD\-GPR\}\}\(\\boldsymbol\{\\theta\}\_\{t\_\{k\},i\}\)\.
2. 2\.*Analysis step:*the discrepancy between the ensemble forecasts𝐱if\\mathbf\{x\}\_\{i\}^\{f\}and the perturbed observations is used to update the parameter ensemble according to the EnKF equations\.
3. 3\.*Ensemble mean:*the calibrated parameter estimate is given by the mean of the analysis ensemble,𝜽¯control,a=1Ne​∑i=1Ne𝜽icontrol,a\\bar\{\\boldsymbol\{\\theta\}\}^\{\\text\{control\},a\}=\\tfrac\{1\}\{N\_\{e\}\}\\sum\_\{i=1\}^\{N\_\{e\}\}\\boldsymbol\{\\theta\}\_\{i\}^\{\\text\{control\},a\}\.

This procedure corresponds to a single\-time\-step EnKF strategy, often applied in the calibration of stationary or time\-averaged fields, where the update of the control parameters relies on observations at a single time instant\. In our study, however, the inclusion of the time index in the parameter vector allows us to extend the assimilation to a multi\-time\-step EnKF\. Consider a sequence of snapshots from one evolution case,

𝐱t0,𝐱t1,𝐱t2,…,𝐱tk,\\mathbf\{x\}\_\{t\_\{0\}\},\\mathbf\{x\}\_\{t\_\{1\}\},\\mathbf\{x\}\_\{t\_\{2\}\},\\ldots,\\mathbf\{x\}\_\{t\_\{k\}\},\(29\)which serve as observations at successive assimilation steps\. Starting from a prior control vector, the EnKF update with𝐱t0\\mathbf\{x\}\_\{t\_\{0\}\}yields an analysis control vector, which is then taken as the background for the update with𝐱t1\\mathbf\{x\}\_\{t\_\{1\}\}, and so on\. In this way, observations from multiple time steps are successively assimilated, and the final analysis control vector incorporates observational information from the entire sequence of snapshots\.

#### 2\.3\.23D\-Var and 4D\-Var combined with CAE\-MLP and OACAE\-MLP

A key challenge in variational data assimilation lies in defining the forward and observation operators, which often involve nonlinear and high\-dimensional mappings\. In particular, integrating deep learning models into DA workflows remains difficult, as most existing software packages do not natively support neural networks within variational schemes\. To bypass this limitation, we rely on the recently proposed Python package TorchDA\[[13](https://arxiv.org/html/2608.11435#bib.bib35)\], which provides a unified framework for coupling deep learning surrogates with variational DA algorithms\. Built on the PyTorch ecosystem, TorchDA streamlines 3D\-Var/4D\-Var workflows by enabling neural networks to serve directly as forward or observation operators, with automatic differentiation for tangent\-linear and adjoint computations, GPU acceleration, and user\-friendly configuration interfaces\. This capability allows us to seamlessly integrate DL\-based ROMs, such as the CAE\-MLP and OACAE\-MLP introduced in Section 3\.2, into variational assimilation pipelines\.

Variational DA seeks to estimate the optimal state or parameter vector𝜽\\boldsymbol\{\\theta\}by minimizing a cost function that balances background information with observational constraints\. Here, we focus on parameter calibration using variational DA with DL\-based observation operators\. In the 3D\-Var case, where assimilation is performed at a single time step, the cost function reads

J3D\-Var​\(𝜽\)=12​\(𝜽−𝜽b\)⊤​𝐁−1​\(𝜽−𝜽b\)\+12​\(𝐲0−ℋ⁡\(𝜽\)\)⊤​𝐑−1​\(𝐲0−ℋ⁡\(𝜽\)\),J\_\{\\textbf\{3D\-Var\}\}\(\\boldsymbol\{\\theta\}\)\\;=\\;\\tfrac\{1\}\{2\}\\big\(\\boldsymbol\{\\theta\}\-\\boldsymbol\{\\theta\}^\{b\}\\big\)^\{\\top\}\\mathbf\{B\}^\{\-1\}\\big\(\\boldsymbol\{\\theta\}\-\\boldsymbol\{\\theta\}^\{b\}\\big\)\\;\+\\;\\tfrac\{1\}\{2\}\\big\(\\mathbf\{y\}\_\{0\}\-\\mathcal\{H\}\(\\boldsymbol\{\\theta\}\)\\big\)^\{\\top\}\\mathbf\{R\}^\{\-1\}\\big\(\\mathbf\{y\}\_\{0\}\-\\mathcal\{H\}\(\\boldsymbol\{\\theta\}\)\\big\),\(30\)where𝜽b\\boldsymbol\{\\theta\}^\{b\}denotes the background parameter vector,𝐁\\mathbf\{B\}the background error covariance matrix,𝐲0\\mathbf\{y\}\_\{0\}the observation,ℋ\\mathcal\{H\}the DL\-based observation operator, and𝐑\\mathbf\{R\}the observation error covariance matrix\.

The 4D\-Var framework extends this principle to a temporal window\[t0,…,tK\]\[t\_\{0\},\\ldots,t\_\{K\}\], by assimilating all observations within this interval:

J4D\-Var​\(𝜽\)=12​\(𝜽−𝜽b\)⊤​𝐁−1​\(𝜽−𝜽b\)\+12​∑k=0K\(𝐲tk−ℋtk​\(𝜽\)\)⊤​𝐑−1​\(𝐲tk−ℋtk​\(𝜽\)\),J\_\{\\textbf\{4D\-Var\}\}\(\\boldsymbol\{\\theta\}\)\\;=\\;\\tfrac\{1\}\{2\}\\big\(\\boldsymbol\{\\theta\}\-\\boldsymbol\{\\theta\}^\{b\}\\big\)^\{\\top\}\\mathbf\{B\}^\{\-1\}\\big\(\\boldsymbol\{\\theta\}\-\\boldsymbol\{\\theta\}^\{b\}\\big\)\\;\+\\;\\tfrac\{1\}\{2\}\\sum\_\{k=0\}^\{K\}\\big\(\\mathbf\{y\}\_\{t\_\{k\}\}\-\\mathcal\{H\}\_\{t\_\{k\}\}\(\\boldsymbol\{\\theta\}\)\\big\)^\{\\top\}\\mathbf\{R\}^\{\-1\}\\big\(\\mathbf\{y\}\_\{t\_\{k\}\}\-\\mathcal\{H\}\_\{t\_\{k\}\}\(\\boldsymbol\{\\theta\}\)\\big\),\(31\)whereℋtk​\(𝜽\)\\mathcal\{H\}\_\{t\_\{k\}\}\(\\boldsymbol\{\\theta\}\)denotes the surrogate\-predicted field at timetkt\_\{k\}corresponding to the parameter vector𝜽\\boldsymbol\{\\theta\}\. Unlike classical 4D\-Var formulations, where the control variable evolves in time according to the governing dynamics, the present 4D\-Var treats the target parameter vector𝜽\\boldsymbol\{\\theta\}as a fixed control variable, which is mapped to multiple observation times through distinct observation operatorsℋtk\\mathcal\{H\}\_\{t\_\{k\}\}\. The minimization ofJ3D\-VarJ\_\{\\textbf\{3D\-Var\}\}andJ4D\-VarJ\_\{\\textbf\{4D\-Var\}\}is carried out iteratively, often using gradient\-based optimizers such as Adam or SGD, with gradients obtained by automatic differentiation of the DL\-based surrogates enabled by TorchDA\. A learning\-rate sensitivity sweep is reported in[E\.3](https://arxiv.org/html/2608.11435#A5.SS3)to examine the stability of this gradient\-based optimization and to identify a suitable range of learning rates\.

We illustrate the application of the framework using the CAE\-MLP model, noting that the procedure for the OACAE\-MLP surrogate is entirely analogous\. In this setting, the reduced\-order surrogate CAE\-MLP serves as the observation operatorℋ\\mathcal\{H\}\. As defined in Section[2\.3\.1](https://arxiv.org/html/2608.11435#S2.SS3.SSS1), the complete parameter vector is𝜽tk=\(𝜽control,𝜽tkfixed\)\\boldsymbol\{\\theta\}\_\{t\_\{k\}\}=\\big\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\},\\;\\boldsymbol\{\\theta\}^\{\\mathrm\{fixed\}\}\_\{t\_\{k\}\}\\big\), which includes the time\-index parametertkt\_\{k\}and is used as input to the surrogate\. The control variables𝜽control\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}act as the independent variables in the objective cost functions\. Accordingly, the 3D\-Var and 4D\-Var cost functions can be written as

J3D\-Var​\(𝜽control\)=\\displaystyle J\_\{\\textbf\{3D\-Var\}\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\)\\;=\\;12​\(𝜽control−𝜽control,b\)⊤​𝐁−1​\(𝜽control−𝜽control,b\)\+\\displaystyle\\tfrac\{1\}\{2\}\\big\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\-\\boldsymbol\{\\theta\}^\{\\mathrm\{control\},b\}\\big\)^\{\\top\}\\mathbf\{B\}^\{\-1\}\\big\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\-\\boldsymbol\{\\theta\}^\{\\mathrm\{control\},b\}\\big\)\\;\+12​\(𝐱tk−ℋtk​\(𝜽control\)\)⊤​𝐑−1​\(𝐱tk−ℋtk​\(𝜽control\)\),\\displaystyle\\tfrac\{1\}\{2\}\\big\(\\mathbf\{x\}\_\{t\_\{k\}\}\-\\mathcal\{H\}\_\{t\_\{k\}\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\)\\big\)^\{\\top\}\\mathbf\{R\}^\{\-1\}\\big\(\\mathbf\{x\}\_\{t\_\{k\}\}\-\\mathcal\{H\}\_\{t\_\{k\}\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\)\\big\),\(32\)J4D\-Var​\(𝜽control\)=\\displaystyle J\_\{\\textbf\{4D\-Var\}\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\)\\;=\\;12​\(𝜽control−𝜽control,b\)⊤​𝐁−1​\(𝜽control−𝜽control,b\)\+\\displaystyle\\tfrac\{1\}\{2\}\\big\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\-\\boldsymbol\{\\theta\}^\{\\mathrm\{control\},b\}\\big\)^\{\\top\}\\mathbf\{B\}^\{\-1\}\\big\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\-\\boldsymbol\{\\theta\}^\{\\mathrm\{control\},b\}\\big\)\\;\+12​∑k=0K\(𝐱tk−ℋtk​\(𝜽control\)\)⊤​𝐑−1​\(𝐱tk−ℋtk​\(𝜽control\)\),\\displaystyle\\tfrac\{1\}\{2\}\\sum\_\{k=0\}^\{K\}\\big\(\\mathbf\{x\}\_\{t\_\{k\}\}\-\\mathcal\{H\}\_\{t\_\{k\}\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\)\\big\)^\{\\top\}\\mathbf\{R\}^\{\-1\}\\big\(\\mathbf\{x\}\_\{t\_\{k\}\}\-\\mathcal\{H\}\_\{t\_\{k\}\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\)\\big\),\(33\)where, for a given control parameter vector𝜽control\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}, the surrogate prediction of the physical field is expressed as

ℋtk​\(𝜽control\)=ℋCAE\-MLP​\(𝜽control,𝜽tkfixed\)=ℱd​\(ℰc​\(𝜽control,𝜽tkfixed\)\)\.\\mathcal\{H\}\_\{t\_\{k\}\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}\)=\\mathcal\{H\}\_\{\\text\{CAE\-MLP\}\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\},\\boldsymbol\{\\theta\}^\{\\mathrm\{fixed\}\}\_\{t\_\{k\}\}\)=\\mathcal\{F\}\_\{d\}\(\\mathcal\{E\}\_\{c\}\(\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\},\\boldsymbol\{\\theta\}^\{\\mathrm\{fixed\}\}\_\{t\_\{k\}\}\)\)\.\(34\)whereℱd\\mathcal\{F\}\_\{d\}denotes the decoder andℰc\\mathcal\{E\}\_\{c\}the parameter\-to\-latent MLP regressor, as introduced in Sec[2\.2\.2](https://arxiv.org/html/2608.11435#S2.SS2.SSS2)\. The variational assimilation proceeds as follows\. First, the background control vector𝜽control,b\\boldsymbol\{\\theta\}^\{\\mathrm\{control\},b\}is defined as the empirical mean of training parameters, and the associated background covariance𝐁\\mathbf\{B\}is estimated from the training dataset as well\. At each observation timetkt\_\{k\}, the CAE\-MLP surrogate predicts the physical field corresponding to the current complete parameter vector\. The discrepancy between the predicted field and the flattened true snapshot𝐱tk\\mathbf\{x\}\_\{t\_\{k\}\}\(used here as the observation\) is then evaluated in the 3D\-Var/4D\-Var cost function\. Gradients of the objective function with respect to𝜽control\\boldsymbol\{\\theta\}^\{\\mathrm\{control\}\}are computed via backpropagation through the CAE\-MLP network using TorchDA, and the control vector is iteratively updated with a gradient\-based optimizer until convergence, yielding the analysis estimate𝜽control,a\\boldsymbol\{\\theta\}^\{\\mathrm\{control\},a\}\. Through this procedure, observational information distributed over multiple time steps can be effectively assimilated into the control parameters, while leveraging the differentiable structure of the CAE\-MLP surrogate\. Since the balance between the background and observation terms depends on the relative scaling of𝐁\\mathbf\{B\}and𝐑\\mathbf\{R\}, we also report a covariance\-scaling sensitivity study in[E\.4](https://arxiv.org/html/2608.11435#A5.SS4)\.

## 3Numerical Experiments

We evaluate the presented and proposed ROM\-DA frameworks on two representative CFDBench problems: the dam\-break flow and the lid\-driven cavity flow\. To emphasize the contribution of the proposed physics\-aware variational DA\-DL\-ROM approach, we adopt the EnKF\-POD\-GPR framework as a baseline, following the methodology introduced in Sec[2\.2\.1](https://arxiv.org/html/2608.11435#S2.SS2.SSS1)and Sec[2\.3\.1](https://arxiv.org/html/2608.11435#S2.SS3.SSS1), and inspired by the idea from\[[40](https://arxiv.org/html/2608.11435#bib.bib29),[41](https://arxiv.org/html/2608.11435#bib.bib28)\]\. In addition to the EnKF\-POD\-GPR baseline, we compare CAE\-MLP and OACAE\-MLP to isolate the effect of observable supervision\. These two models share the same encoder\-decoder backbone, latent dimension, parameter\-to\-latent regression strategy, and variational assimilation setting\. The main difference is that OACAE\-MLP uses observable supervision during the autoencoder training stage, whereas CAE\-MLP relies only on the reconstruction loss\. Therefore, the comparison between CAE\-MLP and OACAE\-MLP serves as a controlled ablation for assessing the effect of observable supervision on latent\-space organization, forward prediction, and inverse calibration\.

To rigorously assess the generalization capability of the surrogate models, data splitting is performed at the case level rather than at the snapshot level\. All snapshots associated with a given simulation case are assigned exclusively to either the training or the test set\. This strategy prevents information leakage across different parameter realizations and provides a realistic evaluation of model performance on unseen configurations\. Both the solution fields and the associated parameter data are normalized using feature\-wise min–max scaling to the range\[0,1\]\[0,1\]\. This normalization ensures consistent scaling across different state variables and parameters, avoids numerical imbalances, and facilitates the convergence of learning algorithms\. The same normalization procedure is applied consistently throughout the reduced\-order forward modeling and DA inverse modeling stages\. Before fixing the final inverse\-modeling setting, we performed preliminary multi\-parameter assimilation tests in which all available candidate parameters were included as control variables\. These tests were used to assess the practical identifiability of each parameter from the available observations\. Parameters that remained weakly identifiable, led to unstable assimilation updates, or produced calibrated errors comparable to or larger than the background errors were fixed in the subsequent experiments\. This reduced control setting avoids over\-parameterizing the inverse problem and enables a clearer assessment of the achievable inverse modeling performance from the available observations\.

### 3\.1The CFDBench dataset

To evaluate data\-driven reduced\-order models and their integration with data assimilation frameworks, we rely on the recently released CFDBench dataset\[[44](https://arxiv.org/html/2608.11435#bib.bib34)\]\. CFDBench provides a large\-scale benchmark for assessing the generalization ability of machine learning surrogates in computational fluid dynamics \(CFD\)\. Among the problems included in CFDBench, the two\-dimensional dam\-break flow and the lid\-driven cavity flow are selected as the primary test cases in this study\.

The operating parameters in CFDBench are systematically varied along three main axes: \(i\) boundary conditions \(e\.g\., inlet or lid velocity\), \(ii\) fluid properties \(densityρ\\rhoand viscosityμ\\mu\), and \(iii\) geometry \(e\.g\., cavity dimensions, tube diameter, barrier height, or cylinder radius\)\. For each flow configuration, cases are organized into subsets corresponding to these parameter variations and their combinations, enabling rigorous evaluation of model extrapolation to unseen operating conditions\.

### 3\.2Dam\-break flow problem

Dam\-break flows constitute a canonical class of gravity\-driven free\-surface problems and are widely used as simplified models for coastal inundation and nearshore wave dynamics in ocean and coastal engineering\. The sudden release of an initially retained water column generates rapidly propagating free\-surface waves characterized by strong nonlinearity, sharp gradients, and transient flow structures\. Owing to these features, dam\-break configurations serve as challenging yet physically meaningful benchmarks for assessing numerical solvers, reduced\-order models, and inverse methodologies in regimes relevant to oceanic and coastal modeling\. In this work, the dam\-break flow is adopted to evaluate the performance of the proposed ROM–DA frameworks under strongly nonlinear dynamics\.

In CFDBench, the dam flow is parameterized by the heighthhand widthwwof the dam obstacle, the inlet velocityui​nu\_\{in\}, the fluid densityρ\\rho, and the fluid viscosityμ\\mu\. By systematically varying these parameters, the dataset spans a wide range of Reynolds numbers \(see Table[1](https://arxiv.org/html/2608.11435#S3.T1)\)\.

Table 1:Parameter ranges for the dam flow problem in CFDBench\.CategoryValuesBoundary conditionui​n∈\{0\.05,0\.1,…,1\}∪\{1\.02,1\.04,…,2\}u\_\{in\}\\in\\\{0\.05,0\.1,\\ldots,1\\\}\\cup\\\{1\.02,1\.04,\\ldots,2\\\}m/sFluid densityρ∈\{0\.1,0\.5,1,2,3,…,10\}\\rho\\in\\\{0\.1,0\.5,1,2,3,\\ldots,10\\\}kg/m3Fluid viscosityμ∈\{10−5,5×10−5,…,5×10−3,10−2\}\\mu\\in\\\{10^\{\-5\},\\,5\\times 10^\{\-5\},\\,\\ldots,\\,5\\times 10^\{\-3\},\\,10^\{\-2\}\\\}Pa⋅\\cdotsBarrier heighth∈\{0\.11,0\.12,0\.13,0\.14,0\.15\}h\\in\\\{0\.11,0\.12,0\.13,0\.14,0\.15\\\}mBarrier widthw∈\{0\.01,0\.02,…,0\.08,0\.09\}w\\in\\\{0\.01,0\.02,\\ldots,0\.08,0\.09\\\}mFor each simulation case, the velocity components\(u,v\)\(u,v\)are stored at successive time steps on a uniform64×6464\\times 64grid, yielding a total of21,70021\{,\}700snapshots across217217evolution cases\. Each snapshot is associated with a set of physical parameters and a temporal index, and can be represented as

\(𝐱t,𝜽\),𝐱t∈ℝ2×64×64,𝜽∈ℝ6,\\left\(\\mathbf\{x\}\_\{t\},\\,\\boldsymbol\{\\theta\}\\right\),\\qquad\\mathbf\{x\}\_\{t\}\\in\\mathbb\{R\}^\{2\\times 64\\times 64\},\\quad\\boldsymbol\{\\theta\}\\in\\mathbb\{R\}^\{6\},where𝐱t\\mathbf\{x\}\_\{t\}denotes the velocity field at timettand𝜽\\boldsymbol\{\\theta\}collects the five physical parameters together with the time index\.

The dataset is organized into three groups of cases:100100cases varying fluid properties\(ρ,μ\)\(\\rho,\\mu\),5050cases varying the cavity geometry\(h,w\)\(h,w\), and6767cases varying the boundary conditionui​nu\_\{in\}\. For model evaluation, data splitting is performed at the case level, ensuring that each simulation case is assigned to either the training or the test set\. This results in173173training cases \(17,30017\{,\}300snapshots\) and4444test cases \(4,4004\{,\}400snapshots\), providing a strict assessment of generalization to unseen parameter configurations\.

The first row of Figure[3](https://arxiv.org/html/2608.11435#S3.F3)illustrates representative snapshots of the horizontal and vertical velocity components in the dam\-break flow problem\. The presence of gravity\-driven jets, strong velocity gradients, flow separation around the obstacle, and free\-surface\-induced recirculation zones leads to highly transient and nonlinear flow structures\. These features make the dam\-break flow a particularly challenging benchmark for evaluating the reconstruction, prediction, and parameter calibration capabilities of the proposed ROM\-DA frameworks\.

In our numerical experiment, the inlet velocityui​nu\_\{in\}and the barrier heighthhare identified as the primary calibration parameters, while the remaining parameters are set to their true values\. Unless otherwise stated, all reported calibration results correspond to this reduced control setting\.

### 3\.3Lid\-driven cavity flow problem

The lid\-driven cavity flow is a classical benchmark in CFD and provides a controlled setting to investigate both flow reconstruction/prediction and parameter calibration\. In contrast to the dam\-break flow, which is dominated by strongly nonlinear free\-surface dynamics, the cavity flow represents a confined, wall\-bounded shear\-driven system, thereby offering a complementary test case to assess the generality and robustness of the proposed ROM–DA frameworks\. The flow is confined within a square cavity, where the top lid moves with velocityul​i​du\_\{lid\}, while the remaining walls are stationary and impose no\-slip conditions\. In CFDBench, the cavity flow is parameterized by the cavity geometry \(lengthlland widthww\), the lid velocityul​i​du\_\{lid\}, the fluid densityρ\\rho, and the fluid viscosityμ\\mu\. By systematically varying these parameters, the dataset spans a wide range of Reynolds numbers \(see Table[2](https://arxiv.org/html/2608.11435#S3.T2)\)\.

Table 2:Parameter ranges for the cavity flow problem in CFDBench\.CategoryValuesBoundary conditionul​i​d∈\{1,2,3,…,50\}u\_\{lid\}\\in\\\{1,2,3,\\ldots,50\\\}m/sFluid densityρ∈\{0\.1,0\.5,1,…,10\}\\rho\\in\\\{0\.1,0\.5,1,\\ldots,10\\\}kg/m3Fluid viscosityμ∈\{10−5,5×10−5,10−4,…,5×10−3,10−2\}\\mu\\in\\\{10^\{\-5\},\\,5\\times 10^\{\-5\},\\,10^\{\-4\},\\,\\ldots,\\,5\\times 10^\{\-3\},\\,10^\{\-2\}\\\}Pa⋅\\cdotsGeometryl,w∈\{0\.01,0\.02,0\.03,0\.04,0\.05\}l,w\\in\\\{0\.01,0\.02,0\.03,0\.04,0\.05\\\}mFor each simulation case, the velocity components\(u,v\)\(u,v\)are stored at successive time steps on a uniform64×6464\\times 64grid, yielding a total of12,68112\{,\}681snapshots across158158cases\. Each snapshot is associated with a set of physical parameters and a temporal index, and can be represented as

\(𝐱t,𝜽\),𝐱t∈ℝ2×64×64,𝜽∈ℝ6,\\left\(\\mathbf\{x\}\_\{t\},\\,\\boldsymbol\{\\theta\}\\right\),\\qquad\\mathbf\{x\}\_\{t\}\\in\\mathbb\{R\}^\{2\\times 64\\times 64\},\\quad\\boldsymbol\{\\theta\}\\in\\mathbb\{R\}^\{6\},where𝐱t\\mathbf\{x\}\_\{t\}denotes the velocity field at timettand𝜽\\boldsymbol\{\\theta\}collects the five physical parameters together with the time index\.

The dataset is organized into three groups of cases:8484cases varying fluid properties\(ρ,μ\)\(\\rho,\\mu\),2424cases varying the cavity geometry\(l,w\)\(l,w\), and5050cases varying the boundary conditionul​i​du\_\{lid\}\. For model evaluation, data splitting is performed at the case level, ensuring that each simulation case is assigned to either the training or the test set\. This results in144144training cases \(8,0368\{,\}036snapshots\) and1414test cases \(4,6454\{,\}645snapshots\), providing a strict assessment of generalization to unseen parameter configurations\.

The fifth row of Figure[3](https://arxiv.org/html/2608.11435#S3.F3)illustrates representative snapshots of the horizontal and vertical velocity components on the cavity domain\. The emergence of secondary vortices and increasingly complex flow structures at higher Reynolds numbers makes this benchmark particularly suitable for evaluating the reconstruction, prediction, and parameter calibration capabilities of ROM\-DA frameworks\.

In our numerical experiment, the lid velocityul​i​du\_\{lid\}and the fluid densityρ\\rhoare identified as the primary control parameters for calibration, while the remaining parameters are fixed at their true values\. Unless otherwise stated, all reported calibration results correspond to this reduced control setting\.

### 3\.4Reconstruction

We first evaluate the reconstruction capabilities of the three reduced\-order models introduced in Section[2\.2](https://arxiv.org/html/2608.11435#S2.SS2), namely POD, standard CAE, and OACAE\.

To provide a quantitative comparison, we evaluate the average reconstruction errors over the entire test dataset\. For fairness, the latent dimension is fixed to 32 for all three ROMs\. This value is selected as a robust and balanced choice according to the latent\-dimension sensitivity study reported in[E\.2](https://arxiv.org/html/2608.11435#A5.SS2)\. The results, summarized in Table[3](https://arxiv.org/html/2608.11435#S3.T3), are reported in terms of relativeℓ2\\ell\_\{2\}error \(REL\), mean square error \(MSE\), and structural similarity index \(SSIM\)\. These metrics highlight the contrast between the linear POD coefficients and the nonlinear latent manifolds constructed by CAE and OACAE\.

Table 3:Average reconstruction metrics of POD, CAE, and OACAE on the test datasets for the cavity flow and dam flow problems\.Cavity flowDam flowModelREL↓\\downarrowMSE↓\\downarrowSSIM↑\\uparrowREL↓\\downarrowMSE↓\\downarrowSSIM↑\\uparrowPOD0\.0024240\.0000060\.9879870\.0055860\.0000220\.976817CAE0\.0045840\.0000110\.9826910\.0077040\.0000540\.957641OACAE0\.0081110\.0000310\.9699030\.0099660\.0000690\.942009Two main observations can be drawn from Table[3](https://arxiv.org/html/2608.11435#S3.T3)\. First, the OACAE exhibits slightly lower reconstruction accuracy than the standard CAE\. This behavior is expected, since the OACAE is not entirely trained for reconstruction but also incorporates an additional regression loss to align the latent space with physical parameters\. As a result, part of the latent representation is allocated to parameter regression rather than purely minimizing reconstruction error, making the OACAE suboptimal for the reconstruction task alone\. Second, unexpectedly, POD achieves the best reconstruction accuracy\. This contrasts with existing studies on more complex flows, where CAE models usually outperform linear POD in capturing nonlinear structures\. However, the cavity flow and dam flow dataset considered here is relatively less complex, as they are represented on a uniform64×6464\\times 64grid, which reduces the spatial resolution compared with higher\-resolution datasets \(e\.g\.,128×128128\\times 128or256×256256\\times 256\)\.

Figure[14](https://arxiv.org/html/2608.11435#A3.F14)in[C](https://arxiv.org/html/2608.11435#A3)shows one test sample snapshot of the dam flow problem and cavity flow problem with the original field, the reconstructions from the three ROMs, and their error maps\.

### 3\.5Prediction

We now turn to the prediction task, where the goal is to directly infer velocity fields from physical parameters without relying on reference simulations\. In this setting, the three ROMs are employed as surrogate models: POD\-GPR, CAE\-MLP, and OACAE\-MLP\.

To quantitatively evaluate their predictive accuracy, we compute the average errors on the test set, summarized in Table[4](https://arxiv.org/html/2608.11435#S3.T4)\.

Table 4:Average prediction metrics of different predictive models on the test dataset for the cavity flow and dam flow problems\.Cavity flowDam flowModelREL↓\\downarrowMSE↓\\downarrowSSIM↑\\uparrowREL↓\\downarrowMSE↓\\downarrowSSIM↑\\uparrowPOD\-GPR0\.0613420\.0023820\.8998570\.0202750\.0001090\.944738CAE\-MLP0\.0140330\.0000790\.9520160\.0123040\.0000380\.968740OACAE\-MLP0\.0196150\.0001040\.9410410\.0140750\.0000420\.952114The same metrics as in the reconstruction study are reported: relativeℓ2\\ell\_\{2\}error \(REL\), mean square error \(MSE\), and structural similarity index \(SSIM\)\. While both reconstruction and prediction are evaluated on unseen test samples, the prediction task is inherently more difficult, as the models must infer the flow field entirely from physical parameters rather than from a latent encoding of the field itself\.

Two main conclusions can be drawn from Table[4](https://arxiv.org/html/2608.11435#S3.T4)\. First, although POD exhibits strong reconstruction capabilities in Section[3\.4](https://arxiv.org/html/2608.11435#S3.SS4), its predictive extension \(POD\-GPR\) performs markedly worse than the DL\-based surrogates, CAE\-MLP and OACAE\-MLP, across all evaluation metrics, including the REL, MSE, and SSIM\. This result highlights the superior expressiveness of deep learning ROMs for parameter\-to\-field prediction tasks and provides clear justification for their applications in more complex flow configurations\. A second observation concerns the comparison between CAE\-MLP and OACAE\-MLP\. CAE\-MLP achieves superior reconstruction accuracy, as it is solely optimized for field reconstruction\. In contrast, OACAE\-MLP introduces an additional parameter regression constraint that promotes a more structured and physically informed latent representation\. As a result, the performance gap between the two models is substantially reduced in the prediction task, indicating that the physics\-aware latent space learned by OACAE\-MLP enhances parameter\-to\-field generalization despite a slight reduction in pure reconstruction accuracy\.

Figure[3](https://arxiv.org/html/2608.11435#S3.F3)presents representative test cases from the dam flow and cavity flow datasets\. The results indicate that DL\-based surrogates not only outperform POD\-GPR in terms of average performance but also exhibit robust accuracy on individual samples where POD\-GPR fails to produce reliable predictions\.

![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_truth_u.png)\(a\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_no_error_u.png)\(b\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_truth_v.png)\(c\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_no_error_v.png)\(d\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_pod_u.png)\(e\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_pod_err_u.png)\(f\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_pod_v.png)\(g\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_pod_err_v.png)\(h\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_cae_u.png)\(i\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_cae_err_u.png)\(j\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_cae_v.png)\(k\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_cae_err_v.png)\(l\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_oacae_u.png)\(m\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_oacae_err_u.png)\(n\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_oacae_v.png)\(o\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_predictions_oacae_err_v.png)\(p\)

![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_truth_u.png)\(a\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_no_error_u.png)\(b\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_truth_v.png)\(c\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_no_error_v.png)\(d\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_pod_u.png)\(e\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_pod_err_u.png)\(f\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_pod_v.png)\(g\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_pod_err_v.png)\(h\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_cae_u.png)\(i\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_cae_err_u.png)\(j\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_cae_v.png)\(k\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_cae_err_v.png)\(l\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_oacae_u.png)\(m\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_oacae_err_u.png)\(n\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_oacae_v.png)\(o\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_predictions_oacae_err_v.png)\(p\)

Figure 3:Comparison of prediction results \(dam flow and cavity flow\)
### 3\.6Inverse modeling: Parameter calibration

We finally consider the parameter calibration problem, which aims to estimate key physical parameters from observed flow fields\. For the cavity flow, the lid velocityul​i​du\_\{lid\}and fluid densityρ\\rhoare treated as control variables, while for the dam\-break flow, the barrier heighthhand inlet velocityui​nu\_\{in\}are considered\. All remaining parameters are fixed to their ground\-truth values\.

Parameter calibration is performed using variational DA frameworks \(3D\-Var and 4D\-Var\) combined with CAE\-MLP and OACAE\-MLP surrogates\. As a baseline, an Ensemble Kalman Filter \(EnKF\) coupled with a POD\-GPR surrogate is also evaluated\. All experiments are conducted on held\-out test cases\.

For each flow configuration, a representative temporal evolution is selected\. We first examine single\-time\-step assimilation strategies, including EnKF with POD\-GPR and 3D\-Var with CAE\-MLP and OACAE\-MLP, applied over the entire evolution\. We then investigate multi\-time\-step assimilation strategies, namely 2T\-EnKF with POD\-GPR and 4D\-Var with CAE\-MLP and OACAE\-MLP, using multiple time windows extracted from the same evolution\. Since multi\-time\-step assimilation increases the number of surrogate evaluations, we further compare the online runtime of the EnKF\-POD\-GPR and the AE\-MLP 4D\-Var frameworks in[F](https://arxiv.org/html/2608.11435#A6)\.

#### 3\.6\.1Single\-time\-step assimilation\.

We first examine the performance of single\-time\-step assimilation frameworks on one dam flow and cavity flow evolution case\. The dam flow evolution case consists of 100 time steps, while the cavity flow evolution case includes 32 time steps\. The availability of complete temporal evolutions with a sufficient number of time instants allows for a reliable evaluation of single\-time\-step parameter estimation performance\. Table[5](https://arxiv.org/html/2608.11435#S3.T5)reports the average parameter estimation errors, with the background errors included for comparison\. Since all data assimilation procedures are initialized from the same prescribed background state, the comparison between analysis and background errors provides a direct measure of the correction capability of each method\. The considered approaches include the one\-step EnKF \(1T\-EnKF\) coupled with POD\-GPR, as well as 3D\-Var combined with CAE\-MLP and OACAE\-MLP surrogates\.

Table 5:Averageℓ2\\ell\_\{2\}relative errors of parameters for single\-time\-step assimilation\. Both the analysis results and the background are reported for reference\.Dam flowCavity flowMethodui​nu\_\{in\}hhul​i​du\_\{lid\}ρ\\rhoBackground0\.0978730\.6329510\.9514569\.3253861T\-EnKF \(POD–GPR\)0\.1448840\.2983310\.5635083\.3226713D\-Var \(CAE–MLP\)0\.0545950\.1670840\.5624772\.0964633D\-Var \(OACAE–MLP\)0\.0689640\.0956770\.1595652\.533160Several observations can be drawn from these results\. First, in both flow configurations, the lowest estimation errors are consistently achieved by 3D\-Var combined with DL\-based ROM frameworks\. For the dam flow problem, the inlet velocityui​nu\_\{in\}is most accurately estimated using the 3D\-Var\-CAE\-MLP framework, whereas the barrier heighthhis better recovered with 3D\-Var\-OACAE\-MLP\. In the cavity flow case, the most accurate estimates of the lid velocityul​i​du\_\{lid\}are obtained with 3D\-Var\-OACAE\-MLP, while the fluid densityρ\\rhois more accurately estimated using 3D\-Var\-CAE\-MLP\. These results demonstrate the effectiveness of the proposed variational DA\-DL\-ROM framework for parameter calibration and its advantage over the EnKF\-POD\-GPR baseline in the same task\. Second, although the 3D\-Var\-OACAE\-MLP framework exhibits slightly higher estimation errors for one of the parameters in each configuration \(namelyui​nu\_\{in\}for the dam flow andρ\\rhofor the cavity flow\), it provides substantially improved estimates for the other parameter \(hhandul​i​du\_\{lid\}, respectively\)\. Since the objective of the data assimilation procedure is to optimize the control variable vector, lower estimation errors in the analyzed parameter vector directly reflect improved calibration performance\. These results suggest that the physics\-informed latent representation learned by OACAE can enhance parameter calibration in the proposed variational\-DA\-DL\-ROM framework\.

To further investigate different inverse modeling performances between the CAE\-MLP and OACAE\-MLP, beyond the average parameter estimation errors reported in Table[5](https://arxiv.org/html/2608.11435#S3.T5), it is also essential to assess the impact of parameter calibration in the physical space\. In particular, we compare the prediction errors of the flow fields obtained using the true parameters and the calibrated \(analysis\) parameters\. To this end, we recall that the 3D\-Var cost function \([2\.3\.2](https://arxiv.org/html/2608.11435#S2.Ex4)\) consists of a prior term and an observation term\. In the present setting, the observations correspond to the true physical fields, such that the observation term directly measures the discrepancy between the predicted and true flow fields\. Moreover, because the observation term is given a large weight, its contribution dominates the cost function by several orders of magnitude over the prior term\.

\(a\)Dam flow — OACAE\-MLP\(b\)Dam flow — CAE\-MLP\(c\)Cavity flow — OACAE\-MLP\(d\)Cavity flow — CAE\-MLP
Figure 4:Comparison of 3D\-Var observation mismatch in the physical space, evaluated at each time step of the selected evolution case, using the true parameters and the calibrated \(analysis\) parameters for different surrogate models \(OACAE\-MLP and CAE\-MLP\) and flow configurations \(dam flow and cavity flow\)\.As a consequence, the effectiveness of variational data assimilation can be assessed through the 3D\-Var validation loss, defined as the value of the observation term in the 3D\-Var cost function, which provides a physically meaningful criterion for comparing different parameter estimates\. Since the observation operator is a learned surrogate rather than the full\-order model, the surrogate cannot exactly represent all spatial structures of the full\-order flow\. The inverse optimization may therefore adjust the control parameters to compensate for unresolved or distorted information in the reduced\-order model\. Consequently, the true physical parameters do not necessarily minimize the surrogate\-induced observation mismatch\. In particular, obtaining a lower validation loss with the analysis parameters than with the true parameters at a given time step serves as a necessary consistency check for the optimization procedure, indicating improved flow predictions within the surrogate\-based framework\. Figure[4](https://arxiv.org/html/2608.11435#S3.F4)illustrates the temporal evolution of the validation loss in the physical space for both flow configurations, comparing the predictions obtained using the true and calibrated parameters\. For the OACAE\-MLP framework, the validation loss associated with the analysis parameters is consistently lower than or equal to that obtained with the true parameters at all time steps, indicating robust convergence of the variational assimilation procedure\. By contrast, in the CAE\-MLP framework, although similar trends are observed at most time steps, several instances exhibit higher validation losses with the analysis parameters, suggesting a lack of convergence at specific time steps and less reliable flow predictions, which can be attributed to a less smooth latent space\. Overall, these results further demonstrate the effectiveness and enhanced robustness of the proposed physics\-informed variational DA\-DL\-ROM framework for parameter calibration\.

#### 3\.6\.2Multi\-time\-step assimilation\.

We now examine the performance of multi\-time\-step assimilation frameworks\. To ensure a consistent basis for comparison with the single\-time\-step experiments, we consider the same evolution cases from the dam flow and cavity flow datasets\. Specifically, we design ten comparative experiments between the two\-time\-step EnKF \(2T\-EnKF\) and the 4D\-Var strategies\. In each experiment, one snapshot selected from𝐱t=0\\mathbf\{x\}\_\{t=0\}to𝐱t=9\\mathbf\{x\}\_\{t=9\}is used as the first observation, while the final snapshot𝐱t=tend\\mathbf\{x\}\_\{t=t\_\{\\mathrm\{end\}\}\}is fixed as the second observation, withtend=99t\_\{\\mathrm\{end\}\}=99for the dam flow evolution andtend=31t\_\{\\mathrm\{end\}\}=31for the cavity flow evolution\. The background state for multi\-time\-step assimilation is the estimated results from the corresponding single\-time\-step assimilation\. This setup ensures that both ensemble\-based and variational methods assimilate exactly the same observational information from two time instants\. Moreover, since the first assimilation step coincides with that used in the single\-time\-step experiments, the corresponding results can be directly reused to facilitate a complementary comparison between stationary and non\-stationary data assimilation strategies\. The resulting distributions of the parameter estimates in the parameter space for both dam and cavity flow cases are illustrated in Figure[5](https://arxiv.org/html/2608.11435#S3.F5)\.

\(a\)Inverse modeling results in parameter space \(Dam flow\)\(b\)Inverse modeling results in parameter space \(Cavity flow\)
Figure 5:Distribution of calibrated parameters \(ten realizations for each method\) for the dam flow and cavity flow problems\. Scatter markers indicate the estimated\(uin,h\)\(u\_\{\\mathrm\{in\}\},h\)and\(ulid,ρ\)\(u\_\{\\mathrm\{lid\}\},\\rho\)parameter pairs obtained using different data assimilation frameworks\. The black cross denotes the background state, while the star indicates the ground\-truth parameters\. Ellipses represent the2​σ2\\sigmacovariance bounds associated with each method\. Overall, the variational DA\-DL\-ROM frameworks yield more concentrated parameter estimates around the target than the baseline approaches\. In particular, the 3D\-Var\-OACAE\-MLP and 4D\-Var\-OACAE\-MLP strategies exhibit the tightest distributions around the reference values for both flow configurations\.The 4D\-Var procedure is initialized following the 3D\-Var analysis and incorporates additional observational information over multiple time instants\. As a result, the corresponding analysis is expected to achieve improved predictive capability in the physical space\. To assess this expected advantage of 4D\-Var over 3D\-Var, we compare the 4D\-Var validation losses \(observation term in the 4D\-Var cost function \([2\.3\.2](https://arxiv.org/html/2608.11435#S2.Ex5)\)\) obtained using the analysis parameters produced by the two methods\.

\(a\)Dam flow — OACAE\-MLP\(b\)Dam flow — CAE\-MLP\(c\)Cavity flow — OACAE\-MLP\(d\)Cavity flow — CAE\-MLP
Figure 6:Comparison of 4D\-Var observation mismatch in the physical space, evaluated at selected 4D\-Var assimilation time steps, using the true parameters and the analysis parameters from 4D\-Var and 3D\-Var methods for different surrogate models \(OACAE\-MLP and CAE\-MLP\) and flow configurations \(dam flow and cavity flow\)\.Figure[6](https://arxiv.org/html/2608.11435#S3.F6)compares the 4D\-Var validation losses obtained using the true parameters and the analysis parameters from 3D\-Var and 4D\-Var for different surrogate models and flow configurations\. For the OACAE\-MLP surrogate, the 4D\-Var framework consistently achieves lower or comparable validation losses than 3D\-Var for both dam and cavity flows, demonstrating that incorporating temporal consistency effectively improves physical\-space predictions when a physics\-informed latent representation is employed\. By contrast, for the CAE\-MLP surrogate, although 4D\-Var outperforms 3D\-Var in most cases, noticeable degradation is observed for certain test windows, particularly in the cavity flow configuration, indicating reduced robustness\. Moreover, in the majority of tests, the validation losses obtained with the 4D\-Var analysis parameters are lower than or equal to those obtained with the true parameters, highlighting the strong capability of the proposed inverse modeling framework to achieve physically consistent parameter calibration\.

While the validation losses reported above evaluate the physical\-space consistency at assimilation time steps, the ultimate objective of parameter calibration is to improve long\-term prediction of flow dynamics\. To this end, we further assess the predictive performance of the calibrated parameters over the entire temporal evolution\. Figure[7](https://arxiv.org/html/2608.11435#S3.F7)reports the meanℓ2\\ell\_\{2\}relative errors with standard deviations computed over all time steps of the evolution for different calibration strategies\. For each estimate, the discrepancy of its predicted trajectory with the reference is averaged over all time steps, yielding a single error value per estimate\. The long\-term predictive performance of the calibrated parameters is evaluated based on two complementary criteria: prediction robustness and prediction accuracy\. Robustness is assessed through the uncertainty of the prediction errors, quantified by the standard deviation across different calibration instances, while accuracy is evaluated by comparing the mean prediction error with that obtained using the true parameters\. As shown in Figure[7](https://arxiv.org/html/2608.11435#S3.F7), the variational DA\-DL\-ROM frameworks, particularly when combined with the OACAE\-MLP surrogate, consistently exhibit low prediction uncertainty for both dam and cavity flow configurations\. Notably, for each flow configuration, the uncertainty associated with the 3D\-Var\-OACAE\-MLP framework is even lower than that of the 4D\-Var\-CAE\-MLP framework, highlighting the strong impact of physics\-aware latent representations on stabilizing the assimilation process\. In terms of accuracy, the prediction errors obtained using the calibrated parameters are either lower than or comparable to those obtained using true parameters\. For the dam flow configuration, the 4D\-Var\-OACAE\-MLP framework achieves lower prediction errors than the reference level, whereas for the cavity flow configuration, the corresponding errors remain close to those obtained with the true parameters\. In both cases, the OACAE\-based predictions are consistently closer to the reference than their CAE\-based counterpart\. A quantitative comparison further highlights this difference\. For the dam flow, the relative deviations with respect to the true prediction are 2\.80% and 2\.10% for the OACAE\-based framework using 4D\-Var and 3D\-Var, whereas the corresponding deviations for the CAE\-based framework reach 6\.77% for 4D\-Var and 18\.05% for 3D\-Var\. For the cavity flow, the OACAE\-based framework yields relative deviations of 1\.79% for 4D\-Var and 20\.54% for 3D\-Var, while the CAE\-based framework exhibits substantially larger deviations of 15\.38% for 4D\-Var and 141\.54% for 3D\-Var\. By contrast, the EnKF\-POD\-GPR baseline exhibits substantially larger prediction errors and uncertainties, highlighting its limited robustness for long\-term prediction, particularly in the cavity flow configuration where the 2T\-EnKF strategy leads to increased errors over the entire evolution\. Overall, these results demonstrate that the proposed variational DA\-DL\-ROM framework not only improves parameter estimation but also leads to accurate and robust long\-term predictions in physical space\.

\(a\)Entire evolution prediction error \(Dam flow\)\(b\)Entire evolution prediction error \(Cavity flow\)
Figure 7:Mean relativeℓ2\\ell\_\{2\}errors in the physical space over the entire temporal evolution using parameters estimated by different data assimilation frameworks\.To further illustrate the physical impact of parameter calibration, in Figure[8](https://arxiv.org/html/2608.11435#S3.F8), we visualize representative flow fields obtained using the background parameters, the parameters calibrated by 3D\-Var and 4D\-Var, and the corresponding true parameter with the OACAE\-MLP model\.

![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_bg_u.png)\(a\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_bg_err_u.png)\(b\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_bg_v.png)\(c\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_bg_err_v.png)\(d\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_3d_u.png)\(e\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_3d_err_u.png)\(f\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_3d_v.png)\(g\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_3d_err_v.png)\(h\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_4d_u.png)\(i\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_4d_err_u.png)\(j\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_4d_v.png)\(k\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_4d_err_v.png)\(l\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_tr_u.png)\(m\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_tr_error_u.png)\(n\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_tr_v.png)\(o\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_da_tr_err_v.png)\(p\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_bg_u.png)\(q\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_bg_err_u.png)\(r\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_bg_v.png)\(s\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_bg_err_v.png)\(t\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_3d_u.png)\(u\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_3d_err_u.png)\(v\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_3d_v.png)\(w\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_3d_err_v.png)\(x\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_4d_u.png)\(y\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_4d_err_u.png)\(z\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_4d_v.png)\(aa\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_4d_err_v.png)\(ab\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_tr_u.png)\(ac\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_tr_error_u.png)\(ad\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_tr_v.png)\(ae\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_da_tr_err_v.png)\(af\)

Figure 8:
#### 3\.6\.3Robustness to different evolution cases\.

As discussed in Section[3\.6\.1](https://arxiv.org/html/2608.11435#S3.SS6.SSS1), the effectiveness of the proposed physics\-informed variational DA\-DL\-ROM framework for parameter calibration is first demonstrated using all time steps from a fixed evolution case in the test dataset\. To further assess the robustness of the proposed framework across different evolution cases, we perform an additional sensitivity analysis on the dam flow test dataset\. The dam flow test dataset contains 44 distinct evolution cases, each consisting of 100 time steps, which provides a suitable basis for evaluating case\-to\-case robustness\. For each evolution case, a single snapshot at the same time index \(the 50th time step\) is selected for assimilation, and the same single\-time\-step data assimilation procedures are applied without any modification to the data assimilation settings or surrogate models\. As a result, each framework produces 44 independent parameter estimates, enabling consistent comparison of robustness across different evolution cases\.

Figure[9](https://arxiv.org/html/2608.11435#S3.F9)shows the calibrated parameter values for each case and each framework, together with the corresponding ground\-truth and background values for reference\. Each marker corresponds to the parameter estimate obtained from an evolution case using single\-time\-step data assimilation\. The horizontal dashed line indicates the background value, while the circle markers denote the ground\-truth parameters\. Results obtained with different data assimilation frameworks are shown for comparison\. The variational DA\-DL\-ROM approaches yield more concentrated and consistent parameter estimates across cases, whereas larger dispersion is observed for the EnKF\-POD\-GPR baseline, indicating improved case\-to\-case robustness of the proposed framework\.

Figure 9:Distribution of calibrated parameters across different evolution cases for the dam flow problem\.Table[6](https://arxiv.org/html/2608.11435#S3.T6)reports the average parameter estimation errors together with their standard deviations over all 44 test cases\.

Table 6:Parameter estimation errors for the dam flow problem using single\-time\-step data assimilation across 44 evolution cases\. The results show that the variational DA\-DL\-ROM frameworks yield significantly lower estimation errors and reduced parameter\-space variability compared with the EnKF\-POD\-GPR baseline\. In particular, the 3D\-Var\-OACAE\-MLP approach achieves the lowest errors and the smallest uncertainty for both parameters, indicating improved robustness of parameter calibration across different evolution cases\.ui​nu\_\{in\}hhMethodMSE↓\\downarrowSTD↓\\downarrowMSE↓\\downarrowSTD↓\\downarrow1T\-EnKF\-POD\-GPR8\.70×10−38\.70\\times 10^\{\-3\}9\.18×10−29\.18\\times 10^\{\-2\}2\.24×10−22\.24\\times 10^\{\-2\}1\.49×10−11\.49\\times 10^\{\-1\}3D\-Var\-CAE\-MLP6\.58×10−36\.58\\times 10^\{\-3\}7\.86×10−27\.86\\times 10^\{\-2\}2\.58×10−32\.58\\times 10^\{\-3\}4\.98×10−24\.98\\times 10^\{\-2\}3D\-Var\-OACAE\-MLP4\.66×𝟏𝟎−𝟒\\mathbf\{4\.66\\times 10^\{\-4\}\}2\.13×𝟏𝟎−𝟐\\mathbf\{2\.13\\times 10^\{\-2\}\}2\.04×𝟏𝟎−𝟒\\mathbf\{2\.04\\times 10^\{\-4\}\}1\.40×𝟏𝟎−𝟐\\mathbf\{1\.40\\times 10^\{\-2\}\}While the parameter\-space statistics provide a direct measure of calibration accuracy, it is still necessary to examine whether the observed case\-to\-case robustness is preserved in the physical space\. To this end, Figure[10](https://arxiv.org/html/2608.11435#S3.F10)reports the physical\-space prediction errors obtained using the calibrated parameters for each evolution case\. For each method, the calibrated parameters are used to generate predictions over the entire temporal evolution of that case\. The resulting prediction errors are then computed and averaged over time in the physical space and summarized using their mean and standard deviation across all 44 evolution cases\.

Figure 10:Case\-to\-case physical\-space prediction errors \(relativeℓ2\\ell\_\{2\}error\) for the dam flow problem\. All data assimilation frameworks reduce both the prediction error and its uncertainty compared with the background\. In particular, the variational DA\-DL\-ROM approaches yield lower average errors and reduced uncertainty than the EnKF\-POD\-GPR baseline\. Moreover, the 3D\-Var\-OACAE\-MLP framework achieves even lower prediction error and reduced uncertainty than that obtained using the true parameters\.Across the tested evolution cases, consistent trends are observed\. In particular, the variational DA–DL–ROM frameworks continue to provide more stable parameter estimates and improved physical\-space predictions compared with the EnKF\-POD\-GPR baseline\. Moreover, the OACAE\-MLP surrogate exhibits greater robustness than the CAE\-MLP surrogate\. These results indicate that the main conclusions drawn above are not sensitive to the choice of test case\.

### 3\.7Robustness under degraded\-observation settings

To evaluate the robustness of the proposed framework for inverse problem under more realistic measurement conditions, we consider four degraded\-observation scenarios for the dam flow example: noisy observations, low\-resolution observations, randomly masked observations, and block\-wise partial observations\. These scenarios represent cases where the available observations are corrupted, spatially coarsened, or incomplete\.

For the noisy\-observation experiment, additive Gaussian noise is applied to the full\-field observation\. For each sample, a random noise fieldε\\varepsilonis generated and rescaled to satisfy

‖ε‖2‖y‖2=η,\\frac\{\\\|\\varepsilon\\\|\_\{2\}\}\{\\\|y\\\|\_\{2\}\}=\\eta,where

η∈\{0,0\.01,0\.05,0\.10,0\.20\}\\eta\\in\\\{0,0\.01,0\.05,0\.10,0\.20\\\}denotes the prescribed relative noise level\. The noisy observation is then defined as

yη=y\+ε\.y^\{\\eta\}=y\+\\varepsilon\.
For the low\-resolution observation experiment, the full\-resolution field is downsampled using average pooling over non\-overlapping spatial blocks\. Given a downsampling factorss, the low\-resolution observation is defined as

y\(s\)=Ps​\(y\),y^\{\(s\)\}=P\_\{s\}\(y\),wherePsP\_\{s\}denotes the average\-pooling operator with kernel size and stride equal toss\. We consider

s∈\{1,2,4,8,16\},s\\in\\\{1,2,4,8,16\\\},corresponding to observation resolutions

64×64,32×32,16×16,8×8,4×4,64\\times 64,\\quad 32\\times 32,\\quad 16\\times 16,\\quad 8\\times 8,\\quad 4\\times 4,respectively\. Note thats=1s=1corresponds to the original full\-field observation\.

For the randomly masked\-observation experiment, binary spatial masks are applied to the full\-resolution field\. The mask is defined over spatial locations and is shared by both physical channels, so that either both velocity components are observed or both are masked at a given grid point\. LetMρ∈\{0,1\}H×WM\_\{\\rho\}\\in\\\{0,1\\\}^\{H\\times W\}denote a binary spatial mask, whereρ\\rhois the masked ratio\. The partial observation is written as

yρ=Mρ​\(y\),y^\{\\rho\}=M\_\{\\rho\}\(y\),whereMρM\_\{\\rho\}extracts the observed entries fromyyaccording to the mask\. We use

ρ∈\{0,18,38,58,78\},\\rho\\in\\left\\\{0,\\frac\{1\}\{8\},\\frac\{3\}\{8\},\\frac\{5\}\{8\},\\frac\{7\}\{8\}\\right\\\},corresponding to observed ratios

100%,87\.5%,62\.5%,37\.5%,12\.5%\.100\\%,\\quad 87\.5\\%,\\quad 62\.5\\%,\\quad 37\.5\\%,\\quad 12\.5\\%\.The random masks are generated with a fixed seed and are nested, so that higher masked ratios remove a superset of the grid points removed at lower masked ratios\.

For the block\-wise partial observation experiment, the full resolution spatial domain is divided into four non\-overlapping blocks of equal size using a2×22\\times 2partition\. At each time snapshot, only one block is retained as the observation, while the remaining three blocks are unobserved\. The same block mask is applied to both physical channels\. Specifically, we consider

yb=Bb​\(y\),b∈\{A,B,C,D\},y^\{b\}=B\_\{b\}\(y\),\\qquad b\\in\\\{A,B,C,D\\\},whereBbB\_\{b\}extracts the entries ofyylocated in blockbb\. The four blocks correspond to the lower\-left, lower\-right, upper\-left, and upper\-right quadrants of the spatial domain, respectively\. In addition, the full\-field observation is included as a reference case\. Therefore, the tested block\-observation cases are

Full,A,B,C,D\.\\mathrm\{Full\},\\quad A,\\quad B,\\quad C,\\quad D\.
Figure[11](https://arxiv.org/html/2608.11435#S3.F11)shows representative examples of the different observation settings, including clean full\-field observations, noisy observations, low\-resolution observations, randomly masked observations, and block\-wise partial observations\.

![Refer to caption](https://arxiv.org/html/2608.11435v1/fig_observation_operator_overview.png)Figure 11:Visualization of different observation settings, including clean full\-field observations, noisy observations, low\-resolution observations, random masked observations, and partial observations\.For each degraded\-observation setting, the observation operator in the variational cost function is modified consistently with the available observation\. More precisely, letH⁡\(θ\)H\(\\theta\)denote the full\-field surrogate prediction obtained from CAE\-MLP or OACAE\-MLP\. For a given observation transformationGG, the observation term is evaluated as

Jo​\(θ\)=12​\(G⁡\(y\)−G⁡\(H⁡\(θ\)\)\)T​RG−1​\(G⁡\(y\)−G⁡\(H⁡\(θ\)\)\),J\_\{o\}\(\\theta\)=\\frac\{1\}\{2\}\\left\(G\(y\)\-G\(H\(\\theta\)\)\\right\)^\{T\}R\_\{G\}^\{\-1\}\\left\(G\(y\)\-G\(H\(\\theta\)\)\\right\),whereGGis the identity operator for full\-field and noisy observations, the average\-pooling operatorPsP\_\{s\}for low\-resolution observations, the masking operatorMρM\_\{\\rho\}for randomly masked observations, and the block operatorBbB\_\{b\}for block\-wise partial observations\. The covariance matrixRGR\_\{G\}is defined in the corresponding observation space, with a dimension consistent with the number of observed entries\.

Figure[12](https://arxiv.org/html/2608.11435#S3.F12)reports the resulting parameter\-calibration errors under the four degraded\-observation scenarios\. In most degraded\-observation settings, the OACAE\-MLP framework achieves lower calibration errors than the CAE\-MLP framework\. Exceptions are observed in several highly degraded cases, including low\-resolution observations at16×1616\\times 16,8×88\\times 8, and4×44\\times 4, randomly masked observations with only12\.5%12\.5\\%observed entries, and the block\-observation settingDD\. In these cases, the difference between CAE\-MLP and OACAE\-MLP becomes less pronounced, which is expected when the available observations contain limited parameter\-sensitive information\. More importantly, the OACAE\-MLP framework exhibits lower variability across all observation settings, as indicated by the smaller standard deviations\. Overall, both CAE\-MLP and OACAE\-MLP show robustness under degraded\-observation settings, demonstrating the stability of the proposed DL\-ROM\-DA framework itself\. Within this robust framework, OACAE\-MLP further improves the calibration accuracy and reduces the variability of the inverse estimates\.

\(a\)Noisy observations\(b\)Low\-resolution observations\(c\)Masked observations\(d\)Block observations
Figure 12:Parameter calibration performances under degraded\-observation settings\. For each setting, inverse modeling performances of CAE\-MLP and OACAE\-MLP are compared using 3D\-Var\. The curves report the mean error over multiple observations, and the shaded regions indicate standard deviation\.

## 4Conclusion and Future work

This work aims to address the challenge of forward prediction and inverse parameter calibration in high\-dimensional parametric dynamical systems\. This requires a reduced\-order surrogate that not only accurately reconstructs or predicts flow fields but also preserves parameter\-relevant information essential for inverse modeling\. To this end, we developed an end\-to\-end differentiable physics\-aware latent\-space framework that combines observable\-augmented autoencoder\-based reduced\-order modeling with variational data assimilation\.

The numerical results on the CFDbench benchmark show that reconstruction accuracy alone is not sufficient to determine the quality of a surrogate model for inverse problems\. Although POD achieves the best reconstruction accuracy, its reconstruction\-optimal projection basis may not preserve the parameter\-informative directions needed for inverse calibration\. In contrast, the observable\-augmented autoencoder slightly sacrifices pure reconstruction accuracy, but provides a latent representation that is better aligned with the physical quantities involved in the inverse problem\. More precisely, when integrated into a variational data assimilation framework, the observable\-augmented model yields low physical\-space prediction errors and reduced variability across calibration instances\. These advantages persist under noisy, low\-resolution, randomly masked, and block\-wise partial observations\.

Since the present method relies on a deterministic surrogate, the reported parameter errors reflect the combined effects of the potential ill\-posedness of the inverse problem, observation uncertainty, and surrogate\-model discrepancy\. A complete decomposition of surrogate\-induced bias and a systematic uncertainty quantification of the calibrated parameters are therefore left for future work\[[60](https://arxiv.org/html/2608.11435#bib.bib56),[49](https://arxiv.org/html/2608.11435#bib.bib57)\]\. Moreover, since the numerical experiments mainly assess model performance within the available data distribution, reliable extrapolation is generally not guaranteed\. Future work will address these limitations by extending the framework to more realistic experimental data and by developing uncertainty\-aware and probabilistic variants of the proposed framework\[[27](https://arxiv.org/html/2608.11435#bib.bib43),[51](https://arxiv.org/html/2608.11435#bib.bib50)\]\. Potential directions include probabilistic manifold learning, Bayesian reduced\-order modeling, and ensemble\-based data\-assimilation strategies\. Further applications to higher\-Reynolds\-number turbulent flows, higher\-resolution simulations, structural mechanics\[[7](https://arxiv.org/html/2608.11435#bib.bib19),[35](https://arxiv.org/html/2608.11435#bib.bib59)\], and fluid\-structure interaction problems will also be considered\. Finally, quantum neural networks\[[57](https://arxiv.org/html/2608.11435#bib.bib51),[34](https://arxiv.org/html/2608.11435#bib.bib52),[21](https://arxiv.org/html/2608.11435#bib.bib53)\]and hybrid quantum–classical learning architectures may be investigated as a longer\-term direction for physics\-informed reduced\-order modeling and inverse problems, although their integration into the present data\-assimilation framework would require a dedicated methodological study\.

## Data and code availability

## Author contributions

Qiyao Zhou: Methodology, Software, Validation, Formal analysis, Investigation, Writing \- Original Draft, Visualization

Xujia Zhu: Conceptualization, Methodology, Formal analysis, Investigation, Writing \- Review and Editing, Visualization

Pierre Joli: Resources, Supervision, Formal analysis, Writing \- Review and Editing

Yu Cong: Resources, Supervision, Writing \- Review and Editing

Sibo Cheng: Project administration, Methodology, Software, Resources, Supervision, Formal analysis, Writing \- Review and Editing

## Acknowledgement

Qiyao gratefully acknowledges the UnivEvry, Université Paris\-Saclay, for support through a Ph\.D\. scholarship\. Sibo Cheng acknowledges the support of the French Agence Nationale de la Recherche \(ANR\) under reference ANR\-22\-CPJ2\-0143\-01\.

## References

- \[1\]\(2022\)Deep neural network and yuki algorithm for inner damage characterization based on elastic boundary displacement\.InInternational Conference of Steel and Composite for Engineering Structures,pp\. 220–233\.Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p1.1)\.
- \[2\]M\. Asch, M\. Bocquet, and M\. Nodet\(2016\)Data assimilation: methods, algorithms, and applications\.Society for Industrial and Applied Mathematics,Philadelphia, PA\.External Links:[Document](https://dx.doi.org/10.1137/1.9781611974546)Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[3\]R\. N\. Asl, Y\. Yamazaki, K\. Taghikhani,et al\.\(2025\)A physics\-informed meta\-learning framework for the continuous solution of parametric pdes on arbitrary geometries\.arXiv preprint arXiv:2504\.02459\.Cited by:[§1\.1](https://arxiv.org/html/2608.11435#S1.SS1.p3.1)\.
- \[4\]G\. K\. Batchelor\(1967\)An introduction to fluid dynamics\.Cambridge University Press\.Cited by:[§1\.1](https://arxiv.org/html/2608.11435#S1.SS1.p4.3)\.
- \[5\]B\. Benaissa, N\. A\. Hocine, S\. Khatir,et al\.\(2021\)YUKI algorithm and pod\-rbf for elastostatic and dynamic crack identification\.J\. Comput\. Sci\.55,pp\. 101451\.External Links:ISSN 1877\-7503,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.jocs.2021.101451),[Link](https://www.sciencedirect.com/science/article/pii/S1877750321001319)Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p1.1)\.
- \[6\]M\. Bocquet, J\. Brajard, A\. Carrassi,et al\.\(2020\)Bayesian inference of chaotic dynamics by merging data assimilation, machine learning and expectation\-maximization\.Found\. Data Sci\.2\(1\),pp\. 55–80\.External Links:ISSN 2639\-8001,[Link](http://dx.doi.org/10.3934/fods.2020004),[Document](https://dx.doi.org/10.3934/fods.2020004)Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[7\]S\. Brivio, S\. Fresca, and A\. Manzoni\(2025\)Handling geometrical variability in nonlinear reduced order modeling through continuous geometry\-aware dl\-roms\.Comput\. Methods Appl\. Mech\. Eng\.442,pp\. 117989\.Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p3.1),[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[8\]S\. Burela, P\. Krah, and J\. Reiss\(2025\)Parametric model order reduction for a wildland fire model via the shifted pod\-based deep learning method\.Adv\. Comput\. Math\.51\(1\),pp\. 9\.Cited by:[§1\.1](https://arxiv.org/html/2608.11435#S1.SS1.p3.1)\.
- \[9\]A\. Carrassi, M\. Bocquet, L\. Bertino,et al\.\(2018\)Data assimilation in the geosciences: an overview of methods, issues, and perspectives\.WIREs Clim\. Chang\.9\(5\),pp\. e535\.External Links:[Document](https://dx.doi.org/10.1002/wcc.535)Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[10\]S\. Cheng, J\. Chen, C\. Anastasiou, P\. Angeli, O\. K\. Matar, Y\. Guo, C\. C\. Pain, and R\. Arcucci\(2023\)Generalised latent assimilation in heterogeneous reduced spaces with machine learning surrogate models\.Journal of Scientific Computing94\(1\),pp\. 11\.Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[11\]S\. Cheng, Y\. Jin, S\. P\. Harrison,et al\.\(2022\)Parameter flexible wildfire prediction using machine learning techniques: forward and inverse modelling\.Remote Sens\.14\(13\)\.External Links:[Link](https://www.mdpi.com/2072-4292/14/13/3228),ISSN 2072\-4292,[Document](https://dx.doi.org/10.3390/rs14133228)
- \[12\]S\. Cheng, C\. Liu, Y\. Guo,et al\.\(2024\)Efficient deep data assimilation with sparse observations and time\-varying sensors\.J\. Comput\. Phys\.496,pp\. 112581\.External Links:ISSN 0021\-9991,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.jcp.2023.112581),[Link](https://www.sciencedirect.com/science/article/pii/S0021999123006769)
- \[13\]S\. Cheng, J\. Min, C\. Liu,et al\.\(2025\)TorchDA: a python package for performing data assimilation with deep learning forward and transformation functions\.Comput\. Phys\. Commun\.306,pp\. 109359\.External Links:ISSN 0010\-4655,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.cpc.2024.109359),[Link](https://www.sciencedirect.com/science/article/pii/S0010465524002820)Cited by:[§2\.3\.2](https://arxiv.org/html/2608.11435#S2.SS3.SSS2.p1.1)\.
- \[14\]S\. Cheng, C\. Quilodrán\-Casas, S\. Ouala,et al\.\(2023\)Machine learning with data assimilation and uncertainty quantification for dynamical systems: a review\.IEEE/CAA J\. Autom\. Sin\.10\(6\),pp\. 1361–1387\.External Links:[Document](https://dx.doi.org/10.1109/JAS.2023.123537)Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[15\]S\. Cheng, Y\. Zhuang, L\. Kahouadji, C\. Liu, J\. Chen, O\. K\. Matar, and R\. Arcucci\(2024\)Multi\-domain encoder–decoder neural networks for latent data assimilation in dynamical systems\.Computer Methods in Applied Mechanics and Engineering430,pp\. 117201\.Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[16\]C\. Dai, D\. Yang, C\. Zhang,et al\.\(2025\)Parameter identification of fluid field based on cfd reduced\-order model and 3d\-var data assimilation\.Ann\. Nucl\. Energy219,pp\. 111459\.External Links:ISSN 0306\-4549,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.anucene.2025.111459),[Link](https://www.sciencedirect.com/science/article/pii/S0306454925002762)Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p2.1),[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[17\]J\. de Frutos and J\. Novo\(2000\)A spectral element method for the navier–stokes equations with improved accuracy\.SIAM J\. Numer\. Anal\.38\(3\),pp\. 799–819\.External Links:[Document](https://dx.doi.org/10.1137/S0036142999351984),[Link](https://doi.org/10.1137/S0036142999351984),https://doi\.org/10\.1137/S0036142999351984
- \[18\]S\. Dealbera, S\. Raynaud, C\. Granero Belinchon,et al\.\(2025\)The advantages of data assimilation in parametric space rather than classic grid space\.EGUsphere2025,pp\. 1–8\.External Links:[Link](https://egusphere.copernicus.org/preprints/2025/egusphere-2025-5907/),[Document](https://dx.doi.org/10.5194/egusphere-2025-5907)Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[19\]S\. Deshpande, H\. Rappel, M\. Hobbs,et al\.\(2025\)Gaussian process regression\+ deep neural network autoencoder for probabilistic surrogate modeling in nonlinear mechanics of solids\.Comput\. Methods Appl\. Mech\. Eng\.437,pp\. 117790\.
- \[20\]A\. Ern and J\. Guermond\(2004\)Theory and practice of finite elements\.
- \[21\]A\. Farea, S\. Khan, and M\. S\. Celebi\(2025\)QCPINN: quantum\-classical physics\-informed neural networks for solving pdes\.External Links:2503\.16678,[Link](https://arxiv.org/abs/2503.16678)Cited by:[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[22\]N\. Franco, A\. Manzoni, and P\. Zunino\(2022\)A deep learning approach to reduced order modelling of parameter dependent partial differential equations\.Math\. Comput\.92\(340\),pp\. 483–524\.External Links:ISSN 0025\-5718,[Link](http://dx.doi.org/10.1090/mcom/3781),[Document](https://dx.doi.org/10.1090/mcom/3781)Cited by:[§E\.2](https://arxiv.org/html/2608.11435#A5.SS2.p2.1)\.
- \[23\]S\. Fresca, L\. Dede’, and A\. Manzoni\(2021\)A comprehensive deep learning\-based approach to reduced order modeling of nonlinear time\-dependent parametrized pdes\.J\. Sci\. Comput\.87\(2\),pp\. 61\.Cited by:[§1\.1](https://arxiv.org/html/2608.11435#S1.SS1.p3.1),[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p1.1)\.
- \[24\]K\. Fukami and K\. Taira\(2023\)Grasping extreme aerodynamics on a low\-dimensional manifold\.Nat\. Commun\.14\(1\),pp\. 6480\.Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p4.1)\.
- \[25\]K\. Fukami and K\. Taira\(2025\)Observable\-augmented manifold learning for multi\-source turbulent flow data\.J\. Fluid Mech\.1010,pp\. R4\.External Links:[Document](https://dx.doi.org/10.1017/jfm.2025.383)Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p4.1),[§2\.2\.2](https://arxiv.org/html/2608.11435#S2.SS2.SSS2.p1.1)\.
- \[26\]H\. Gao, S\. Kaltenbach, and P\. Koumoutsakos\(2025\)Generative learning of the solution of parametric partial differential equations using guided diffusion models and virtual observations\.Comput\. Methods Appl\. Mech\. Eng\.435,pp\. 117654\.
- \[27\]J\. Guo and D\. Xiao\(2026\)Parametric probabilistic manifold decomposition for nonlinear model reduction\.arXiv preprint arXiv:2601\.07278\.Cited by:[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[28\]J\. Harrison and E\. Yeung\(2025\)Estimating varying parameters in dynamical systems: a modular framework using switch detection, optimization, and sparse regression\.arXiv preprint arXiv:2412\.16198\.
- \[29\]R\. Huang, C\. Heaney, and M\. Van Reeuwijk\(2025\)Parameter estimation for land\-surface models using machine learning libraries\.pp\.\.External Links:[Document](https://dx.doi.org/10.48550/arXiv.2505.02979)
- \[30\]B\. R\. Hunt, E\. J\. Kostelich, and I\. Szunyogh\(2007\)Efficient data assimilation for spatiotemporal chaos: a local ensemble transform kalman filter\.Phys\. D: Nonlinear Phenom\.230\(1\),pp\. 112–126\.Note:Data AssimilationExternal Links:ISSN 0167\-2789,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.physd.2006.11.008),[Link](https://www.sciencedirect.com/science/article/pii/S0167278906004647)
- \[31\]R\. E\. Kalman\(1960\)A new approach to linear filtering and prediction problems\.
- \[32\]B\. Koc, S\. Rubino, T\. C\. Rebollo,et al\.\(2022\)Residual data\-driven variational multiscale reduced order models for parameter dependent problems\.arXiv preprint arXiv:2208\.00059\.
- \[33\]L\. D\. Landau and E\. M\. Lifshitz\(1987\)Fluid mechanics\.Pergamon Press\.Cited by:[§1\.1](https://arxiv.org/html/2608.11435#S1.SS1.p4.3)\.
- \[34\]H\. Li, Y\. Lou, and D\. Xiao\(2025\)Quantum machine learning for efficient reduced order modelling of turbulent flows\.External Links:2511\.18552,[Link](https://arxiv.org/abs/2511.18552)Cited by:[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[35\]X\. Li, H\. Bolandi, M\. Masmoudi,et al\.\(2024\)Mechanics\-informed autoencoder enables automated detection and localization of unforeseen structural damage\.External Links:2402\.15492,[Link](https://arxiv.org/abs/2402.15492)Cited by:[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[36\]Z\. Li, B\. Dong, and P\. Zhang\(2025\)State\-observation augmented diffusion model for nonlinear assimilation with unknown dynamics\.J\. Comput\. Phys\.,pp\. 114240\.
- \[37\]C\. Liu, R\. Fu, D\. Xiao,et al\.\(2022\)Enkf data\-driven reduced order assimilation system\.Eng\. Anal\. Bound\. Elem\.139,pp\. 46–55\.Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[38\]J\. Liu, Z\. Wang, Q\. Li,et al\.\(2025\)A hybrid data assimilation and dynamic mode decomposition approach for xenon dynamic prediction of nuclear reactor cores\.Nucl\. Sci\. Eng\.199\(6\),pp\. 888–906\.Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p2.1)\.
- \[39\]X\. Liu, Z\. Wang, H\. Ji,et al\.\(2024\)Application and comparison of several adaptive sampling algorithms in reduced order modeling\.Heliyon10\(15\)\.Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1)\.
- \[40\]E\. Lumet, M\. C\. Rochoux, T\. Jaravel,et al\.\(2025\)Uncertainty\-aware surrogate modeling for urban air pollutant dispersion prediction\.Build\. Environ\.267,pp\. 112287\.External Links:ISSN 0360\-1323,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.buildenv.2024.112287),[Link](https://www.sciencedirect.com/science/article/pii/S0360132324011296)Cited by:[Appendix A](https://arxiv.org/html/2608.11435#A1.p2.1),[Appendix A](https://arxiv.org/html/2608.11435#A1.p3.1),[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p2.1),[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1),[§2\.2\.1](https://arxiv.org/html/2608.11435#S2.SS2.SSS1.p1.1),[§2\.2\.1](https://arxiv.org/html/2608.11435#S2.SS2.SSS1.p3.1),[§3](https://arxiv.org/html/2608.11435#S3.p1.1)\.
- \[41\]E\. Lumet, M\. C\. Rochoux, T\. Jaravel,et al\.\(2026\)Surrogate\-based ensemble data assimilation for reducing uncertainty in large\-eddy simulation of microscale pollutant dispersion\.Build\. Environ\.287,pp\. 113863\.External Links:ISSN 0360\-1323,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.buildenv.2025.113863),[Link](https://www.sciencedirect.com/science/article/pii/S0360132325013332)Cited by:[§1\.3](https://arxiv.org/html/2608.11435#S1.SS3.p2.1),[§2\.3\.1](https://arxiv.org/html/2608.11435#S2.SS3.SSS1.p2.1),[§3](https://arxiv.org/html/2608.11435#S3.p1.1)\.
- \[42\]M\. Luo and J\. Wu\(2026\)Fractal invariance\-constrained deep learning for spatial\-temporal prediction of turbulent flows\.Appl\. Math\. Model\.,pp\. 116754\.
- \[43\]M\. Luo, S\. Zhong, J\. Wu,et al\.\(2026\)A hybrid conv\-lstm network with skip connections for nonlinear reduced\-order modeling of spatiotemporal flow fields\.Ocean Eng\.347,pp\. 124033\.
- \[44\]Y\. Luo, Y\. Chen, and Z\. Zhang\(2023\)CFDBench: a large\-scale benchmark for machine learning methods in fluid dynamics\.arXiv preprint arXiv:2310\.05963\.Cited by:[§3\.1](https://arxiv.org/html/2608.11435#S3.SS1.p1.1)\.
- \[45\]B\. Melinc and Ž\. Zaplotnik\(2024\)3D\-var data assimilation using a variational autoencoder\.Q\. J\. R\. Meteorol\. Soc\.150\(761\),pp\. 2273–2295\.
- \[46\]A\. T\. Mohan and D\. V\. Gaitonde\(2018\)A deep learning based approach to reduced order modeling for turbulent flow control using lstm neural networks\.arXiv preprint arXiv:1804\.09269\.Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p2.1)\.
- \[47\]F\. Pichi, B\. Moya, and J\. S\. Hesthaven\(2024\)A graph convolutional autoencoder approach to model order reduction for parametrized pdes\.J\. Comput\. Phys\.501,pp\. 112762\.External Links:ISSN 0021\-9991,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.jcp.2024.112762),[Link](https://www.sciencedirect.com/science/article/pii/S0021999124000111)Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p3.1)\.
- \[48\]C\. Quilodrán\-Casas, R\. Arcucci, L\. Mottet,et al\.\(2021\)Adversarial autoencoders and adversarial lstm for improved forecasts of urban air pollution simulations\.arXiv preprint arXiv:2104\.06297\.
- \[49\]P\. Reiser, J\. E\. Aguilar, A\. Guthke,et al\.\(2025\)Uncertainty quantification and propagation in surrogate\-based bayesian inference\.Stat\. and Comput\.35\(3\)\.External Links:ISSN 1573\-1375,[Link](http://dx.doi.org/10.1007/s11222-025-10597-8),[Document](https://dx.doi.org/10.1007/s11222-025-10597-8)Cited by:[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[50\]A\. Taflove\(1995\)Computational electrodynamics the finite\-difference time\-domain method\.
- \[51\]M\. Titsias and N\. D\. Lawrence\(2010\)Bayesian gaussian process latent variable model\.InProceedings of the Thirteenth International Conference on Artificial Intelligence and Statistics,Y\. W\. Teh and M\. Titterington \(Eds\.\),Proceedings of Machine Learning Research, Vol\.9,Chia Laguna Resort, Sardinia, Italy,pp\. 844–851\.External Links:[Link](https://proceedings.mlr.press/v9/titsias10a.html)Cited by:[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[52\]M\. Tomasetto, A\. Manzoni, and F\. Braghin\(2026\)Real‐time optimal control of high‐dimensional parametrized systems by deep learning‐based reduced order models\.Int\. J\. Numer\. Methods Eng\.127,pp\.\.External Links:[Document](https://dx.doi.org/10.1002/nme.70237)
- \[53\]J\. H\. Tu\(2013\)Dynamic mode decomposition: theory and applications\.Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p2.1)\.
- \[54\]H\. K\. Versteeg and W\. Malalasekera\(2007\)An introduction to computational fluid dynamics: the finite volume method\.Pearson Education\.
- \[55\]K\. Vlachas, T\. Simpson, A\. Garland,et al\.\(2025\)Reduced order modeling conditioned on monitored features for response and error bounds estimation in engineered systems\.Mech\. Syst\. Signal Process\.226,pp\. 112261\.External Links:ISSN 0888\-3270,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.ymssp.2024.112261),[Link](https://www.sciencedirect.com/science/article/pii/S0888327024011609)Cited by:[§1\.1](https://arxiv.org/html/2608.11435#S1.SS1.p3.1)\.
- \[56\]W\. Wang, C\. Gong, X\. Jia,et al\.\(2025\)A kernel ridge regression combining nonlinear roms for accurate flow\-field reconstruction with discontinuities\.Aerosp\. Sci\. Technol\.166,pp\. 110549\.External Links:ISSN 1270\-9638,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.ast.2025.110549),[Link](https://www.sciencedirect.com/science/article/pii/S1270963825006200)Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p2.1)\.
- \[57\]Y\. Xiao, L\. Yang, C\. Shu,et al\.\(2024\)Physics\-informed quantum neural network for solving forward and inverse problems of partial differential equations\.Phys\. Fluids36\(9\)\.Cited by:[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[58\]J\. Xu and K\. Duraisamy\(2020\)Multi\-level convolutional autoencoder networks for parametric prediction of spatio\-temporal dynamics\.Comput\. Methods Appl\. Mech\. Eng\.372,pp\. 113379\.Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p3.1)\.
- \[59\]T\. Yang, J\. Rosca, K\. Narasimhan,et al\.\(2022\)Learning physics constrained dynamics using autoencoders\.InAdvances in Neural Information Processing Systems,S\. Koyejo, S\. Mohamed, A\. Agarwal, D\. Belgrave, K\. Cho, and A\. Oh \(Eds\.\),Vol\.35,pp\. 17157–17172\.External Links:[Link](https://proceedings.neurips.cc/paper_files/paper/2022/file/6d5e035724687454549b97d6c805dc84-Paper-Conference.pdf)Cited by:[§1\.2](https://arxiv.org/html/2608.11435#S1.SS2.p4.1)\.
- \[60\]J\. Zhang, Q\. Zheng, D\. Chen,et al\.\(2020\)Surrogate‐based bayesian inverse modeling of the hydrological system: an adaptive approach considering surrogate approximation error\.Water Resour\. Res\.56\(1\)\.External Links:ISSN 1944\-7973,[Link](http://dx.doi.org/10.1029/2019WR025721),[Document](https://dx.doi.org/10.1029/2019wr025721)Cited by:[§4](https://arxiv.org/html/2608.11435#S4.p3.1)\.
- \[61\]O\. C\. Zienkiewicz, R\. L\. Taylor, J\. Z\. Zhu,et al\.\(1997\)The finite element method\.

## Appendix AGaussian process regression of POD coefficients

Once the POD basis𝐋𝒳,q\\mathbf\{L\}\_\{\\mathcal\{X\},q\}is computed and the training snapshots are projected onto the reduced space, the corresponding POD coefficients are obtained for each training parameter instance\. Collecting these reduced representations over the training set yields the POD coefficient matrix

𝐱^=\[\{kj​\(𝜽\(0\)\)\}j=1q,…,\{kj​\(𝜽\(Ntrain−1\)\)\}j=1q\]∈ℝq×Ntrain,\\hat\{\\mathbf\{x\}\}=\\Big\[\\\{k\_\{j\}\(\\boldsymbol\{\\theta\}^\{\(0\)\}\)\\\}\_\{j=1\}^\{q\},\\;\\ldots,\\;\\\{k\_\{j\}\(\\boldsymbol\{\\theta\}^\{\(N\_\{\\text\{train\}\}\-1\)\}\)\\\}\_\{j=1\}^\{q\}\\Big\]\\in\\mathbb\{R\}^\{q\\times N\_\{\\text\{train\}\}\},which serves as the regression target for the subsequent surrogate modeling\. Following the common assumption that POD coefficients are approximately decorrelated, the regression task is split intoqqindependent scalar problems, one for each mode\.

We adopt a Gaussian Process Regression \(GPR\) model for each coefficientkjk\_\{j\}, with an additive Gaussian noise\[[40](https://arxiv.org/html/2608.11435#bib.bib29)\]:

kj​\(𝜽\)=fj​\(𝜽\)\+εj,fj∼𝒢​𝒫​\(0,rj\),εj∼𝒩⁡\(0,sj2\),k\_\{j\}\(\\boldsymbol\{\\theta\}\)=f\_\{j\}\(\\boldsymbol\{\\theta\}\)\+\\varepsilon\_\{j\},\\quad f\_\{j\}\\sim\\mathcal\{GP\}\\big\(0,\\,r\_\{j\}\\big\),\\quad\\varepsilon\_\{j\}\\sim\\mathcal\{N\}\(0,s\_\{j\}^\{2\}\),\(35\)whererjr\_\{j\}is the covariance function defined onΩθ2\\Omega\_\{\\theta\}^\{2\}andsj2s\_\{j\}^\{2\}is the noise variance\. Given training samples\{𝜽\(i\),kj\(i\)\}i=1Nt​r​a​i​n\\\{\\boldsymbol\{\\theta\}^\{\(i\)\},k\_\{j\}^\{\(i\)\}\\\}\_\{i=1\}^\{N\_\{train\}\}, the posterior distribution of the predicted POD coefficientkj∗​\(𝜽∗\)k\_\{j\}^\{\*\}\(\\boldsymbol\{\\theta\}^\{\*\}\)at a new parameter𝜽∗\\boldsymbol\{\\theta\}^\{\*\}, conditional on the training set\{Θt​r​a​i​n,𝐊jt​r​a​i​n\}=\{\(𝜽\(i\),kj\(i\)\)\}i=1Nt​r​a​i​n\\\{\\Theta^\{train\},\\mathbf\{K\}\_\{j\}^\{train\}\\\}=\\\{\(\\boldsymbol\{\\theta\}^\{\(i\)\},k\_\{j\}^\{\(i\)\}\)\\\}\_\{i=1\}^\{N\_\{train\}\}, is

kj∗​\(𝜽∗\)\|\{Θt​r​a​i​n,𝐊jt​r​a​i​n\}∼𝒩⁡\(μj,σGP2​\(kj∗\)\),k\_\{j\}^\{\*\}\(\\boldsymbol\{\\theta\}^\{\*\}\)\\;\\big\|\\;\\\{\\Theta^\{train\},\\mathbf\{K\}\_\{j\}^\{train\}\\\}\\;\\sim\\;\\mathcal\{N\}\\\!\\big\(\\mu\_\{j\},\\,\\sigma^\{2\}\_\{\\mathrm\{GP\}\}\(k\_\{j\}^\{\*\}\)\\big\),\(36\)with

μj\\displaystyle\\mu\_\{j\}=rj​\(𝜽∗,Θt​r​a​i​n\)​\[rj​\(Θt​r​a​i​n,Θt​r​a​i​n\)\+sj2​𝐈\]−1​𝐊jt​r​a​i​n,\\displaystyle=r\_\{j\}\(\\boldsymbol\{\\theta\}^\{\*\},\\Theta^\{train\}\)\\Big\[\\,r\_\{j\}\(\\Theta^\{train\},\\Theta^\{train\}\)\+s\_\{j\}^\{2\}\\mathbf\{I\}\\,\\Big\]^\{\-1\}\\mathbf\{K\}\_\{j\}^\{train\},\(37a\)σGP2​\(kj∗\)\\displaystyle\\sigma^\{2\}\_\{\\mathrm\{GP\}\}\(k\_\{j\}^\{\*\}\)=rj​\(𝜽∗,𝜽∗\)\+sj2\\displaystyle=r\_\{j\}\(\\boldsymbol\{\\theta\}^\{\*\},\\boldsymbol\{\\theta\}^\{\*\}\)\+s\_\{j\}^\{2\}−rj​\(𝜽∗,Θt​r​a​i​n\)​\[rj​\(Θt​r​a​i​n,Θt​r​a​i​n\)\+sj2​𝐈\]−1​rj​\(Θt​r​a​i​n,𝜽∗\)\.\\displaystyle\\quad\-r\_\{j\}\(\\boldsymbol\{\\theta\}^\{\*\},\\Theta^\{train\}\)\\Big\[\\,r\_\{j\}\(\\Theta^\{train\},\\Theta^\{train\}\)\+s\_\{j\}^\{2\}\\mathbf\{I\}\\,\\Big\]^\{\-1\}r\_\{j\}\(\\Theta^\{train\},\\boldsymbol\{\\theta\}^\{\*\}\)\.\(37b\)
For regression, we trainqqindependent Gaussian processes using a multi\-output wrapper that enables parallel computation\. This implementation directly adopts the code architecture proposed in\[[40](https://arxiv.org/html/2608.11435#bib.bib29)\]\. Each GP employs a radial basis function kernel of the form

rj​\(𝜽,𝜽′\)=ρj​exp⁡\(−12​‖𝜽−𝜽′‖Λj2\),r\_\{j\}\(\\boldsymbol\{\\theta\},\\boldsymbol\{\\theta\}^\{\\prime\}\)=\\rho\_\{j\}\\exp\\\!\\Big\(\-\\tfrac\{1\}\{2\}\\\|\\boldsymbol\{\\theta\}\-\\boldsymbol\{\\theta\}^\{\\prime\}\\\|^\{2\}\_\{\\Lambda\_\{j\}\}\\Big\),whereΛj=diag⁡\(λj,1−2,…,λj,dθ−2\)\\Lambda\_\{j\}=\\mathrm\{diag\}\(\\lambda\_\{j,1\}^\{\-2\},\\ldots,\\lambda\_\{j,d\_\{\\theta\}\}^\{\-2\}\)contains the characteristic length\-scales associated with each parameter dimension, andρj\\rho\_\{j\}denotes the maximum allowable covariance\. For convenience, we adopt the notation‖𝐚‖𝐀2=𝐚⊤​𝐀−1​𝐚\\\|\\mathbf\{a\}\\\|\_\{\\mathbf\{A\}\}^\{2\}=\\mathbf\{a\}^\{\\top\}\\mathbf\{A\}^\{\-1\}\\mathbf\{a\}\.

The GP predictive variance reflects two distinct sources of uncertainty: \(i\) observation noise in the training data, represented by the termsj2​𝐈s\_\{j\}^\{2\}\\mathbf\{I\}, and \(ii\) regression uncertainty, which increases with the distance between new input parameters𝜽∗\\boldsymbol\{\\theta\}^\{\*\}and the training setΘtrain\\Theta^\{\\mathrm\{train\}\}\. The set of hyperparameters for each GP is given by

γj=\(ρj,sj,λj,1,…,λj,dθ\)∈ℝ\+dθ\+2,\\gamma\_\{j\}=\\big\(\\rho\_\{j\},\\,s\_\{j\},\\,\\lambda\_\{j,1\},\\ldots,\\lambda\_\{j,d\_\{\\theta\}\}\\big\)\\in\\mathbb\{R\}\_\{\+\}^\{d\_\{\\theta\}\+2\},and is determined by maximizing the log\-marginal likelihood \(MLL\) with multiple random restarts\. Finally, the posterior predictions of the reduced coordinates

𝐱^∗=\[\{kj∗​\(𝜽∗\(0\)\)\}j=1q,…,\{kj∗​\(𝜽∗\(Ntest−1\)\)\}j=1q\]∈ℝq×Ntest,\\hat\{\\mathbf\{x\}\}^\{\*\}=\\Big\[\\\{k\_\{j\}^\{\*\}\(\\boldsymbol\{\\theta\}^\{\*\(0\)\}\)\\\}\_\{j=1\}^\{q\},\\;\\ldots,\\;\\\{k\_\{j\}^\{\*\}\(\\boldsymbol\{\\theta\}^\{\*\(N\_\{\\text\{test\}\}\-1\)\}\)\\\}\_\{j=1\}^\{q\}\\Big\]\\in\\mathbb\{R\}^\{q\\times N\_\{\\text\{test\}\}\},are mapped back to the full physical space through the POD basis computed from the training set, thus performing the inverse POD reconstruction to predict the velocity fields corresponding to new parameter configurations in the test dataset\. For clarity, the complete offline training and online prediction workflows of POD\-GPR surrogate are summarized in Fig[13](https://arxiv.org/html/2608.11435#A1.F13)\.

Figure 13:
## Appendix BNeural network architectures

Table 7:Neural Network structure of the OACAE encoder/decoder \(input shape: 2×\\times64×\\times64, output shape: 2×\\times64×\\times64\)\.Layer \(type\)Output shapeActivation*Encoder*Input\(2,64,64\)\(2,\\,64,\\,64\)Conv2d\(3×3\)\(3\\times 3\)\(16,64,64\)\(16,\\,64,\\,64\)TanhMaxPooling\(2×2\)\(2\\times 2\)\(16,32,32\)\(16,\\,32,\\,32\)Conv2d\(3×3\)\(3\\times 3\)\(8,32,32\)\(8,\\,32,\\,32\)TanhMaxPooling\(2×2\)\(2\\times 2\)\(8,16,16\)\(8,\\,16,\\,16\)Conv2d\(3×3\)\(3\\times 3\)\(4,16,16\)\(4,\\,16,\\,16\)TanhMaxPooling\(2×2\)\(2\\times 2\)\(4,8,8\)\(4,\\,8,\\,8\)Flatten\(256\)\(256\)Dense\(→256\)\(256\\\!\\to\\\!256\)\(256\)\(256\)TanhDense\(→128\)\(256\\\!\\to\\\!128\)\(128\)\(128\)TanhDense\(→64\)\(128\\\!\\to\\\!64\)\(64\)\(64\)TanhDense\(→32\)\(64\\\!\\to\\\!32\)\(32\)\(32\)TanhLatent \(Dense→3232\\\!\\to\\\!32\)\(32\)\(32\)*Decoder*Dense\(→32\)\(32\\\!\\to\\\!32\)\(32\)\(32\)TanhDense\(→64\)\(32\\\!\\to\\\!64\)\(64\)\(64\)TanhDense\(→128\)\(64\\\!\\to\\\!128\)\(128\)\(128\)TanhDense\(→256\)\(128\\\!\\to\\\!256\)\(256\)\(256\)TanhDense\(→256\)\(256\\\!\\to\\\!256\)\(256\)\(256\)TanhReshape\(4,8,8\)\(4,\\,8,\\,8\)Conv2d\(3×3\)\(3\\times 3\)\(8,8,8\)\(8,\\,8,\\,8\)TanhConvTranspose2d\(4×4\)\(4\\times 4\)\(16,16,16\)\(16,\\,16,\\,16\)TanhConvTranspose2d\(4×4\)\(4\\times 4\)\(32,32,32\)\(32,\\,32,\\,32\)TanhConvTranspose2d\(4×4\)\(4\\times 4\)\(2,64,64\)\(2,\\,64,\\,64\)Table 8:Neural Network structure of the observable branch \(MLPℰa\\mathcal\{E\}\_\{a\}\) used in OACAE training \(input shape: 32, output shape: 6\)\.Layer \(type\)Output shapeActivationInput \(latent\)\(32\)\(32\)Dense\(→32\)\(32\\\!\\to\\\!32\)\(32\)\(32\)TanhDense\(→64\)\(32\\\!\\to\\\!64\)\(64\)\(64\)TanhDense\(→32\)\(64\\\!\\to\\\!32\)\(32\)\(32\)TanhDense\(→6\)\(32\\\!\\to\\\!6\)\(6\)\(6\)Table 9:Neural Network structure of the parameter\-to\-latent regressor \(MLPℰb\\mathcal\{E\}\_\{b\}, input shape: 6, output shape: 32\)\.Layer \(type\)Output shapeActivationInput \(parameters\)\(6\)\(6\)Dense\(→50\)\(6\\\!\\to\\\!50\)\(50\)\(50\)ReLUDense\(→200\)\(50\\\!\\to\\\!200\)\(200\)\(200\)ReLUDense\(→200\)\(200\\\!\\to\\\!200\)\(200\)\(200\)ReLUDense\(→50\)\(200\\\!\\to\\\!50\)\(50\)\(50\)ReLUDense\(→32\)\(50\\\!\\to\\\!32\)\(32\)\(32\)
## Appendix CReconstruction snapshots

![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_truth_u.png)\(a\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_no_error_u.png)\(b\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_truth_v.png)\(c\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_no_error_v.png)\(d\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_pod_u.png)\(e\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_pod_err_u.png)\(f\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_pod_v.png)\(g\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_pod_err_v.png)\(h\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_cae_u.png)\(i\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_cae_err_u.png)\(j\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_cae_v.png)\(k\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_cae_err_v.png)\(l\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_oacae_u.png)\(m\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_oacae_err_u.png)\(n\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_oacae_v.png)\(o\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/dam_reconstruction_oacae_err_v.png)\(p\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_truth_u.png)\(q\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_no_error_u.png)\(r\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_truth_v.png)\(s\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_no_error_v.png)\(t\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_pod_u.png)\(u\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_pod_err_u.png)\(v\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_pod_v.png)\(w\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_pod_err_v.png)\(x\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_cae_u.png)\(y\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_cae_err_u.png)\(z\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_cae_v.png)\(aa\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_cae_err_v.png)\(ab\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_oacae_u.png)\(ac\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_oacae_err_u.png)\(ad\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_oacae_v.png)\(ae\)
![Refer to caption](https://arxiv.org/html/2608.11435v1/cavity_reconstructions_oacae_err_v.png)\(af\)

Figure 14:Dam and Cavity reconstruction
## Appendix DQuantitative analysis of the physics\-aware latent space

The reconstruction and prediction errors reported in Section[3\.4](https://arxiv.org/html/2608.11435#S3.SS4)and Section[3\.5](https://arxiv.org/html/2608.11435#S3.SS5)evaluate the accuracy of the reduced\-order surrogates in the physical space\. However, these metrics do not directly indicate whether the learned latent variables are physically informative\. In this part, we further analyze the latent representations learned by standard CAE and OACAE\.

For each parameter case, the corresponding latent codes form a temporal trajectory in the latent space\. A latent representation that is favorable for inverse modeling should satisfy two desirable properties\. First, latent trajectories associated with different parameter configurations should be distinguishable, so that different physical regimes are not strongly overlapped\. Second, for a fixed parameter configuration, the latent trajectory should evolve smoothly in time, so that the multi\-time\-step surrogate mapping remains consistent for variational assimilation\.

Letzi,t∈ℝdz\_\{i,t\}\\in\\mathbb\{R\}^\{d\}denote the latent code of thett\-th snapshot in theii\-th parameter case, wherei=1,…,Ni=1,\\ldots,N,t=1,…,Tt=1,\\ldots,T, andddis the latent dimension\. For each parameter case, we define the latent centroid and the within\-case variance as

μi=1T​∑t=1Tzi,t,σi2=1T​∑t=1T‖zi,t−μi‖22\.\\mu\_\{i\}=\\frac\{1\}\{T\}\\sum\_\{t=1\}^\{T\}z\_\{i,t\},\\qquad\\sigma\_\{i\}^\{2\}=\\frac\{1\}\{T\}\\sum\_\{t=1\}^\{T\}\\\|z\_\{i,t\}\-\\mu\_\{i\}\\\|\_\{2\}^\{2\}\.The average within\-case variance is then given by

σwithin2=1N​∑i=1Nσi2\.\\sigma\_\{\\mathrm\{within\}\}^\{2\}=\\frac\{1\}\{N\}\\sum\_\{i=1\}^\{N\}\\sigma\_\{i\}^\{2\}\.This quantity measures the average dispersion of each temporal latent trajectory around its case centroid\. To quantify the separation between different parameter cases, we compute the pairwise distance between case centroids,

di​j=‖μi−μj‖2,i≠j,d\_\{ij\}=\\\|\\mu\_\{i\}\-\\mu\_\{j\}\\\|\_\{2\},\\qquad i\\neq j,and define the average between\-case distance as

dbetween=2N⁡\(N−1\)​∑1≤i<j≤Ndi​j\.d\_\{\\mathrm\{between\}\}=\\frac\{2\}\{N\(N\-1\)\}\\sum\_\{1\\leq i<j\\leq N\}d\_\{ij\}\.Since the absolute scale of the latent space may differ between CAE and OACAE, we use the normalized case\-separability ratio

Rsep=dbetweenσwithin2\.R\_\{\\mathrm\{sep\}\}=\\frac\{d\_\{\\mathrm\{between\}\}\}\{\\sqrt\{\\sigma\_\{\\mathrm\{within\}\}^\{2\}\}\}\.A larger value ofRsepR\_\{\\mathrm\{sep\}\}indicates that different parameter cases are more clearly separated relative to the temporal dispersion within each case\.

In addition to case\-level separability, we quantify the temporal regularity of the latent trajectories\. Because the latent coordinates may have different scales, each latent dimension is first standardized over the whole dataset\. Denoting the standardized latent code byz~i,t\\tilde\{z\}\_\{i,t\}, we define the first\-order temporal smoothness indicator

Si\(1\)=1T−1​∑t=1T−1‖z~i,t\+1−z~i,t‖2,S\_\{i\}^\{\(1\)\}=\\frac\{1\}\{T\-1\}\\sum\_\{t=1\}^\{T\-1\}\\\|\\tilde\{z\}\_\{i,t\+1\}\-\\tilde\{z\}\_\{i,t\}\\\|\_\{2\},and the second\-order temporal smoothness indicator

Si\(2\)=1T−2​∑t=2T−1‖z~i,t\+1−2​z~i,t\+z~i,t−1‖2\.S\_\{i\}^\{\(2\)\}=\\frac\{1\}\{T\-2\}\\sum\_\{t=2\}^\{T\-1\}\\\|\\tilde\{z\}\_\{i,t\+1\}\-2\\tilde\{z\}\_\{i,t\}\+\\tilde\{z\}\_\{i,t\-1\}\\\|\_\{2\}\.The global smoothness indicators are obtained by averaging over all parameter cases:

S\(1\)=1N​∑i=1NSi\(1\),S\(2\)=1N​∑i=1NSi\(2\)\.S^\{\(1\)\}=\\frac\{1\}\{N\}\\sum\_\{i=1\}^\{N\}S\_\{i\}^\{\(1\)\},\\qquad S^\{\(2\)\}=\\frac\{1\}\{N\}\\sum\_\{i=1\}^\{N\}S\_\{i\}^\{\(2\)\}\.Smaller values ofS\(1\)S^\{\(1\)\}andS\(2\)S^\{\(2\)\}indicate smoother temporal evolution in the latent space\.

Table[10](https://arxiv.org/html/2608.11435#A4.T10)reports the latent\-space metrics on the training set\. Compared with CAE, OACAE yields a smaller within\-case variance and a substantially larger normalized separability ratio\. In particular,RsepR\_\{\\mathrm\{sep\}\}increases from1\.5016531\.501653for CAE to3\.4743833\.474383for OACAE\. The first\- and second\-order temporal smoothness indicators are also reduced, indicating that the OACAE latent trajectories are more regular in time\.

Table 10:Comparison of latent representations between CAE and OACAE on the training dataset\.ModelS\(1\)↓S^\{\(1\)\}\\downarrowS\(2\)↓S^\{\(2\)\}\\downarrowσwithin2\\sigma\_\{\\mathrm\{within\}\}^\{2\}dbetweend\_\{\\mathrm\{between\}\}Rsep↑R\_\{\\mathrm\{sep\}\}\\uparrowCAE0\.322924±0\.0737120\.322924\\pm 0\.0737120\.199233±0\.0663400\.199233\\pm 0\.0663406\.6485073\.8719671\.501653OACAE0\.145713±0\.0333350\.145713\\pm 0\.0333350\.104804±0\.0517890\.104804\\pm 0\.0517890\.7030902\.9132873\.474383Figure[15](https://arxiv.org/html/2608.11435#A4.F15)provides a complementary visualization of the latent manifolds using t\-SNE projections\. The CAE latent trajectories exhibit stronger overlap and less organized case\-level structure, whereas the OACAE latent trajectories are more structured according to parameter\-induced dynamical regimes\. These visual and quantitative results support the interpretation that observable supervision improves the parameter consistency of the learned latent representation\.

\(a\)CAE![Refer to caption](https://arxiv.org/html/2608.11435v1/train_oacae_tsne_3d.png)\(b\)OACAE

Figure 15:Visualization of latent manifolds of dam flow evolutions for different boundary conditions \(inlet velocity\)\.
## Appendix ESensitivity analyses

### E\.1Sensitivity to training loss weights

We first examine the sensitivity of the proposed AE\-MLP variational DA framework to the training loss weightsα\\alphaandβ\\beta\. We fixβ=1\\beta=1and vary

α∈\{0,0\.01,0\.05,0\.1,0\.2,0\.5\},\\alpha\\in\\\{0,0\.01,0\.05,0\.1,0\.2,0\.5\\\},and fixα=0\.05\\alpha=0\.05and vary

β∈\{0\.5,1,2\}\.\\beta\\in\\\{0\.5,1,2\\\}\.For each setting, we evaluate the autoencoder reconstruction error, the AE\-MLP prediction error, and the full\-observation 3D\-Var inverse\-calibration error\.

Figure 16:Sensitivity analysis ofα\\alphaandβ\\betafor forward modeling\.Figure 17:Sensitivity analysis ofα\\alphaandβ\\betafor inverse modeling\.
### E\.2Sensitivity to latent dimension

We further study the sensitivity of the proposed framework to the latent\-space dimension\. Both CAE\-MLP and OACAE\-MLP are evaluated using latent dimensions

q∈\{2,6,13,32,64\}\.q\\in\\\{2,6,13,32,64\\\}\.For all tested latent dimensions, the weighting parameters are fixed at values ofα=0\.05\\alpha=0\.05andβ=1\\beta=1\. The comparison includes the autoencoder reconstruction error, the AE\-MLP prediction error, and the full\-observation 3D\-Var inverse\-calibration error\.

Figure 18:Sensitivity analysis of AE latent dimension\.As shown in Figure[18](https://arxiv.org/html/2608.11435#A5.F18), very small latent dimensions, such asq=2q=2andq=6q=6, lead to larger errors because the latent space does not have sufficient capacity to represent the dominant flow structures and the parameter\-relevant information\. Increasing the latent dimension improves both reconstruction and prediction performance, but the improvement saturates aroundq=32q=32\. For the inverse 3D\-Var calibration task, all tested latent dimensions substantially reduce the parameter error compared with the background estimate\. The OACAE\-MLP framework shows a clear improvement fromq=2q=2toq=13q=13, and then remains stable forq=32q=32andq=64q=64\. As a practical heuristic, the POD energy spectrum can be used as a conservative reference for selecting a preliminary latent dimension, since POD provides the optimal linear low\-rank approximation of the snapshot data in the mean\-square sense\. In addition, if the solution manifold is mainly parameterized bypeffp\_\{\\mathrm\{eff\}\}effective variables, a simple manifold\-based heuristic suggests testing latent dimensions aroundpeffp\_\{\\mathrm\{eff\}\}and2​peff\+12p\_\{\\mathrm\{eff\}\}\+1\[[22](https://arxiv.org/html/2608.11435#bib.bib58)\]\. In the present datasets, the effective number of generating variables is approximatelypeff=6p\_\{\\mathrm\{eff\}\}=6, which motivates the tested dimensionsq=6q=6andq=13q=13, together with larger valuesq=32q=32andq=64q=64\. This argument is used only as a heuristic reference rather than as a strict optimality criterion\. The final choice is based on the empirical sensitivity results for reconstruction, prediction, and inverse calibration\.

### E\.3Sensitivity to variational optimization learning rate

We also investigate the sensitivity of the 3D\-Var optimization to the learning rate used in the gradient\-based minimization\. We performed a learning\-rate sweep over

η∈\{0\.009,0\.03,0\.05,0\.07,0\.09,0\.13\},\\eta\\in\\\{0\.009,\\,0\.03,\\,0\.05,\\,0\.07,\\,0\.09,\\,0\.13\\\},for both CAE\-MLP and OACAE\-MLP under the same assimilation setting\. The corresponding objective histories are presented in Figures[19](https://arxiv.org/html/2608.11435#A5.F19)\(a\) and[19](https://arxiv.org/html/2608.11435#A5.F19)\(b\)\. For both models, all tested learning rates eventually approach similar objective plateaus for the representative observation, indicating that the converged objective is relatively insensitive to the learning rate within the tested range\. However, the convergence speed and transient behavior depend on the optimizer setting\. The smallest learning rate gives smooth but slow convergence, whereas larger learning rates accelerate the initial decrease but may introduce stronger transient oscillations or occasional overshooting\. Figures[19](https://arxiv.org/html/2608.11435#A5.F19)\(c\) and[19](https://arxiv.org/html/2608.11435#A5.F19)\(d\) further examine convergence at the selected learning rateη=0\.05\\eta=0\.05for observations taken at different time instances\. Although the initial objective values and final plateaus vary with the observation, the OACAE\-MLP trajectories are more tightly clustered, while the CAE\-MLP trajectories show a wider spread in final objective values\. This suggests more uniform convergence behavior for OACAE\-MLP across different observations\.

The statistical calibration results are summarized in Figure[20](https://arxiv.org/html/2608.11435#A5.F20)\. The mean final objective remains nearly constant for0\.03≤ηopt≤0\.090\.03\\leq\\eta\_\{\\mathrm\{opt\}\}\\leq 0\.09\. Although CAE\-MLP can yield a lower mean final objective value, this should not be interpreted as better calibration, because CAE\-MLP and OACAE\-MLP define different nonlinear mappings and therefore different optimization landscapes\. The parameter\-recovery results show that OACAE\-MLP achieves lower mean relative errors for the calibrated parameters over the tested learning\-rate range\. It also exhibits smaller standard deviations for most intermediate learning rates\. Overall, the interval

0\.03≤ηopt≤0\.090\.03\\leq\\eta\_\{\\mathrm\{opt\}\}\\leq 0\.09provides the best compromise between convergence speed, parameter accuracy, and robustness\.

\(a\)CAE 3dvar history fixing observation varying learning rate\(b\)OACAE 3dvar history fixing observation varying learning rate\(c\)CAE 3dvar history fixing learning rate varying observations\(d\)OACAE 3dvar history fixing learning rate varying observations
Figure 19:Gradient\-based optimization objective histories\.\(a\)Mean final objective values varying learning rate\(b\)Parameter calibration errors varying learning rate
Figure 20:Sensitivity of the 3D\-Var optimization results to the learning rate\.
### E\.4Sensitivity to background and observation covariance scaling

The variational data\-assimilation objective depends on the background covariance matrix𝐁\\mathbf\{B\}and the observation covariance matrix𝐑\\mathbf\{R\}\. These matrices control the relative weights of the background term and the observation term\. To assess the sensitivity of the calibration results to this balance, we perform additional experiments by scaling𝐁\\mathbf\{B\}and𝐑\\mathbf\{R\}\. Let𝐁0\\mathbf\{B\}\_\{0\}and𝐑0\\mathbf\{R\}\_\{0\}denote the reference covariance matrices used in the main experiments\. We consider scaled covariance matrices of the form

𝐁=λ𝐁​𝐁0,𝐑=λ𝐑​𝐑0,\\mathbf\{B\}=\\lambda\_\{\\mathbf\{B\}\}\\mathbf\{B\}\_\{0\},\\qquad\\mathbf\{R\}=\\lambda\_\{\\mathbf\{R\}\}\\mathbf\{R\}\_\{0\},whereλ𝐁\\lambda\_\{\\mathbf\{B\}\}andλ𝐑\\lambda\_\{\\mathbf\{R\}\}were varied over the range

λ𝐁,λ𝐑∈\{10−4,10−3,10−2,10−1,1,10,102,103,104\}\.\\lambda\_\{\\mathbf\{B\}\},\\lambda\_\{\\mathbf\{R\}\}\\in\\left\\\{10^\{\-4\},10^\{\-3\},10^\{\-2\},10^\{\-1\},1,10,10^\{2\},10^\{3\},10^\{4\}\\right\\\}\.
Figure[21](https://arxiv.org/html/2608.11435#A5.F21)reports the calibration performance under different covariance\-scaling settings\. When𝐁\\mathbf\{B\}is scaled to be very small, the background term becomes overly dominant, forcing the analysis to remain too close to the background state and leading to larger parameter errors \(Figure[21](https://arxiv.org/html/2608.11435#A5.F21)left\)\. Conversely, when𝐑\\mathbf\{R\}is scaled to be very large, the observation term is strongly downweighted, weakening the observational correction and also increasing the parameter error\(Figure[21](https://arxiv.org/html/2608.11435#A5.F21)right\)\. These results indicate that the proposed framework is stable with respect to reasonable choices of B and R and the conclusions of the main experiments are not tied to a single covariance choice\.

Figure 21:Sensitivity of the assimilation results to the scaling of the background covariance matrix𝐁\\mathbf\{B\}and the observation covariance matrix𝐑\\mathbf\{R\}\. The left panel varies𝐁=λ𝐁​𝐁0\\mathbf\{B\}=\\lambda\_\{\\mathbf\{B\}\}\\mathbf\{B\}\_\{0\}while fixing𝐑=𝐑0\\mathbf\{R\}=\\mathbf\{R\}\_\{0\}, and the right panel varies𝐑=λ𝐑​𝐑0\\mathbf\{R\}=\\lambda\_\{\\mathbf\{R\}\}\\mathbf\{R\}\_\{0\}while fixing𝐁=𝐁0\\mathbf\{B\}=\\mathbf\{B\}\_\{0\}\.

## Appendix FComputational efficiency

Finally, we compare the online computational efficiency of the proposed differentiable variational assimilation framework with an ensemble\-based POD\-GPR assimilation baseline\. The purpose of this experiment is to evaluate the cost of solving the inverse problem after the offline surrogate training stage has been completed\. Therefore, the reported wall\-clock time corresponds to the online calibration stage only\.

Figure 22:Online runtime comparison between the EnKF\-POD\-GPR baseline and the differentiable 4D\-Var frameworks using CAE\-MLP and OACAE\-MLP surrogates\. The dashed EnKF curve denotes the estimated runtime for the 100\-snapshot case, and the 4D\-Var results are obtained with learning rate0\.0050\.005\.The POD\-GPR ensemble\-based strategy requires repeated surrogate evaluations for all ensemble members and for each assimilated time snapshot\. In contrast, the AE\-MLP variational framework solves the inverse problem by directly optimizing the physical control parameters through automatic differentiation and the optimization can be accelerated on a GPU\. Figure[22](https://arxiv.org/html/2608.11435#A6.F22)compares the wall\-clock time for different numbers of assimilated time snapshots\. The results show that the proposed AE\-MLP variational framework is substantially more efficient than the ensemble\-based POD\-GPR baseline\. This advantage becomes more pronounced as the number of assimilated snapshots increases\.

\*

Similar Articles

Physics-conforming Latent Twins

arXiv cs.LG

Physics-conforming Latent Twins is a framework for learning latent surrogate solution operators that enforce physical principles such as conservation laws and dissipative inequalities by design, using a constraint-transfer approach and structure-preserving latent dynamics.