Structure-preserving fidelity diagnostics for partial differential equations
Organizations: MRC Biostatistics Unit, University of Cambridge, Cambridge, United Kingdom · King’s College, University of Cambridge, Cambridge, United Kingdom
Abstract
Machine learning surrogates of partial differential equations (PDEs) are commonly trained on data generated by numerical solvers. Therefore, the training data obey not the PDE but the modified equation of the numerical scheme, and the test loss, computed on data from the same solver, cannot distinguish a model that learned the physics from one that learned the discretization error. Here, we show that the difference is nevertheless visible in the trained model. We propose a suite of structure-preserving fidelity diagnostics, simple scores that measure whether a learned model respects one known property of the physics, such as a conserved quantity, a symmetry or a Hamiltonian structure, without access to the training data or the governing equation. Over five systems, the diagnostics track the error with respect to the true dynamics with a rank correlation of up to , while the test loss is uncorrelated or anti-correlated with it. For the Burgers, Korteweg-de Vries and Navier-Stokes equations, the diagnostics separate numerical schemes that the test loss ranks as equivalent, including schemes of the same order. order.
Figures & tables
| diagnostic | tests whether the model | requires | rollout | detects, spread across learned models |
|---|---|---|---|---|
| Conservation | ||||
| changes a quantity at the rate of the physics | and its rate | no | artificial viscosity and dispersion, Burgers ; non-conservative forms, KdV ; energy–enstrophy pair, Navier–Stokes ; Euler schemes, oscillators | |
| keeps along its own rollout | yes | non-symplectic schemes, oscillators | ||
| Symmetry | ||||
| commutes with a symmetry | the symmetry | no | dispersive error, Burgers (blind to viscosity); time reversal, oscillators | |
| Structure | ||||
| scheme | of the learned model | ||||
|---|---|---|---|---|---|
| spectral | |||||
| skew, 4th | |||||
| skew, 2nd | |||||
| flux, 2nd | |||||
| advective, 2nd |
PDE Surrogate Modeling
PDE surrogate modeling builds fast approximations of partial differential equation solutions for simulation, prediction, and design.
See all 331 papers