NOWS: Neural Operator Warm Starts for Accelerating Iterative Solvers
About
Partial differential equations (PDEs) underpin quantitative descriptions across the physical sciences and engineering, yet high-fidelity simulation remains a major computational bottleneck for many-query, real-time, and design tasks. Data-driven surrogates can be strikingly fast but are often unreliable when applied outside their training distribution. Here we introduce Neural Operator Warm Starts (NOWS), a hybrid strategy that harnesses learned solution operators to accelerate classical iterative solvers by producing high-quality initial guesses for Krylov methods such as conjugate gradient and GMRES. NOWS leaves existing discretizations and solver infrastructures intact, integrating seamlessly with finite-difference, finite-element, isogeometric analysis, finite volume method, etc. Across our benchmarks, the learned initialization consistently reduces iteration counts and end-to-end runtime, resulting in a reduction of the computational time of up to 90 %, while preserving the stability and convergence guarantees of the underlying numerical algorithms. By combining the rapid inference of neural operators with the rigor of traditional solvers, NOWS provides a practical and trustworthy approach to accelerate high-fidelity PDE simulations.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| PDE solving | Poisson | Time (s)2.1 | 55 | |
| Solving Partial Differential Equations (PDEs) | Darcy Flow | CG Time (s)13.8 | 6 | |
| Solving PDE | Smoke Plume | Total Iterations1.05e+4 | 4 | |
| Solving Partial Differential Equations (PDEs) | Smoke Plume | CG Time (s)97.37 | 4 | |
| Solving Partial Differential Equations (PDEs) | Plate & Voids | CG Time (s)2.04e+4 | 2 | |
| Solving Partial Differential Equations (PDEs) | Burgers | CG Time (s)236.4 | 2 |