Search papers, labs, and topics across Lattice.
This paper introduces Variational Grey-Box Dynamics Matching (VGB-DM), a novel method that integrates incomplete physics-based models (ODEs/PDEs) into generative flow matching frameworks to learn dynamics from observational trajectories. VGB-DM models a structured variational distribution with two latent encodings: one for missing stochasticity and multi-modal velocity, and another to encode physics parameters with a physics-informed prior. Experiments on ODE/PDE problems demonstrate that VGB-DM achieves performance comparable to or better than fully data-driven and previous grey-box methods, while retaining the interpretability of the underlying physics model.
Unlock the power of interpretable physics-based models in generative AI, even with incomplete physics, using a novel grey-box method that rivals fully data-driven approaches.
Deep generative models such as flow matching and diffusion models have shown great potential in learning complex distributions and dynamical systems, but often act as black-boxes, neglecting underlying physics. In contrast, physics-based simulation models described by ODEs/PDEs remain interpretable, but may have missing or unknown terms, unable to fully describe real-world observations. We bridge this gap with a novel grey-box method that integrates incomplete physics models directly into generative models. Our approach learns dynamics from observational trajectories alone, without ground-truth physics parameters, in a simulation-free manner that avoids scalability and stability issues of Neural ODEs. The core of our method lies in modelling a structured variational distribution within the flow matching framework, by using two latent encodings: one to model the missing stochasticity and multi-modal velocity, and a second to encode physics parameters as a latent variable with a physics-informed prior. Furthermore, we present an adaptation of the framework to handle second-order dynamics. Our experiments on representative ODE/PDE problems show that our method performs on par with or superior to fully data-driven approaches and previous grey-box baselines, while preserving the interpretability of the physics model. Our code is available at https://github.com/DMML-Geneva/VGB-DM.
