Approximate Newton Policy Gradient Algorithms
- 26 September 2023
- journal article
- Published by Society for Industrial & Applied Mathematics (SIAM) in SIAM Journal on Scientific Computing
- Vol. 45 (5) , A2585-A2609
Abstract
Policy gradient algorithms have been widely applied to Markov decision processes and reinforcement learning problems in recent years. Regularization with various entropy functions is often used to encourage exploration and improve stability. This paper proposes an approximate Newton method for the policy gradient algorithm with entropy regularization. In the case of Shannon entropy, the resulting algorithm reproduces the natural policy gradient algorithm. For other entropy functions, this method results in brand-new policy gradient algorithms. We prove that all these algorithms enjoy Newton-type quadratic convergence and that the corresponding gradient flow converges globally to the optimal solution. We use synthetic and industrial-scale examples to demonstrate that the proposed approximate Newton method typically converges in single-digit iterations, often orders of magnitude faster than other state-of-the-art algorithms.Keywords
All Related Versions
This publication has 14 references indexed in Scilit:
- Rates of superlinear convergence for classical quasi-Newton methodsMathematical Programming, 2021
- Mirror Descent Algorithms for Minimizing Interacting Free EnergyJournal of Scientific Computing, 2020
- The Information Geometry of Mirror DescentIEEE Transactions on Information Theory, 2015
- Primal-dual subgradient methods for convex problemsMathematical Programming, 2007
- Convex Optimization2004
- Fundamentals of Convex Analysis2001
- A Comparison of Iterative Methods for Solving Nonsymmetric Linear SystemsActa Applicandae Mathematicae, 1998
- Simple statistical gradient-following algorithms for connectionist reinforcement learningMachine Learning, 1992
- Bi-CGSTAB: A Fast and Smoothly Converging Variant of Bi-CG for the Solution of Nonsymmetric Linear SystemsSIAM Journal on Scientific and Statistical Computing, 1992
- A characterization of superlinear convergence and its application to quasi-Newton methodsMathematics of Computation, 1974