mlsanityData to 14 Oct 2026

Amortized Score-Hamiltonian Policy Iteration: A Grid-Free Scheme for Relaxed Stochastic Control Problems

arXiv 2610.04285 ↗First posted 3 Oct 2026math.OC cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

We develop an amortized, grid-free implementation of continuous Langevin dynamics based policy-value iteration for entropy-regularized, infinite-horizon relaxed stochastic control problems. The improvement rate of the exact iteration is a discounted aggregate of relative Fisher information between the policy and the Gibbs law of its Hamiltonian. The associated score residual is the velocity with which the control's Langevin dynamics transport its law. We project this velocity onto a conditional sampler shared across states, instead of one Langevin dynamics per state, and the value dynamics onto a parametric critic, estimating both projections at sampled states to obtain coupled actor--critic flows. The score loss measures the actor's agreement with the current critic, while the policy-evaluation residual measures the critic's agreement with the actor. We also derive gradient and Hessian residuals, including a Feynman--Kac representation for the gradient equation, to control errors not detected by the projected value iteration. An exact decomposition of the HJB residual combines these errors into a policy-suboptimality bound under verification and logarithmic Sobolev assumptions. In the linear-quadratic class, both projections are exact and recover the pointwise iteration, and we provide numerical experiments on general models to demonstrate the coupled actor--critic learning in high-dimensions.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 38 of its 62 references that are in the map. First posted on arXiv 3 Oct 2026.

Builds on

38 of 62 references.

References · 38

  1. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  2. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
  3. 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
  4. 2017DGM: A deep learning algorithm for solving partial differential equationscited by 678
  5. 2017Reinforcement Learning with Deep Energy-Based Policiescited by 669
  6. 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learningcited by 562
  7. 2017Solving high-dimensional partial differential equations using deep learningcited by 556
  8. 2023IDQL: Implicit Q-Learning as an Actor-Critic Method with Diffusion Policiescited by 325
  9. 2024Diffusion Policy Policy Optimizationcited by 280
  10. 2023Learning a Diffusion Model Policy from Rewards via Q-Score Matchingcited by 138
  11. 2021Policy Gradient and Actor-Critic Learning in Continuous Time and Space: Theory and Algorithmscited by 101
  12. 2024Diffusion Actor-Critic with Entropy Regulatorcited by 94
26 more
  1. 2022q-Learning in Continuous Timecited by 91
  2. 2023Diffusion Models for Reinforcement Learning: A Surveycited by 91
  3. 2021Policy Evaluation and Temporal-Difference Learning in Continuous Time and Space: A Martingale Approachcited by 84
  4. 2017Machine learning approximation algorithms for high-dimensional fully nonlinear partial differential equations and second-order backward stochastic differential equationscited by 65
  5. 2022Neural Galerkin Schemes with Active Learning for High-Dimensional Evolution Equationscited by 56
  6. 2021Evolutional Deep Neural Networkcited by 50
  7. 2018Policy Optimization as Wasserstein Gradient Flowscited by 34
  8. 2021Exploratory HJB equations and their convergencecited by 32
  9. 2023Policy Optimization for Continuous Reinforcement Learningcited by 31
  10. 2022Convergence of policy gradient methods for finite-horizon exploratory linear-quadratic control problemscited by 25
  11. 2023A Fisher-Rao gradient flow for entropy-regularised Markov decision processes in Polish spacescited by 23
  12. 2020Regularity and stability of feedback relaxed controlscited by 13
  13. 2023Fast Policy Learning for Linear Quadratic Control with Entropy Regularizationcited by 13
  14. 2025Accuracy of Discretely Sampled Stochastic Policies in Continuous-time Reinforcement Learningcited by 13
  15. 2024Entropy annealing for policy mirror descent in continuous time and spacecited by 12
  16. 2024Policy Iteration for Exploratory Hamilton--Jacobi--Bellman Equationscited by 12
  17. 2024Regret of exploratory policy improvement and q-learningcited by 12
  18. 2020A modified MSA for stochastic control problemscited by 10
  19. 2022Convergence of Policy Iteration for Entropy-Regularized Stochastic Control Problemscited by 10
  20. 2023A Policy Gradient Framework for Stochastic Optimal Control Problems with Global Convergence Guaranteecited by 10
  21. 2020Gradient Flows for Regularized Stochastic Control Problemscited by 9
  22. 2024Solving Time-Continuous Stochastic Optimal Control Problems: Algorithm Design and Convergence Analysis of Actor-Critic Flowcited by 8
  23. 2022Linear convergence of a policy gradient method for some finite horizon continuous time control problemscited by 7
  24. 2024Convergence Analysis for Entropy-Regularized Control Problems: A Probabilistic Approachcited by 6
  25. 2024Mirror Descent for Stochastic Control Problems with Measure-valued Controlscited by 4
  26. 2025Continuous Q-Score Matching: Diffusion Guided Reinforcement Learning for Continuous-Time Controlcited by 4

Built on by

Not cited yet.