The Hamilton-Jacobi Theory of Deep Learning
Not on the map. Why?
Abstract (arXiv)
In this paper, training a neural network is identified, exactly, as a search through Hamilton--Jacobi initial-value problems: each gradient step selects the initial data of a viscous Hamilton--Jacobi equation whose Hopf--Cole propagator best fits the observations; at inference, the input is the spatial point at which that solution is evaluated and the initial condition is already encoded in the weights. The correspondence is exact for log-sum-exp layers, with ReLU, sigmoid, SiLU, and GELU each an exact limit, gradient, or moment of the same object, and exact in composition across depth and width, with a quantified error at finite depth that vanishes in the joint limit. It is structural for residual networks, transformers, and recurrent networks (RNNs, LSTMs, SSMs), each discretizing the same class of equations, at a named and quantified approximation error. A single deformation parameter varepsilon unifies all four perspectives (network, tropical algebra, viscous PDE, convex optimization) in a commutative diagram closed under Lipschitz conditions. Quantitative consequences include: the minimax optimal generalization rate O(n-1/(d+2)) for fixed t; adversarial robustness controlled by varepsilon; backpropagation as the co-state equation of the Hamiltonian system for residual networks (Pontryagin Maximum Principle); scaling exponents consistent with data intrinsic dimension via PDE quadrature; and a closed-form O(N) influence function (softmax attribution weights πj) whose entropy landscape undergoes fold bifurcations as varepsilon increases, each merging attribution basins.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 66 references that are in the map. First posted on arXiv 27 May 2026.
Builds on
14 of 66 references.
References · 14
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 879
- 2017Deep Neural Networks as Gaussian Processescited by 622
- 2020Scaling Laws for Autoregressive Generative Modelingcited by 522
- 2021Explaining Neural Scaling Lawscited by 363
- 2020A Neural Scaling Law from the Dimension of the Data Manifoldcited by 52
- 2024Transformers Learn Nonlinear Features In Context: Nonconvex Mean-field Dynamics on the Attention Landscapecited by 46
- 2019Deep Learning Theory Review: An Optimal Control and Dynamical Systems Perspectivecited by 27
- 2019Overcoming the curse of dimensionality for some Hamilton--Jacobi partial differential equations via neural network architecturescited by 22
- 2024Scaling Laws for Linear Complexity Language Modelscited by 19
- 2022Neural Networks are Decision Treescited by 17
- 2025Neural ODE Transformers: Analyzing Internal Dynamics and Adaptive Fine-tuningcited by 13
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2023Stabilizing RNN Gradients through Pre-training
- 2025Hyperbolic-PDE GNN: Spectral Graph Neural Networks in the Perspective of A System of Hyperbolic Partial Differential Equations
- 2026Traversing the solution space of neural networks with Hessian Null Space Continuation
- 2025High-Dimensional Learning Dynamics of Quantized Models with Straight-Through Estimator
- 2025Optimal generalisation and learning transition in extensive-width shallow neural networks near interpolation
- 2023Efficient Neural PDE-Solvers using Quantization Aware Training
- 2026Representation Gap: Explaining the Unreasonable Effectiveness of Neural Networks from a Geometric Perspective
- 2024Neural Hamilton: Can A.I. Understand Hamiltonian Mechanics?