Learning Unknown Markov Decision Processes: A Thompson Sampling Approach
Not on the map. Why?
Abstract (arXiv)
We consider the problem of learning an unknown Markov Decision Process (MDP) that is weakly communicating in the infinite horizon setting. We propose a Thompson Sampling-based reinforcement learning algorithm with dynamic episodes (TSDE). At the beginning of each episode, the algorithm generates a sample from the posterior distribution over the unknown model parameters. It then follows the optimal stationary policy for the sampled model for the rest of the episode. The duration of each episode is dynamically determined by two stopping criteria. The first stopping criterion controls the growth rate of episode length. The second stopping criterion happens when the number of visits to any state-action pair is doubled. We establish tilde O(HS√AT) bounds on expected regret under a Bayesian setting, where S and A are the sizes of the state and action spaces, T is time, and H is the bound of the span. This regret bound matches the best available bound for weakly communicating MDPs. Numerical results show it to perform better than existing algorithms for infinite horizon MDPs.
Key facts. Cited by 65 papers in the map (data through 7 Oct 2026). First posted on arXiv 14 Sep 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 12; 2023: 10; 2024: 18; 2025: 7; 2026: 4.
65 papers.
Sort:
By area:
- 2021Recent Advances in Reinforcement Learning in Financecited by 102
- 2022Jump-Start Reinforcement Learningcited by 102
- 2021Reinforcement Learning, Bit by Bitcited by 59
- 2021Randomized Exploration for Reinforcement Learning with General Value Function Approximationcited by 49
- 2021Simple Agent, Complex Environment: Efficient Reinforcement Learning with Agent Statescited by 42
- 2022Stochastic first-order methods for average-reward Markov decision processescited by 30
- 2024MaxInfoRL: Boosting exploration in reinforcement learning through information gain maximizationcited by 30
- 2023Sharper Model-free Reinforcement Learning for Average-reward Markov Decision Processescited by 28
- 2023Regret Analysis of Policy Gradient Algorithm for Infinite Horizon Average Reward Markov Decision Processescited by 22
- 2021Nearly Minimax Optimal Regret for Learning Infinite-horizon Average-reward MDPs with Linear Function Approximationcited by 18
- 2021Improved Regret Bound and Experience Replay in Regularized Policy Iterationcited by 18
- 2021Concave Utility Reinforcement Learning with Zero-Constraint Violationscited by 18
38 more
- 2021Online Learning for Unknown Partially Observable MDPscited by 17
- 2021Online Learning for Stochastic Shortest Path Model via Posterior Samplingcited by 16
- 2021Adaptive Transmission Scheduling in Wireless Networks for Asynchronous Federated Learningcited by 15
- 2025Toward Efficient Exploration by Large Language Model Agentscited by 15
- 2024Bayesian Design Principles for Offline-to-Online Reinforcement Learningcited by 14
- 2022Model-Based Reinforcement Learning with Multinomial Logistic Function Approximationcited by 13
- 2024Achieving Tractable Minimax Optimal Regret in Average Reward MDPscited by 13
- 2022Optimistic Posterior Sampling for Reinforcement Learning with Few Samples and Tight Guaranteescited by 12
- 2022On learning Whittle index policy for restless bandits with scalable regretcited by 9
- 2023Social Contract AI: Aligning AI Assistants with Implicit Group Normscited by 9
- 2024Reinforcement Learning for Infinite-Horizon Average-Reward Linear MDPs via Approximation by Discounted-Reward MDPscited by 7
- 2023Safe Posterior Sampling for Constrained MDPs with Bounded Constraint Violationcited by 6
- 2024Sample-efficient Learning of Infinite-horizon Average-reward MDPs with General Function Approximationcited by 6
- 2022Augmented RBMLE-UCB Approach for Adaptive Control of Linear Quadratic Systemscited by 5
- 2023Bridging Imitation and Online Reinforcement Learning: An Optimistic Talecited by 5
- 2023Langevin Thompson Sampling with Logarithmic Communication: Bandits and Reinforcement Learningcited by 5
- 2025The Sample Complexity of Online Reinforcement Learning: A Multi-model Perspectivecited by 5
- 2021Learning Augmented Index Policy for Optimal Service Placement at the Network Edgecited by 4
- 2022Posterior Sampling for Continuing Environmentscited by 4
- 2021A Bayesian Learning Algorithm for Unknown Zero-sum Stochastic Games with an Arbitrary Opponentcited by 3
- 2022Efficient Algorithms for Learning to Control Bandits with Unobserved Contextscited by 3
- 2022Actor-Critic based Improper Reinforcement Learningcited by 3
- 2024NeoRL: Efficient Exploration for Nonepisodic RLcited by 3
- 2025Best Policy Learning from Trajectory Preference Feedbackcited by 3
- 2025No-Regret Thompson Sampling for Finite-Horizon Markov Decision Processes with Gaussian Processescited by 3
- 2022Slowly Changing Adversarial Bandit Algorithms are Efficient for Discounted MDPscited by 2
- 2023Bayesian Learning of Optimal Policies in Markov Decision Processes with Countably Infinite State-Spacecited by 2
- 2024Efficient Exploration in Average-Reward Constrained Reinforcement Learning: Achieving Near-Optimal Regret With Posterior Samplingcited by 2
- 2024Robust Thompson Sampling Algorithms Against Reward Poisoning Attackscited by 2
- 2024Generalized Bayesian deep reinforcement learningcited by 2
- 2025Minimax Optimal Reinforcement Learning with Quasi-Optimismcited by 2
- 2021Reinforcement Learning for Markovian Bandits: Is Posterior Sampling more Scalable than Optimism?cited by 1
- 2022An Empirical Evaluation of Posterior Sampling for Constrained Reinforcement Learningcited by 1
- 2024Policy Zooming: Adaptive Discretization-based Infinite-Horizon Average-Reward Reinforcement Learningcited by 1
- 2024A Federated Online Restless Bandit Framework for Cooperative Resource Allocationcited by 1
- 2024Learning Infinite-Horizon Average-Reward Linear Mixture MDPs of Bounded Spancited by 1
- 2025Q-learning with Posterior Samplingcited by 1
- 2025A Bit of Freedom Goes a Long Way: Classical and Quantum Algorithms for Reinforcement Learning under a Generative Modelcited by 1
50 most cited of 65 All 65 (JSON)
- 2026Finite Constant Frontiers and Auditable Regret Certificates for Average-Reward Reinforcement Learning7 Aug 2026
- 2026Bayesian learning for the stochastic shortest path problem3 Jun 2026
- 2026Learning Weakly Communicating Average-Reward CMDPs: Strong Duality and Improved Regret12 May 2026
- 2026Bayesian Learning in Episodic Zero-Sum Games21 Mar 2026
- 2025No-Regret Thompson Sampling for Finite-Horizon Markov Decision Processes with Gaussian Processes23 Oct 2025
- 2025A Bit of Freedom Goes a Long Way: Classical and Quantum Algorithms for Reinforcement Learning under a Generative Model30 Jul 2025
- 2025Q-learning with Posterior Sampling1 Jun 2025
- 2025Toward Efficient Exploration by Large Language Model Agents29 Apr 2025
- 2025Minimax Optimal Reinforcement Learning with Quasi-Optimism2 Mar 2025
- 2025Best Policy Learning from Trajectory Preference Feedback31 Jan 2025
- 2025The Sample Complexity of Online Reinforcement Learning: A Multi-model Perspective27 Jan 2025
- 2024MaxInfoRL: Boosting exploration in reinforcement learning through information gain maximization16 Dec 2024
38 more
- 2024Generalized Bayesian deep reinforcement learning16 Dec 2024
- 2024Risk-Aware Decision Making in Restless Bandits: Theory and Algorithms for Planning and Learning30 Oct 2024
- 2024Provably Adaptive Average Reward Reinforcement Learning for Metric Spaces25 Oct 2024
- 2024Robust Thompson Sampling Algorithms Against Reward Poisoning Attacks25 Oct 2024
- 2024Learning Infinite-Horizon Average-Reward Linear Mixture MDPs of Bounded Span19 Oct 2024
- 2024A Federated Online Restless Bandit Framework for Cooperative Resource Allocation12 Jun 2024
- 2024Achieving Tractable Minimax Optimal Regret in Average Reward MDPs3 Jun 2024
- 2024NeoRL: Efficient Exploration for Nonepisodic RL3 Jun 2024
- 2024Bayesian Design Principles for Offline-to-Online Reinforcement Learning31 May 2024
- 2024Efficient Exploration in Average-Reward Constrained Reinforcement Learning: Achieving Near-Optimal Regret With Posterior Sampling29 May 2024
- 2024Policy Zooming: Adaptive Discretization-based Infinite-Horizon Average-Reward Reinforcement Learning29 May 2024
- 2024Reinforcement Learning for Infinite-Horizon Average-Reward Linear MDPs via Approximation by Discounted-Reward MDPs23 May 2024
- 2024Thompson Sampling for Infinite-Horizon Discounted Decision Processes14 May 2024
- 2024Structured Reinforcement Learning for Delay-Optimal Data Transmission in Dense mmWave Networks25 Apr 2024
- 2024Sample-efficient Learning of Infinite-horizon Average-reward MDPs with General Function Approximation19 Apr 2024
- 2024A resource-constrained stochastic scheduling algorithm for homeless street outreach and gleaning edible food15 Mar 2024
- 2024Model-Free Approximate Bayesian Learning for Large-Scale Conversion Funnel Optimization12 Jan 2024
- 2023Social Contract AI: Aligning AI Assistants with Implicit Group Norms26 Oct 2023
- 2023Posterior Sampling-based Online Learning for Episodic POMDPs16 Oct 2023
- 2023Provably Efficient Exploration in Constrained Reinforcement Learning:Posterior Sampling Is All You Need27 Sep 2023
- 2023Regret Analysis of Policy Gradient Algorithm for Infinite Horizon Average Reward Markov Decision Processes5 Sep 2023
- 2023Learning Optimal Admission Control in Partially Observable Queueing Networks4 Aug 2023
- 2023Sharper Model-free Reinforcement Learning for Average-reward Markov Decision Processes28 Jun 2023
- 2023Langevin Thompson Sampling with Logarithmic Communication: Bandits and Reinforcement Learning15 Jun 2023
- 2023Bayesian Learning of Optimal Policies in Markov Decision Processes with Countably Infinite State-Space5 Jun 2023
- 2023Bridging Imitation and Online Reinforcement Learning: An Optimistic Tale20 Mar 2023
- 2023Safe Posterior Sampling for Constrained MDPs with Bounded Constraint Violation27 Jan 2023
- 2022Model-Based Reinforcement Learning with Multinomial Logistic Function Approximation27 Dec 2022
- 2022Posterior Sampling for Continuing Environments29 Nov 2022
- 2022Optimistic Posterior Sampling for Reinforcement Learning with Few Samples and Tight Guarantees28 Sep 2022
- 2022An Empirical Evaluation of Posterior Sampling for Constrained Reinforcement Learning8 Sep 2022
- 2022Actor-Critic based Improper Reinforcement Learning19 Jul 2022
- 2022Slowly Changing Adversarial Bandit Algorithms are Efficient for Discounted MDPs18 May 2022
- 2022Stochastic first-order methods for average-reward Markov decision processes11 May 2022
- 2022Jump-Start Reinforcement Learning5 Apr 2022
- 2022Double Thompson Sampling in Finite stochastic Games21 Feb 2022
- 2022On learning Whittle index policy for restless bandits with scalable regret7 Feb 2022
- 2022Efficient Algorithms for Learning to Control Bandits with Unobserved Contexts2 Feb 2022
50 newest of 65 All 65 (JSON)
- 2025Best Policy Learning from Trajectory Preference Feedbackcited by 3
- 2026Bayesian Learning in Episodic Zero-Sum Gamescited by 0
- 2026Bayesian Learning in Episodic Zero-Sum Games21 Mar 2026
- 2025Best Policy Learning from Trajectory Preference Feedback31 Jan 2025
- 2025Toward Efficient Exploration by Large Language Model Agentscited by 15
- 2025Toward Efficient Exploration by Large Language Model Agents29 Apr 2025
- 2021Recent Advances in Reinforcement Learning in Financecited by 102
- 2022Jump-Start Reinforcement Learningcited by 102
- 2021Reinforcement Learning, Bit by Bitcited by 59
- 2021Randomized Exploration for Reinforcement Learning with General Value Function Approximationcited by 49
- 2021Simple Agent, Complex Environment: Efficient Reinforcement Learning with Agent Statescited by 42
- 2022Stochastic first-order methods for average-reward Markov decision processescited by 30
- 2024MaxInfoRL: Boosting exploration in reinforcement learning through information gain maximizationcited by 30
- 2023Sharper Model-free Reinforcement Learning for Average-reward Markov Decision Processescited by 28
- 2023Regret Analysis of Policy Gradient Algorithm for Infinite Horizon Average Reward Markov Decision Processescited by 22
- 2021Nearly Minimax Optimal Regret for Learning Infinite-horizon Average-reward MDPs with Linear Function Approximationcited by 18
- 2021Improved Regret Bound and Experience Replay in Regularized Policy Iterationcited by 18
- 2021Concave Utility Reinforcement Learning with Zero-Constraint Violationscited by 18
12 most cited of 60 All 60 (JSON)
- 2026Finite Constant Frontiers and Auditable Regret Certificates for Average-Reward Reinforcement Learning7 Aug 2026
- 2026Bayesian learning for the stochastic shortest path problem3 Jun 2026
- 2026Learning Weakly Communicating Average-Reward CMDPs: Strong Duality and Improved Regret12 May 2026
- 2025No-Regret Thompson Sampling for Finite-Horizon Markov Decision Processes with Gaussian Processes23 Oct 2025
- 2025A Bit of Freedom Goes a Long Way: Classical and Quantum Algorithms for Reinforcement Learning under a Generative Model30 Jul 2025
- 2025Q-learning with Posterior Sampling1 Jun 2025
- 2025Minimax Optimal Reinforcement Learning with Quasi-Optimism2 Mar 2025
- 2025The Sample Complexity of Online Reinforcement Learning: A Multi-model Perspective27 Jan 2025
- 2024MaxInfoRL: Boosting exploration in reinforcement learning through information gain maximization16 Dec 2024
- 2024Generalized Bayesian deep reinforcement learning16 Dec 2024
- 2024Risk-Aware Decision Making in Restless Bandits: Theory and Algorithms for Planning and Learning30 Oct 2024
- 2024Provably Adaptive Average Reward Reinforcement Learning for Metric Spaces25 Oct 2024
12 newest of 60 All 60 (JSON)