INC: An Indirect Neural Corrector for Auto-Regressive Hybrid PDE Solvers

TL;DR

The paper introduces INC, an indirect neural correction method for hybrid PDE solvers that reduces long-term errors and stabilizes simulations, enabling faster and more reliable scientific computations.

Contribution

INC integrates learned corrections into PDE governing equations, significantly reducing error amplification and improving long-term stability without architectural constraints.

Findings

Up to 158.7% improvement in long-term trajectory accuracy.

Stabilizes blowups under aggressive coarsening.

Yields orders of magnitude speed-ups in 3D turbulence simulations.

Abstract

When simulating partial differential equations, hybrid solvers combine coarse numerical solvers with learned correctors. They promise accelerated simulations while adhering to physical constraints. However, as shown in our theoretical framework, directly applying learned corrections to solver outputs leads to significant autoregressive errors, which originate from amplified perturbations that accumulate during long-term rollouts, especially in chaotic regimes. To overcome this, we propose the Indirect Neural Corrector ($\mathrm{INC}$), which integrates learned corrections into the governing equations rather than applying direct state updates. Our key insight is that $\mathrm{INC}$ reduces the error amplification on the order of $\Delta t^{-1} + L$, where $\Delta t$ is the timestep and $L$ the Lipschitz constant. At the same time, our framework poses no architectural requirements and integrates seamlessly with arbitrary neural networks and solvers. We test $\mathrm{INC}$ in extensive benchmarks, covering numerous differentiable solvers, neural backbones, and test cases ranging from a 1D chaotic system to 3D turbulence. $\mathrm{INC}$ improves the long-term trajectory performance ($R^2$) by up to 158.7%, stabilizes blowups under aggressive coarsening, and for complex 3D turbulence cases yields speed-ups of several orders of magnitude. $\mathrm{INC}$ thus enables stable, efficient PDE emulation with formal error reduction, paving the way for faster scientific and engineering simulations with reliable physics guarantees. Our source code is available at https://github.com/tum-pbs/INC