Search papers, labs, and topics across Lattice.
This paper introduces a physics-informed neural operator training framework that decomposes PDEs into individual physical operators using operator splitting methods. The framework trains separate neural operators for non-linear operators and approximates linear operators with fixed finite-difference convolutions, forming a modular mixture-of-experts architecture. By formulating the problem as a neural ODE, the approach achieves better convergence and generalization to unseen physics, demonstrated on incompressible and compressible Navier-Stokes equations, while also enabling continuous-in-time predictions and temporal extrapolation.
Physics-informed neural operators can now generalize to unseen physical regimes and extrapolate in time by explicitly encoding the underlying operator structure and decomposing PDEs into interpretable components.
Neural operators have emerged as promising surrogate models for solving partial differential equations (PDEs), but struggle to generalise beyond training distributions and are often constrained to a fixed temporal discretisation. This work introduces a physics-informed training framework that addresses these limitations by decomposing PDEs using operator splitting methods, training separate neural operators to learn individual non-linear physical operators while approximating linear operators with fixed finite-difference convolutions. This modular mixture-of-experts architecture enables generalisation to novel physical regimes by explicitly encoding the underlying operator structure. We formulate the modelling task as a neural ordinary differential equation (ODE) where these learned operators constitute the right-hand side, enabling continuous-in-time predictions through standard ODE solvers and implicitly enforcing PDE constraints. Demonstrated on incompressible and compressible Navier-Stokes equations, our approach achieves better convergence and superior performance when generalising to unseen physics. The method remains parameter-efficient, enabling temporal extrapolation beyond training horizons, and provides interpretable components whose behaviour can be verified against known physics.
DeepMind