cs.LGJul 25, 2026

Finite-Time Analysis of the Natural Policy Gradient in Finite-Horizon Markov Decision Processes

Authors: Asha BaruaSajad Khodadadian

Organizations: Grado Department of Industrial and Systems Engineering, Virginia Polytechnic Institute and State University, Blacksburg, Virginia 24061, USA.

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 ηt=ηη_t=η, we prove that NPG converges sublinearly with a rate of O(H2/t)\mathcal{O}(H^{2}/t) after tt iterations, where HH 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((11ϑρ)t)\mathcal{O}\left(\left(1-\frac{1}{\vartheta_ρ}\right)^t\right) for a problem-dependent constant ϑρ>1\vartheta_ρ> 1, and the horizon-only robust schedule of the form ηt=η0(H/(H1))tη_t=η_0(H/(H-1))^t where η0>0η_0>0 and H2H \geq 2, attains this same geometric rate.

Explore similar work

CardsList