Model-based RL with Optimistic Posterior Sampling: Structural Conditions and Sample Complexity
Alekh Agarwal, Tong Zhang
Abstract
We propose a general framework to design posterior sampling methods for model-based RL. We show that the proposed algorithms can be analyzed by reducing regret to Hellinger distance in conditional probability estimation. We further show that optimistic posterior sampling can control this Hellinger distance, when we measure model error via data likelihood. This technique allows us to design and analyze unified posterior sampling algorithms with state-of-the-art sample complexity guarantees for many model-based RL settings. We illustrate our general result in many special cases, demonstrating the versatility of our framework.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 24a83a7e-cbd0-4e16-a2bd-79cec59216e1Cited by top-tier papers23
- Making RL with Preference-based Feedback Efficient via RandomizationRunzhe Wu, Wen SunICLR 2024 · 44 citations
- Maximize to Explore: One Objective Function Fusing Estimation, Planning, and ExplorationZhihan Liu, Miao Lu, Wei Xiong, Han Zhong et al.NeurIPS 2023 · 30 citations
- Learning in POMDPs is Sample-Efficient with Hindsight ObservabilityJonathan Lee, Alekh Agarwal, Christoph Dann, Tong ZhangICML 2023 · 25 citations
- Feel-Good Thompson Sampling for Contextual Dueling BanditsXuheng Li, Heyang Zhao, Quanquan GuICML 2024 · 19 citations
- Bayesian Design Principles for Frequentist Sequential LearningYunbei Xu, Assaf ZeeviICML 2023 · 19 citations
Builds on11
- Planning to Explore via Self-Supervised World ModelsRamanan Sekar, Oleh Rybkin, Kostas Daniilidis, Pieter Abbeel et al.ICML 2020 · 489 citations
- Model-Based Reinforcement Learning with Value-Targeted RegressionAlex Ayoub, Zeyu Jia, Csaba Szepesvári, Mengdi Wang et al.ICML 2020 · 324 citations
- Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret BoundLin Yang, Mengdi WangICML 2020 · 308 citations
- FLAMBE: Structural Complexity and Representation Learning of Low Rank MDPsAlekh Agarwal, Sham M. Kakade, Akshay Krishnamurthy, Wen SunNeurIPS 2020 · 271 citations
- Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient AlgorithmsChi Jin, Qinghua Liu, Sobhan MiryoosefiNeurIPS 2021 · 264 citations
Related papers
- Model-based Reinforcement Learning for Continuous Control with Posterior SamplingYing Fan, Yifei MingICML 2021 · 25 citations
- A Provably Efficient Model-Free Posterior Sampling Method for Episodic Reinforcement LearningChristoph Dann, Mehryar Mohri, Tong Zhang, Julian ZimmertNeurIPS 2021 · 43 citations
- Q-learning with Posterior SamplingPriyank Agrawal, Shipra Agrawal, Azmat AzatiICLR 2026 · 3 citations
- Conservative Dual Policy Optimization for Efficient Model-Based Reinforcement LearningShenao ZhangNeurIPS 2022 · 8 citations
- Posterior Sampling Reinforcement Learning with Gaussian Processes for Continuous Control: Sublinear Regret Bounds for Unbounded State SpacesHamish Flynn, Joe Watson, Ingmar Posner, Jan PetersICML 2026
