Finite-Time Analysis of the Natural Policy Gradient in Finite-Horizon Markov Decision Processes
Asha Barua, Sajad Khodadadian
cs.LG, math.OC, stat.ML
Submitted: 2026-07-25
Comments: 33 pages, 3 figures (each with two subfigures), 1 table. Accepted at the Reinforcement Learning Conference (RLC 2026)
License: http://creativecommons.org/licenses/by/4.0/
The gist: Natural Policy Gradient (NPG) is a well-established Reinforcement Learning algorithm that underlies widely used methods such as Trust Region Policy Optimization and Proximal Policy Optimization, both
Terminology
Abstract
Natural Policy Gradient (NPG) is a well-established Reinforcement Learning algorithm that underlies widely used methods such as Trust Region Policy Optimization and Proximal Policy Optimization, both of which have demonstrated strong empirical success. In this paper, we study exact NPG in finite-horizon Markov Decision Processes with known dynamics and horizon-dependent transition kernels. We provide the first finite-time convergence guarantees for this algorithm in this setting, for which we consider both constant and increasing step size regimes. With a constant step size eta t= eta, we prove that NPG converges sublinearly with a rate of O(H squared/t) after t iterations, where H is the horizon length. We also extend this constant step size analysis to linear MDPs in an exact population-projection oracle under a full support projection distribution, recovering the same sublinear rate as in the tabular setting. Furthermore, with increasing step sizes, we prove that this algorithm achieves a linear convergence rate of O ((1-1 over rho) t) for a problem-dependent constant rho > 1, and the horizon-only robust schedule of the form eta t= eta 0(H/(H-1)) t where eta 0>0 and H at least 2, attains this same geometric rate.
Sources
- Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback
- Elementary Analysis of Policy Gradient Methods
- Proximal Policy Optimization Algorithms
- A Survey of Reinforcement Learning for Large Reasoning Models
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks