Hybrid Iterative Neural Low-Regularity Integrator for Nonlinear Dispersive Equations
About
We propose HIN-LRI, a hybrid framework that augments a classical numerical solver with a neural operator trained to correct the solver's structured truncation error. A base low-regularity integrator provides a consistent first-order approximation to nonlinear dispersive PDEs, while a lightweight neural network, operating on a low-dimensional latent manifold, learns the residual defect that analytical methods cannot close. An explicit time-step scaling on the neural correction ensures that its Lipschitz contribution remains $\mathcal{O}(\tau)$, yielding a Gronwall stability factor bounded uniformly in the step size and independent of the spatial resolution. The network is trained end-to-end through a solver-in-the-loop objective that unrolls the full iteration and penalises trajectory error in a Bourgain-type norm, aligning learning with multi-step solver dynamics rather than isolated one-step targets. Under stated assumptions, the global error satisfies $C(\varepsilon_{net}+\delta)\,\tau^\gamma\ln(1/\tau)$, where $\varepsilon_{net}$ measures the network approximation quality and $\delta$ the training shortfall. Experiments on three dispersive benchmarks with rough data show that HIN-LRI improves accuracy over analytical integrators, splitting methods, and neural PDE surrogates, with stable spatial refinement, effective out-of-distribution transfer, and modest online overhead.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Solving Quadratic Nonlinear Schrödinger Equation | Quadratic Nonlinear Schrödinger Equation (T=1.0, N=1024, γ=0.5) | L2 Error7.82e-6 | 21 | |
| Numerical Integration | Cubic NLS T=1.0, N=1024, γ=0.5 | L2 Error6.52e-6 | 16 | |
| Numerical Integration | KdV equation T=1.0, N=1024, γ=0.5 | L2 Error (τ=2^-4)0.0075 | 4 | |
| Out-of-Distribution Numerical PDE Solving | Cubic Nonlinear Schrödinger Riemann step function OOD (test) | L2 Error5.84e-4 | 3 | |
| Out-of-Distribution Numerical PDE Solving | Cubic Nonlinear Schrödinger Dirac delta pulse OOD (test) | L2 Error8.22e-4 | 3 | |
| Relative Hamiltonian energy drift estimation | Rough initial data (N=1024, τ=2⁻⁸) (long-time evolution) | Relative Hamiltonian Energy Drift (T=10)1.15e-13 | 3 | |
| Out-of-Distribution Numerical PDE Solving | Cubic Nonlinear Schrödinger Variable coeff. c(x) OOD (test) | L2 Error0.0016 | 2 |