Search papers, labs, and topics across Lattice.
This paper introduces a hierarchical Bayesian framework for jointly estimating model parameters and learning unknown dynamics from data across multiple related physical systems. The approach uses a neural network closure model within the ODE/PDE formulation and employs the ensemble Metropolis-Adjusted Langevin Algorithm (MALA) for stable sampling. To address computational costs, a bilevel optimization strategy is used to simultaneously train a surrogate forward model, comparing Fourier Neural Operators (FNO) and parametric Physics-Informed Neural Networks (PINNs) as surrogate architectures.
By jointly inferring model parameters and learning unknown dynamics across related systems, this work offers a more complete picture of complex physical systems where detailed properties and complete laws are unknown.
Inverse problems are the task of calibrating models to match data. They play a pivotal role in diverse engineering applications by allowing practitioners to align models with reality. In many applications, engineers and scientists do not have a complete picture of i) the detailed properties of a system (such as material properties, geometry, initial conditions, etc.); ii) the complete laws describing all dynamics at play (such as friction laws, complicated damping phenomena, and general nonlinear interactions). In this paper, we develop a principled methodology for leveraging data from collections of distinct yet related physical systems to jointly estimate the individual model parameters of each system, and learn the shared unknown dynamics in the form of an ML-based closure model. To robustly infer the unknown parameters for each system, we employ a hierarchical Bayesian framework, which allows for the joint inference of multiple systems and their population-level statistics. To learn the closures, we use a maximum marginal likelihood estimate of a neural network embeded within the ODE/PDE formulation of the problem. To realize this framework we utilize the ensemble Metropolis-Adjusted Langevin Algorithm (MALA) for stable and efficient sampling. To mitigate the computational bottleneck of repetitive forward evaluations in solving inverse problems, we introduce a bilevel optimization strategy to simultaneously train a surrogate forward model alongside the inference. Within this framework, we evaluate and compare distinct surrogate architectures, specifically Fourier Neural Operators (FNO) and parametric Physics-Informed Neural Network (PINNs).