Policy Mirror Descent for Reinforcement Learning: Linear Convergence, New Sampling Complexity, and Generalized Problem Classes
Not on the map. Why?
Abstract (arXiv)
We present new policy mirror descent (PMD) methods for solving reinforcement learning (RL) problems with either strongly convex or general convex regularizers. By exploring the structural properties of these overall highly nonconvex problems we show that the PMD methods exhibit fast linear rate of convergence to the global optimality. We develop stochastic counterparts of these methods, and establish an cal O(1/ε) (resp., cal O(1/ε2)) sampling complexity for solving these RL problems with strongly (resp., general) convex regularizers using different sampling schemes, where ε denote the target accuracy. We further show that the complexity for computing the gradients of these regularizers, if necessary, can be bounded by cal O(logγ ε) [(1-γ)L/μ]1/2log (1/ε) (resp., cal O (logγ ε ) (L/ε)1/2)for problems with strongly (resp., general) convex regularizers. Here γ denotes the discounting factor. To the best of our knowledge, these complexity bounds, along with our algorithmic developments, appear to be new in both optimization and RL literature. The introduction of these convex regularizers also greatly expands the flexibility and applicability of RL models.
Key facts. Cited by 149 papers in the map (data through 7 Oct 2026). Builds on 10 of its 29 references that are in the map. First posted on arXiv 30 Jan 2021.
Builds on
10 of 29 references.
References · 10
- 2020On the Global Convergence Rates of Softmax Policy Gradient Methodscited by 258
- 2020Fast Global Convergence of Natural Policy Gradient Methods with Entropy Regularizationcited by 170
- 2019Neural Policy Gradient Methods: Global Optimality and Rates of Convergencecited by 157
- 2019Adaptive Trust Region Policy Optimization: Global Convergence and Faster Rates for Regularized MDPscited by 113
- 2020Mirror Descent Policy Optimizationcited by 102
- 2019Neural Proximal/Trust Region Policy Optimization Attains Globally Optimal Policycited by 42
- 2020Simple and optimal methods for stochastic variational inequalities, I: operator extrapolationcited by 35
- 2013On the Convergence Properties of Non-Euclidean Extragradient Methods for Variational Inequalities with Generalized Monotone Operatorscited by 29
- 2020Simple and optimal methods for stochastic variational inequalities, II: Markovian noise and policy evaluation in reinforcement learningcited by 22
- 2020Improving Sample Complexity Bounds for (Natural) Actor-Critic Algorithmscited by 14
Built on by
Citing papers per year: 2021: 17; 2022: 28; 2023: 25; 2024: 26; 2025: 23; 2026: 30.
149 papers.
Sort:
By area:
- 2022On the Convergence Rates of Policy Gradient Methodscited by 115
- 2021Provable Benefits of Actor-Critic Methods for Offline Reinforcement Learningcited by 109
- 2021Policy Mirror Descent for Regularized Reinforcement Learning: A Generalized Framework with Linear Convergencecited by 83
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 67
- 2021Fast Policy Extragradient Methods for Competitive Games with Entropy Regularizationcited by 61
- 2025Efficient Online Reinforcement Learning for Diffusion Policycited by 61
- 2022Independent Policy Gradient for Large-Scale Markov Potential Games: Sharper Rates, Function Approximation, and Game-Agnostic Convergencecited by 54
- 2023Stochastic Policy Gradient Methods: Improved Sample Complexity for Fisher-non-degenerate Policiescited by 53
- 2021On the Linear convergence of Natural Policy Gradient Algorithmcited by 48
- 2021Softmax Policy Gradient Methods Can Take Exponential Time to Convergecited by 46
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policiescited by 44
- 2024Principled Penalty-based Methods for Bilevel Reinforcement Learning and RLHFcited by 41
38 more
- 2021Greedification Operators for Policy Optimization: Investigating Forward and Reverse KL Divergencescited by 40
- 2024Sharp Analysis for KL-Regularized Contextual Bandits and RLHFcited by 32
- 2022Mirror Learning: A Unifying Framework of Policy Optimisationcited by 31
- 2022First-order Policy Optimization for Robust Markov Decision Processcited by 31
- 2022Stochastic first-order methods for average-reward Markov decision processescited by 30
- 2023A Novel Framework for Policy Mirror Descent with General Parameterization and Linear Convergencecited by 29
- 2023Last-Iterate Convergent Policy Gradient Primal-Dual Methods for Constrained MDPscited by 27
- 2021Finite-Sample Analysis of Off-Policy Natural Actor-Critic Algorithmcited by 26
- 2021Beyond Exact Gradients: Convergence of Stochastic Soft-Max Policy Gradient Methods with Entropy Regularizationcited by 25
- 2022Faster Last-iterate Convergence of Policy Optimization in Zero-Sum Markov Gamescited by 25
- 2023The Role of Baselines in Policy Gradient Optimizationcited by 25
- 2022Policy Mirror Ascent for Efficient and Independent Learning in Mean Field Gamescited by 24
- 2023Improved Sample Complexity Analysis of Natural Policy Gradient Algorithm with General Parameterization for Infinite Horizon Discounted Reward Markov Decision Processescited by 24
- 2021Finite-Sample Analysis of Off-Policy Natural Actor-Critic with Linear Function Approximationcited by 22
- 2021Policy Optimization for Constrained MDPs with Provable Fast Global Convergencecited by 22
- 2022Policy Optimization over General State and Action Spacescited by 21
- 2023A Fisher-Rao gradient flow for entropy-regularised Markov decision processes in Polish spacescited by 19
- 2021Faster Algorithm and Sharper Analysis for Constrained Markov Decision Processcited by 18
- 2021Understanding the Effect of Stochasticity in Policy Optimizationcited by 18
- 2022Regularized Gradient Descent Ascent for Two-Player Zero-Sum Markov Gamescited by 18
- 2023Reinforcement Learning with General Utilities: Simpler Variance Reduction and Large State-Action Spacecited by 18
- 2025One-Step Flow Policy Mirror Descentcited by 18
- 2022Finite-Time Analysis of Entropy-Regularized Neural Natural Actor-Critic Algorithmcited by 17
- 2022Linear Convergence for Natural Policy Gradient with Log-linear Policy Parametrizationcited by 17
- 2023Optimal Convergence Rate for Exact Policy Mirror Descent in Discounted Markov Decision Processescited by 17
- 2023Improved Regret for Efficient Online Reinforcement Learning with Linear Function Approximationcited by 16
- 2021Linear Convergence of Entropy-Regularized Natural Policy Gradient with Linear Function Approximationcited by 15
- 2021Bregman Gradient Policy Optimizationcited by 15
- 2022Anchor-Changing Regularized Natural Policy Gradient for Multi-Objective Reinforcement Learningcited by 15
- 2023A Finite-Sample Analysis of Payoff-Based Independent Learning in Zero-Sum Stochastic Gamescited by 15
- 2023Policy Gradient Algorithms for Robust MDPs with Non-Rectangular Uncertainty Setscited by 15
- 2022Homotopic Policy Mirror Descent: Policy Convergence, Implicit Regularization, and Improved Sample Complexitycited by 14
- 2023Rate-Optimal Policy Optimization for Linear Markov Decision Processescited by 14
- 2022KL-Entropy-Regularized RL with a Generative Model is Minimax Optimalcited by 13
- 2021On the Sample Complexity and Metastability of Heavy-tailed Policy Search in Continuous Controlcited by 12
- 2022Finite-Time Analysis of Natural Actor-Critic for POMDPscited by 12
- 2022Geometry and convergence of natural policy gradient methodscited by 12
- 2022On the Hidden Biases of Policy Mirror Ascent in Continuous Action Spacescited by 11
50 most cited of 149 All 149 (JSON)
- 2026Fast Regularized Policy Mirror Descent with One-Step TD Updates30 Sep 2026
- 2026Towards Better Training Signal: Advantage Clipped Policy Optimization29 Sep 2026
- 2026An analysis of Mirror-Descent Soft Actor-Critic28 Sep 2026
- 2026PolicyAttention: Softmax Attention Implements Policy Mirror Descent for Closed-Loop Control24 Sep 2026
- 2026Bellman Policy Optimization14 Sep 2026
- 2026Local and Global Stability in Performative Reinforcement Learning6 Sep 2026
- 2026The Dually Flat Geometry of Planning as Inference3 Sep 2026
- 2026Non-KKT Accumulation in Entropic Mirror Descent3 Aug 2026
- 2026Finite-Time Analysis of the Natural Policy Gradient in Finite-Horizon Markov Decision Processes25 Jul 2026
- 2026Active Inference as a Convex Markov Decision Process22 Jul 2026
- 2026Mathematical methods of reinforcement learning8 Jul 2026
- 2026Entropy Regularization Improves Policy Robustness in Continuous-Time Reinforcement Learning3 Jul 2026
38 more
- 2026Priced Motion Through Optimal Faces: A Normal-Fan Geometry for Non-Stationary Adversarial MDPs27 Jun 2026
- 2026Low Variance Trust Region Optimization with Independent Actors and Sequential Updates in Cooperative Multi-agent Reinforcement Learning24 Jun 2026
- 2026Fast Convergence of Policy Regret in Learning Stochastic Optimal Control25 May 2026
- 2026Global linear convergence of entropy-regularized softmax policy gradient beyond tabular MDPs24 May 2026
- 2026Refined Analysis of Entropy-Regularized Actor-Critic23 May 2026
- 2026Delightful Gradients Accelerate Corner Escape12 May 2026
- 2026Revisiting Policy Gradients for Restricted Policy Classes: Escaping Myopic Local Optima with k-step Policy Gradients11 May 2026
- 2026Natural Policy Gradient as Doubly Smoothed Policy Iteration: A Bellman-Operator Framework11 May 2026
- 2026Value Mirror Descent for Reinforcement Learning7 Apr 2026
- 2026Flow Matching Policy Optimization with Mirror Descent and Entropy Constraints18 Mar 2026
- 2026Multi-Agent Reinforcement Learning Counteracts Delayed CSI in Multi-Satellite Systems17 Mar 2026
- 2026A New Kernel Regularity Condition for Distributed Mirror Descent: Broader Coverage and Simpler Analysis13 Mar 2026
- 2026Actor-Accelerated Policy Dual Averaging for Reinforcement Learning in Continuous Action Spaces10 Mar 2026
- 2026Wasserstein Proximal Policy Gradient3 Mar 2026
- 2026Training Generalizable Collaborative Agents via Strategic Risk Aversion25 Feb 2026
- 2026Approximation of Log-Partition Function in Policy Mirror Descent Induces Implicit Regularization for LLM Post-Training5 Feb 2026
- 2026Solving Stochastic Variational Inequalities without the Bounded Variance Assumption5 Feb 2026
- 2026Clipping-Free Policy Optimization for Large Language Models30 Jan 2026
- 2025Policy Mirror Descent with Temporal Difference Learning: Sample Complexity under Online Markov Data30 Dec 2025
- 2025Non-Asymptotic Global Convergence of PPO-Clip18 Dec 2025
- 2025Auto-exploration for online reinforcement learning6 Dec 2025
- 2025Natural Gradient VI: Guarantees for Non-Conjugate Models22 Oct 2025
- 2025Why Policy Gradient Algorithms Work for Undiscounted Total-Reward MDPs21 Oct 2025
- 2025Asynchronous Policy Gradient Aggregation for Efficient Distributed Reinforcement Learning29 Sep 2025
- 2025MDP modeling for multi-stage stochastic programs26 Sep 2025
- 2025Reinforcement Learning with Discrete Diffusion Policies for Combinatorial Action Spaces26 Sep 2025
- 2025On the Convergence of Policy Mirror Descent with Temporal Difference Evaluation23 Sep 2025
- 2025The Geometry of Nonlinear Reinforcement Learning1 Sep 2025
- 2025One-Step Flow Policy Mirror Descent31 Jul 2025
- 2025Statistical and Algorithmic Foundations of Reinforcement Learning19 Jul 2025
- 2025On the Effect of Regularization in Policy Mirror Descent11 Jul 2025
- 2025Convergence and Sample Complexity of First-Order Methods for Agnostic Reinforcement Learning6 Jul 2025
- 2025PPO in the Fisher-Rao geometry4 Jun 2025
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPs26 May 2025
- 2025Ordering-based Conditions for Global Convergence of Policy Gradient Methods2 Apr 2025
- 2025Analysis of On-policy Policy Gradient Methods under the Distribution Mismatch28 Mar 2025
- 2025Convergence of Policy Mirror Descent Beyond Compatible Function Approximation16 Feb 2025
- 2025Small steps no more: Global convergence of stochastic gradient bandits for arbitrary learning rates11 Feb 2025
50 newest of 149 All 149 (JSON)
- 2021Provable Benefits of Actor-Critic Methods for Offline Reinforcement Learningcited by 109
- 2023Inverse Reinforcement Learning with the Average Reward Criterioncited by 8
- 2024Towards the Transferability of Rewards Recovered via Regularized Inverse Reinforcement Learningcited by 6
- 2024Exploiting Structure in Offline Multi-Agent RL: The Benefits of Low Interaction Rankcited by 6
- 2022Robust Imitation via Mirror Descent Inverse Reinforcement Learningcited by 5
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPscited by 5
- 2024Convergence of a model-free entropy-regularized inverse reinforcement learning algorithmcited by 2
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPs26 May 2025
- 2024Exploiting Structure in Offline Multi-Agent RL: The Benefits of Low Interaction Rank1 Oct 2024
- 2024Towards the Transferability of Rewards Recovered via Regularized Inverse Reinforcement Learning3 Jun 2024
- 2024Convergence of a model-free entropy-regularized inverse reinforcement learning algorithm25 Mar 2024
- 2023Inverse Reinforcement Learning with the Average Reward Criterion24 May 2023
- 2022Robust Imitation via Mirror Descent Inverse Reinforcement Learning20 Oct 2022
- 2021Provable Benefits of Actor-Critic Methods for Offline Reinforcement Learning19 Aug 2021
- 2025Efficient Online Reinforcement Learning for Diffusion Policycited by 61
- 2025One-Step Flow Policy Mirror Descentcited by 18
- 2025Reinforcement Learning with Discrete Diffusion Policies for Combinatorial Action Spacescited by 7
- 2026Flow Matching Policy Optimization with Mirror Descent and Entropy Constraintscited by 1
- 2026Flow Matching Policy Optimization with Mirror Descent and Entropy Constraints18 Mar 2026
- 2025Reinforcement Learning with Discrete Diffusion Policies for Combinatorial Action Spaces26 Sep 2025
- 2025One-Step Flow Policy Mirror Descent31 Jul 2025
- 2025Efficient Online Reinforcement Learning for Diffusion Policy1 Feb 2025
- 2022On the Convergence Rates of Policy Gradient Methodscited by 115
- 2021Policy Mirror Descent for Regularized Reinforcement Learning: A Generalized Framework with Linear Convergencecited by 83
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 67
- 2021Fast Policy Extragradient Methods for Competitive Games with Entropy Regularizationcited by 61
- 2022Independent Policy Gradient for Large-Scale Markov Potential Games: Sharper Rates, Function Approximation, and Game-Agnostic Convergencecited by 54
- 2023Stochastic Policy Gradient Methods: Improved Sample Complexity for Fisher-non-degenerate Policiescited by 53
- 2021On the Linear convergence of Natural Policy Gradient Algorithmcited by 48
- 2021Softmax Policy Gradient Methods Can Take Exponential Time to Convergecited by 46
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policiescited by 44
- 2021Greedification Operators for Policy Optimization: Investigating Forward and Reverse KL Divergencescited by 40
- 2022Mirror Learning: A Unifying Framework of Policy Optimisationcited by 31
- 2022First-order Policy Optimization for Robust Markov Decision Processcited by 31
12 most cited of 131 All 131 (JSON)
- 2026Fast Regularized Policy Mirror Descent with One-Step TD Updates30 Sep 2026
- 2026An analysis of Mirror-Descent Soft Actor-Critic28 Sep 2026
- 2026PolicyAttention: Softmax Attention Implements Policy Mirror Descent for Closed-Loop Control24 Sep 2026
- 2026Local and Global Stability in Performative Reinforcement Learning6 Sep 2026
- 2026The Dually Flat Geometry of Planning as Inference3 Sep 2026
- 2026Non-KKT Accumulation in Entropic Mirror Descent3 Aug 2026
- 2026Finite-Time Analysis of the Natural Policy Gradient in Finite-Horizon Markov Decision Processes25 Jul 2026
- 2026Active Inference as a Convex Markov Decision Process22 Jul 2026
- 2026Mathematical methods of reinforcement learning8 Jul 2026
- 2026Entropy Regularization Improves Policy Robustness in Continuous-Time Reinforcement Learning3 Jul 2026
- 2026Priced Motion Through Optimal Faces: A Normal-Fan Geometry for Non-Stationary Adversarial MDPs27 Jun 2026
- 2026Low Variance Trust Region Optimization with Independent Actors and Sequential Updates in Cooperative Multi-agent Reinforcement Learning24 Jun 2026
12 newest of 131 All 131 (JSON)