Geometry and convergence of natural policy gradient methods
Not on the map. Why?
Abstract (arXiv)
We study the convergence of several natural policy gradient (NPG) methods in infinite-horizon discounted Markov decision processes with regular policy parametrizations. For a variety of NPGs and reward functions we show that the trajectories in state-action space are solutions of gradient flows with respect to Hessian geometries, based on which we obtain global convergence guarantees and convergence rates. In particular, we show linear convergence for unregularized and regularized NPG flows with the metrics proposed by Kakade and Morimura and co-authors by observing that these arise from the Hessian geometries of conditional entropy and entropy respectively. Further, we obtain sublinear convergence rates for Hessian geometries arising from other convex functions like log-barriers. Finally, we interpret the discrete-time NPG methods with regularized rewards as inexact Newton methods if the NPG is defined with respect to the Hessian geometry of the regularizer. This yields local quadratic convergence rates of these methods for step size equal to the penalization strength.
Key facts. Cited by 12 papers in the map (data through 14 Oct 2026). Builds on 40 of its 83 references that are in the map. First posted on arXiv 3 Nov 2022.
Builds on
40 of 83 references.
References · 40
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2013Playing Atari with Deep Reinforcement Learningcited by 3,274
- 2015Trust Region Policy Optimizationcited by 2,580
- 2021Convex optimizationcited by 2,515
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2014New insights and perspectives on the natural gradient methodcited by 443
- 2020On the Global Convergence Rates of Softmax Policy Gradient Methodscited by 264
- 2018Global Convergence of Policy Gradient Methods for the Linear Quadratic Regulatorcited by 250
- 2017A unified view of entropy-regularized Markov decision processescited by 182
- 2020Fast Global Convergence of Natural Policy Gradient Methods with Entropy Regularizationcited by 176
- 2021Policy Mirror Descent for Reinforcement Learning: Linear Convergence, New Sampling Complexity, and Generalized Problem Classescited by 156
- 2019Global Optimality Guarantees For Policy Gradient Methodscited by 150
28 more
- 2019Global Convergence of Policy Gradient Methods to (Almost) Locally Optimal Policiescited by 124
- 2022On the Convergence Rates of Policy Gradient Methodscited by 119
- 2021Policy Mirror Descent for Regularized Reinforcement Learning: A Generalized Framework with Linear Convergencecited by 84
- 2021Reward is enough for convex MDPscited by 84
- 2019A Survey of Deep Reinforcement Learning in Video Gamescited by 73
- 2020On Linear Convergence of Policy Gradient Methods for Finite MDPscited by 52
- 2021Softmax Policy Gradient Methods Can Take Exponential Time to Convergecited by 48
- 2021On the Linear convergence of Natural Policy Gradient Algorithmcited by 48
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policiescited by 46
- 2015Natural Neural Networkscited by 44
- 2021Leveraging Non-uniformity in First-order Non-convex Optimizationcited by 44
- 2022Convergence and sample complexity of natural policy gradient primal-dual methods for constrained MDPscited by 27
- 2018Hessian Riemannian gradient flows in convex programmingcited by 26
- 2020Optimization of Graph Neural Networks with Natural Gradient Descentcited by 26
- 2018An explicit analysis of the entropic penalty in linear programmingcited by 24
- 2020Efficient Wasserstein Natural Gradients for Reinforcement Learningcited by 23
- 2022Convergence of Policy Gradient for Entropy Regularized MDPs with Neural Network Approximation in the Mean-Field Regimecited by 21
- 2018Natural gradient via optimal transportcited by 20
- 2021Bregman Gradient Policy Optimizationcited by 17
- 2022Linear Convergence for Natural Policy Gradient with Log-linear Policy Parametrizationcited by 17
- 2019Kernelized Wasserstein Natural Gradientcited by 12
- 2022Invariance Properties of the Natural Gradient in Overparametrised Systemscited by 11
- 2021The Geometry of Memoryless Stochastic Policy Optimization in Infinite-Horizon POMDPscited by 9
- 2022Efficient Natural Gradient Descent Methods for Large-Scale PDE-Based Optimization Problemscited by 7
- 2021Hessian informed mirror descentcited by 6
- 2021Convergence and Optimality of Policy Gradient Methods in Weakly Smooth Settingscited by 5
- 2021Dimension-Free Rates for Natural Policy Gradient in Multi-Agent Reinforcement Learningcited by 4
- 2022Solving infinite-horizon POMDPs with memoryless stochastic policies in state-action spacecited by 2
Built on by
Citing papers per year: 2023: 1; 2024: 4; 2025: 4; 2026: 3.
12 papers.
- 2024Embedding Safety into RL: A New Take on Trust Region Methodscited by 10
- 2024Fisher-Rao Gradient Flows of Linear Programs and State-Action Natural Policy Gradientscited by 5
- 2025Central Path Proximal Policy Optimizationcited by 3
- 2023Policy Optimization via Adv2: Adversarial Learning on Advantage Functionscited by 2
- 2024Linear Convergence of Independent Natural Policy Gradient in Games with Entropy Regularizationcited by 1
- 2024Optimal Rates of Convergence for Entropy Regularization in Discounted Markov Decision Processescited by 1
- 2025The Geometry of Nonlinear Reinforcement Learningcited by 1
- 2025Beyond Distributions: Geometric Action Control for Continuous Reinforcement Learningcited by 1
- 2026Cheap Thrills: Effective Amortized Optimization Using Inexpensive Labelscited by 1
- 2025Convergence Properties of Natural Gradient Descent for Minimizing KL Divergencecited by 0
- 2026Priced Motion Through Optimal Faces: A Normal-Fan Geometry for Non-Stationary Adversarial MDPscited by 0
- 2026The Dually Flat Geometry of Planning as Inferencecited by 0
- 2026The Dually Flat Geometry of Planning as Inference3 Sep 2026
- 2026Priced Motion Through Optimal Faces: A Normal-Fan Geometry for Non-Stationary Adversarial MDPs27 Jun 2026
- 2026Cheap Thrills: Effective Amortized Optimization Using Inexpensive Labels5 Mar 2026
- 2025Beyond Distributions: Geometric Action Control for Continuous Reinforcement Learning11 Nov 2025
- 2025The Geometry of Nonlinear Reinforcement Learning1 Sep 2025
- 2025Central Path Proximal Policy Optimization31 May 2025
- 2025Convergence Properties of Natural Gradient Descent for Minimizing KL Divergence27 Apr 2025
- 2024Embedding Safety into RL: A New Take on Trust Region Methods5 Nov 2024
- 2024Optimal Rates of Convergence for Entropy Regularization in Discounted Markov Decision Processes6 Jun 2024
- 2024Linear Convergence of Independent Natural Policy Gradient in Games with Entropy Regularization4 May 2024
- 2024Fisher-Rao Gradient Flows of Linear Programs and State-Action Natural Policy Gradients28 Mar 2024
- 2023Policy Optimization via Adv2: Adversarial Learning on Advantage Functions25 Oct 2023