ArXiv

Error-Conditioned Neural Solvers

Authors
Haina Jiang, Liam Wang, Peng-Chen Chen...
Categories
cs.LG, cs.AI, cs.CV, math.NA
arXiv
https://arxiv.org/abs/2606.27354v1
PDF
https://arxiv.org/pdf/2606.27354v1

Brief

Error-Conditioned Neural Solvers (ENS) address limitations of neural surrogates and hybrid residual-minimization methods by passing the PDE residual field as an explicit input at each iteration so the network learns a correction policy instead of numerically minimizing the residual. The authors prove residual minimization can be a poor proxy in ill-conditioned systems and empirically show ENS outperforms hybrids across four PDE families—up to 10× on Kolmogorov flow—while reducing compute and improving robustness to distribution shift.

Why it matters

Error-Conditioned Neural Solvers (ENS) (Jiang et al., arXiv 2026-06-25) feed the PDE residual field into the network at every iteration so the model learns an update policy to correct its own errors, avoiding expensive classical hybrid optimizers (gradient descent / Gauss–Newton) and their instability.

Key details

  • Across four PDE families, ENS attains the highest prediction accuracy in the majority of settings, with improvements up to 10× on turbulent Kolmogorov flow, and it generalizes under distribution shift (zero-shot parameter changes and cross-equation transfer), especially in ill-conditioned regimes where residual minimization is unreliable.
Source evidence

Abstract

Neural surrogate models offer fast approximate mappings from PDE parameters to solutions, but they typically treat solving as a purely statistical task: once trained, they struggle to correct their own constraint violations and extrapolate beyond the training distribution. Recent hybrid methods promote physical correctness by targeting the PDE residual via gradient descent or Gauss--Newton steps, but inherit the compute cost and instability of the underlying classical optimizers. We show, theoretically and empirically, that numerically minimizing the PDE residual can be an unreliable proxy for reconstruction accuracy in ill-conditioned systems, explaining why these methods often do not make accurate predictions despite achieving low residuals. We propose error-conditioned Neural Solvers (ENS), built on a different principle: rather than an optimization target, the PDE residual field is passed as a direct input to the network at each iteration, enabling it to read the spatial structure of its own errors and learn an update policy to iteratively correct its predictions. Across four PDE families, ENS attains the highest prediction accuracy in the large majority of settings, with gains reaching $10\times$ on turbulent Kolmogorov flow, while avoiding the expensive compute cost of hybrid methods. ENS's learned correction policy generalizes under distribution shift, including zero-shot parameter changes and cross-equation transfer, where its relative advantage is largest in the ill-conditioned regimes where residual minimization is least reliable. Project website: https://neuralsolver.github.io/.