Stabilized neural Hamilton--Jacobi--Bellman solvers: Error analysis and applications in model-based reinforcement learning
About
Physics-informed neural solvers offer a promising route to model-based reinforcement learning in continuous time, where optimal feedback synthesis is governed by Hamilton--Jacobi--Bellman (HJB) equations. Practical implementations often occupy a regime that is neither a classical grid method nor a continuous-PDE PINN: the value function is represented by a neural network, finite-difference HJB policy-evaluation operators are evaluated by network queries at shifted points, and residuals are minimized by random continuous collocation. This regime preserves the stabilized finite-difference policy-evaluation structure while avoiding grid-based value unknowns. We develop an error theory for this hybrid regime. Interpreting finite differences as shift operators acting on neural networks, we prove a population $L^2$ stability estimate for one policy-evaluation step with learned dynamics. The bound separates residual error, initial and exterior-collar mismatch, policy mismatch, and model-identification error, with an explicit gradient amplification factor for learned dynamics, while the underlying linear evaluation stability remains free of hidden inverse-viscosity blow-up. We further give a finite-sample collocation certificate and a conditional multi-step propagation result through greedy policy improvement. Experiments on compact-control LQR upto 64 dimensions, Allen--Cahn control, pendulum, Hopper, and 3D quadrotor benchmarks compare against representative model-based and model-free RL baselines, demonstrating the predicted residual, policy-mismatch, and learned-model error trends.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Optimal Control | Quadrotor 3D | Final Cost30.22 | 7 | |
| Nonlinear control | Duffing Oscillator | Evaluation Cost1.92 | 5 | |
| Nonlinear control | Spacecraft | Evaluation Cost3.645 | 5 | |
| Optimal Control | LQR 4D | Evaluation Cost1.056 | 5 | |
| Optimal Control | LQR 8D | Evaluation Cost3.569 | 5 | |
| Optimal Control | LQR 16D | Evaluation Cost9.895 | 5 | |
| Optimal Control | LQR 32D | Evaluation Cost17.72 | 5 | |
| Optimal Control | LQR 64D | Evaluation Cost34.073 | 5 | |
| PDE control | Allen-Cahn 10D | Evaluation Cost66.29 | 5 | |
| PDE control | Allen-Cahn 20D | Evaluation Time119.8 | 5 |