Efficiently Solving MDPs with Stochastic Mirror Descent
Not on the map. Why?
Abstract (arXiv)
We present a unified framework based on primal-dual stochastic mirror descent for approximately solving infinite-horizon Markov decision processes (MDPs) given a generative model. When applied to an average-reward MDP with Atot total state-action pairs and mixing time bound tmix our method computes an ε-optimal policy with an expected widetildeO(tmix2 Atot ε-2) samples from the state-transition matrix, removing the ergodicity dependence of prior art. When applied to a γ-discounted MDP with Atot total state-action pairs our method computes an ε-optimal policy with an expected widetildeO((1-γ)-4 Atot ε-2) samples, matching the previous state-of-the-art up to a (1-γ)-1 factor. Both methods are model-free, update state values and policies simultaneously, and run in time linear in the number of samples taken. We achieve these results through a more general stochastic mirror descent framework for solving bilinear saddle-point problems with simplex and box domains and we demonstrate the flexibility of this framework by providing further applications to constrained MDPs.
Key facts. Cited by 52 papers in the map (data through 7 Oct 2026). First posted on arXiv 28 Aug 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 11; 2023: 8; 2024: 7; 2025: 6; 2026: 6.
52 papers.
Sort:
By area:
- 2021Reward is enough for convex MDPscited by 82
- 2021Towards Tight Bounds on the Sample Complexity of Average-reward MDPscited by 32
- 2022Stochastic first-order methods for average-reward Markov decision processescited by 30
- 2023Sharper Model-free Reinforcement Learning for Average-reward Markov Decision Processescited by 28
- 2023Last-Iterate Convergent Policy Gradient Primal-Dual Methods for Constrained MDPscited by 27
- 2021Decentralized Local Stochastic Extra-Gradient for Variational Inequalitiescited by 25
- 2022Proximal Point Imitation Learningcited by 25
- 2022Near Sample-Optimal Reduction-based Policy Learning for Average Reward MDPcited by 25
- 2021Quantum Algorithms for Reinforcement Learning with a Generative Modelcited by 24
- 2024Span-Based Optimal Sample Complexity for Weakly Communicating and General Average Reward MDPscited by 16
- 2021Bregman Gradient Policy Optimizationcited by 15
- 2021Distributed Methods with Compressed Communication for Solving Variational Inequalities, with Theoretical Guaranteescited by 15
38 more
- 2024Dealing with unbounded gradients in stochastic saddle-point optimizationcited by 15
- 2022Optimal variance-reduced stochastic approximation in Banach spacescited by 14
- 2022Optimal Algorithms for Decentralized Stochastic Variational Inequalitiescited by 14
- 2023First Order Methods with Markovian Noise: from Acceleration to Variational Inequalitiescited by 14
- 2023Reducing Blackwell and Average Optimality to Discounted MDPs via the Blackwell Discount Factorcited by 13
- 2023Optimal Sample Complexity for Average Reward Markov Decision Processescited by 13
- 2024Stochastic Halpern iteration in normed spaces and applications to reinforcement learningcited by 11
- 2022Smooth Monotone Stochastic Variational Inequalities and Saddle Point Problems: A Surveycited by 10
- 2024Finding good policies in average-reward Markov Decision Processes without prior knowledgecited by 10
- 2021Efficient Performance Bounds for Primal-Dual Reinforcement Learning from Demonstrationscited by 9
- 2021Distributed Saddle-Point Problems Under Similaritycited by 8
- 2022Efficient Global Planning in Large MDPs via Stochastic Primal-Dual Optimizationcited by 8
- 2023Similarity, Compression and Local Steps: Three Pillars of Efficient Communications for Distributed Variational Inequalitiescited by 7
- 2021Long-Term Resource Allocation Fairness in Average Markov Decision Process (AMDP) Environmentcited by 6
- 2022Learning Stationary Nash Equilibrium Policies in n-Player Stochastic Games with Independent Chainscited by 6
- 2022An Efficient Stochastic Algorithm for Decentralized Nonconvex-Strongly-Concave Minimax Optimizationcited by 6
- 2024Truncated Variance Reduced Value Iterationcited by 6
- 2023Span-Based Optimal Sample Complexity for Average Reward MDPscited by 5
- 2024Strongly-polynomial time and validation analysis of policy gradient methodscited by 5
- 2021Conic Blackwell Algorithm: Parameter-Free Convex-Concave Saddle-Point Solvingcited by 4
- 2021Distributionally Robust Optimization with Markovian Datacited by 4
- 2021Decentralized Personalized Federated Learning for Min-Max Problemscited by 4
- 2022Solving optimization problems with Blackwell approachabilitycited by 4
- 2024Computing Optimal Regularizers for Online Linear Optimizationcited by 4
- 2021Primal-Dual Stochastic Mirror Descent for MDPscited by 3
- 2022Algorithm for Constrained Markov Decision Process with Linear Convergencecited by 3
- 2025Span-Agnostic Optimal Sample Complexity and Oracle Inequalities for Average-Reward RLcited by 3
- 2025Efficiently Solving Discounted MDPs with Predictions on Transition Matricescited by 2
- 2021Near Optimal Policy Optimization via REPScited by 1
- 2023Scalable and Independent Learning of Nash Equilibrium Policies in n-Player Stochastic Games with Unknown Independent Chainscited by 1
- 2025Layer-wise Quantization for Quantized Optimistic Dual Averagingcited by 1
- 2025A Bit of Freedom Goes a Long Way: Classical and Quantum Algorithms for Reinforcement Learning under a Generative Modelcited by 1
- 2026Shuffling the Stochastic Mirror Descent via Dual Lipschitz Continuity and Kernel Conditioningcited by 1
- 2026Gaussian Approximation for Asynchronous Q-learningcited by 1
- 2026Safe, Scalable, and Accurate Bayes Posterior Sampling for Large-Data Generalized Linear Mixed Modelscited by 1
- 2025Apprenticeship learning with prior beliefs using inverse optimizationcited by 0
- 2025Shuffling Heuristic in Variational Inequalities: Establishing New Convergence Guaranteescited by 0
- 2026Sample Complexity of Average-Reward Q-Learning: From Single-agent to Federated Reinforcement Learningcited by 0
50 most cited of 52 All 52 (JSON)
- 2026Mathematical methods of reinforcement learning8 Jul 2026
- 2026Safe, Scalable, and Accurate Bayes Posterior Sampling for Large-Data Generalized Linear Mixed Models28 Apr 2026
- 2026Gaussian Approximation for Asynchronous Q-learning8 Apr 2026
- 2026Optimal Variance-Dependent Regret Bounds for Infinite-Horizon MDPs25 Mar 2026
- 2026Shuffling the Stochastic Mirror Descent via Dual Lipschitz Continuity and Kernel Conditioning17 Mar 2026
- 2026Sample Complexity of Average-Reward Q-Learning: From Single-agent to Federated Reinforcement Learning20 Jan 2026
- 2025Shuffling Heuristic in Variational Inequalities: Establishing New Convergence Guarantees4 Sep 2025
- 2025A Bit of Freedom Goes a Long Way: Classical and Quantum Algorithms for Reinforcement Learning under a Generative Model30 Jul 2025
- 2025Apprenticeship learning with prior beliefs using inverse optimization27 May 2025
- 2025Layer-wise Quantization for Quantized Optimistic Dual Averaging20 May 2025
- 2025Efficiently Solving Discounted MDPs with Predictions on Transition Matrices21 Feb 2025
- 2025Span-Agnostic Optimal Sample Complexity and Oracle Inequalities for Average-Reward RL16 Feb 2025
38 more
- 2024Computing Optimal Regularizers for Online Linear Optimization22 Oct 2024
- 2024Strongly-polynomial time and validation analysis of policy gradient methods28 Sep 2024
- 2024Finding good policies in average-reward Markov Decision Processes without prior knowledge27 May 2024
- 2024Truncated Variance Reduced Value Iteration21 May 2024
- 2024Stochastic Halpern iteration in normed spaces and applications to reinforcement learning19 Mar 2024
- 2024Span-Based Optimal Sample Complexity for Weakly Communicating and General Average Reward MDPs18 Mar 2024
- 2024Dealing with unbounded gradients in stochastic saddle-point optimization21 Feb 2024
- 2023Scalable and Independent Learning of Nash Equilibrium Policies in n-Player Stochastic Games with Unknown Independent Chains4 Dec 2023
- 2023Span-Based Optimal Sample Complexity for Average Reward MDPs22 Nov 2023
- 2023Optimal Sample Complexity for Average Reward Markov Decision Processes13 Oct 2023
- 2023Sharper Model-free Reinforcement Learning for Average-reward Markov Decision Processes28 Jun 2023
- 2023Last-Iterate Convergent Policy Gradient Primal-Dual Methods for Constrained MDPs20 Jun 2023
- 2023First Order Methods with Markovian Noise: from Acceleration to Variational Inequalities25 May 2023
- 2023Similarity, Compression and Local Steps: Three Pillars of Efficient Communications for Distributed Variational Inequalities15 Feb 2023
- 2023Reducing Blackwell and Average Optimality to Discounted MDPs via the Blackwell Discount Factor31 Jan 2023
- 2022An Efficient Stochastic Algorithm for Decentralized Nonconvex-Strongly-Concave Minimax Optimization5 Dec 2022
- 2022Near Sample-Optimal Reduction-based Policy Learning for Average Reward MDP1 Dec 2022
- 2022Efficient Global Planning in Large MDPs via Stochastic Primal-Dual Optimization21 Oct 2022
- 2022Proximal Point Imitation Learning22 Sep 2022
- 2022Smooth Monotone Stochastic Variational Inequalities and Saddle Point Problems: A Survey29 Aug 2022
- 2022Algorithm for Constrained Markov Decision Process with Linear Convergence3 Jun 2022
- 2022Stochastic first-order methods for average-reward Markov decision processes11 May 2022
- 2022Solving optimization problems with Blackwell approachability24 Feb 2022
- 2022Optimal Algorithms for Decentralized Stochastic Variational Inequalities6 Feb 2022
- 2022Learning Stationary Nash Equilibrium Policies in n-Player Stochastic Games with Independent Chains28 Jan 2022
- 2022Optimal variance-reduced stochastic approximation in Banach spaces21 Jan 2022
- 2021Efficient Performance Bounds for Primal-Dual Reinforcement Learning from Demonstrations28 Dec 2021
- 2021Quantum Algorithms for Reinforcement Learning with a Generative Model15 Dec 2021
- 2021Distributed Methods with Compressed Communication for Solving Variational Inequalities, with Theoretical Guarantees7 Oct 2021
- 2021Distributed Saddle-Point Problems Under Similarity22 Jul 2021
- 2021Bregman Gradient Policy Optimization23 Jun 2021
- 2021Decentralized Local Stochastic Extra-Gradient for Variational Inequalities15 Jun 2021
- 2021Decentralized Personalized Federated Learning for Min-Max Problems14 Jun 2021
- 2021Towards Tight Bounds on the Sample Complexity of Average-reward MDPs13 Jun 2021
- 2021Distributionally Robust Optimization with Markovian Data12 Jun 2021
- 2021Reward is enough for convex MDPs1 Jun 2021
- 2021Conic Blackwell Algorithm: Parameter-Free Convex-Concave Saddle-Point Solving27 May 2021
- 2021Near Optimal Policy Optimization via REPS17 Mar 2021
50 newest of 52 All 52 (JSON)
- 2021Reward is enough for convex MDPscited by 82
- 2021Towards Tight Bounds on the Sample Complexity of Average-reward MDPscited by 32
- 2022Stochastic first-order methods for average-reward Markov decision processescited by 30
- 2023Sharper Model-free Reinforcement Learning for Average-reward Markov Decision Processescited by 28
- 2023Last-Iterate Convergent Policy Gradient Primal-Dual Methods for Constrained MDPscited by 27
- 2021Decentralized Local Stochastic Extra-Gradient for Variational Inequalitiescited by 25
- 2022Near Sample-Optimal Reduction-based Policy Learning for Average Reward MDPcited by 25
- 2021Quantum Algorithms for Reinforcement Learning with a Generative Modelcited by 24
- 2024Span-Based Optimal Sample Complexity for Weakly Communicating and General Average Reward MDPscited by 16
- 2021Bregman Gradient Policy Optimizationcited by 15
- 2021Distributed Methods with Compressed Communication for Solving Variational Inequalities, with Theoretical Guaranteescited by 15
- 2024Dealing with unbounded gradients in stochastic saddle-point optimizationcited by 15
12 most cited of 49 All 49 (JSON)
- 2026Mathematical methods of reinforcement learning8 Jul 2026
- 2026Safe, Scalable, and Accurate Bayes Posterior Sampling for Large-Data Generalized Linear Mixed Models28 Apr 2026
- 2026Gaussian Approximation for Asynchronous Q-learning8 Apr 2026
- 2026Optimal Variance-Dependent Regret Bounds for Infinite-Horizon MDPs25 Mar 2026
- 2026Shuffling the Stochastic Mirror Descent via Dual Lipschitz Continuity and Kernel Conditioning17 Mar 2026
- 2026Sample Complexity of Average-Reward Q-Learning: From Single-agent to Federated Reinforcement Learning20 Jan 2026
- 2025Shuffling Heuristic in Variational Inequalities: Establishing New Convergence Guarantees4 Sep 2025
- 2025A Bit of Freedom Goes a Long Way: Classical and Quantum Algorithms for Reinforcement Learning under a Generative Model30 Jul 2025
- 2025Layer-wise Quantization for Quantized Optimistic Dual Averaging20 May 2025
- 2025Efficiently Solving Discounted MDPs with Predictions on Transition Matrices21 Feb 2025
- 2025Span-Agnostic Optimal Sample Complexity and Oracle Inequalities for Average-Reward RL16 Feb 2025
- 2024Computing Optimal Regularizers for Online Linear Optimization22 Oct 2024
12 newest of 49 All 49 (JSON)