Finite-Horizon Hamilton--Jacobi--Bellman Equations with State-Dependent Diffusion in Spectral Barron Spaces

arXiv:2609.00529 2026 Training 1 ideas extracted · analyzed Sep 2, 2026

What the math gives to ML

The transferable asset is the paper's parametrix treatment of state-dependent diffusion: instead of approximating a variable-coefficient parabolic operator directly, freeze the diffusion locally, apply an exactly computable Gaussian propagator, and represent the coefficient mismatch as a correction. This suggests a preconditioned neural PDE solver in which the network learns only the residual left after analytic diffusion smoothing, rather than the full HJB solution. The same construction can be implemented as a short-horizon Picard or policy iteration with frozen Gaussian convolutions and automatic-differentiation correction terms. The approach is most promising for high-dimensional stochastic-control surrogates, where grid methods fail and ordinary PINN residual optimization is poorly conditioned.

Ideas from this paper

Mechanism failed 2026

Frozen-Diffusion Parametrix Preconditioner

Replace the raw HJB residual loss of a neural PDE solver with a parametrix-preconditioned fixed-point target. At each local space-time patch, analytically propagate terminal values and source terms through a Gaussian kernel whose covariance uses a frozen diffusion matrix, while asking the network to learn only the variable-coefficient correction. This should reduce the burden on the network to represent stiff high-frequency diffusion dynamics and improve short-horizon convergence.

Useful7/10
Difficulty6/10
Novelty7/10
Paper: Finite-Horizon Hamilton--Jacobi--Bellman Equations with State-Dependent Diffusion in Spectral Barron Spaces arXiv:2609.00529