mlsanityData to 7 Oct 2026

Global linear convergence of entropy-regularized softmax policy gradient beyond tabular MDPs

arXiv 2605.24939 ↗First posted 24 May 2026cs.LG math.OCcited by 0

Not on the map. Why?

Abstract (arXiv)

We study the global convergence of policy gradient for infinite-horizon entropy-regularized Markov decision processes (MDPs) with continuous state and action spaces. We consider log-linear softmax policies with linear function approximation, which extend the tabular softmax parameterization while retaining a tractable policy class. Under Qπτ-realizability for the regularized state-action value function, we first establish a non-uniform Polyak--Lojasiewicz (PL) inequality. The non-uniformity arises through degeneracy of constants associated with the policy geometry, namely the Fisher information matrix or an uncentered feature covariance matrix. We then identify two feature regimes under which this non-uniform constant can be bounded along the gradient flow. For full-affine-span features, we prove radial unboundedness of the KL regularizer and show that the smallest eigenvalue of the Fisher information matrix remains bounded below by an initialization-dependent positive constant. For simplex-valued features, we prove an analogous radial unboundedness result in the subspace orthogonal to the all-ones vector and obtain a uniform lower bound for the smallest eigenvalue of the uncentered covariance matrix. These results imply global linear convergence of the regularized objective along the gradient flow, i.e. suboptimality decaying as O(e-Ct) for some C>0. Our analysis extends the global convergence theory of entropy-regularized softmax policy gradient beyond the tabular setting of Agarwal et al. (2020); Bhandari and Russo (2024); Mei et al. (2020).

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 37 references that are in the map. First posted on arXiv 24 May 2026.

Builds on

22 of 37 references.

References · 22

  1. 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
  2. 2017Reinforcement Learning with Deep Energy-Based Policiescited by 659
  3. 2019A Theory of Regularized Markov Decision Processescited by 279
  4. 2020On the Global Convergence Rates of Softmax Policy Gradient Methodscited by 258
  5. 2018Global Convergence of Policy Gradient Methods for the Linear Quadratic Regulatorcited by 211
  6. 2017A unified view of entropy-regularized Markov decision processescited by 174
  7. 2019On the Theory of Policy Gradient Methods: Optimality, Approximation, and Distribution Shiftcited by 174
  8. 2021Policy Mirror Descent for Reinforcement Learning: Linear Convergence, New Sampling Complexity, and Generalized Problem Classescited by 149
  9. 2019Global Optimality Guarantees For Policy Gradient Methodscited by 146
  10. 2015Taming the Noise in Reinforcement Learning via Soft Updatescited by 125
  11. 2019Frequentist Regret Bounds for Randomized Least-Squares Value Iterationcited by 100
  12. 2021Softmax Policy Gradient Methods Can Take Exponential Time to Convergecited by 46
10 more
  1. 2019LQR through the Lens of First Order Methods: Discrete-time Casecited by 30
  2. 2021Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisitingcited by 27
  3. 2022Policy Optimization over General State and Action Spacescited by 21
  4. 2022Convergence of Policy Gradient for Entropy Regularized MDPs with Neural Network Approximation in the Mean-Field Regimecited by 20
  5. 2022Convergence of policy gradient methods for finite-horizon exploratory linear-quadratic control problemscited by 19
  6. 2023A Fisher-Rao gradient flow for entropy-regularised Markov decision processes in Polish spacescited by 19
  7. 2021Linear Convergence of Entropy-Regularized Natural Policy Gradient with Linear Function Approximationcited by 15
  8. 2021Remarks on input to state stability of perturbed gradient flows, motivated by model-free feedback control learningcited by 7
  9. 2022Polyak-L ojasiewicz inequality on the space of measures and convergence of mean-field birth-death processescited by 6
  10. 2025Rethinking the Global Convergence of Softmax Policy Gradient with Linear Function Approximation: The Case of Multi-Armed Banditscited by 3

Built on by

Not cited yet.