Settling the Horizon-Dependence of Sample Complexity in Reinforcement Learning
Yuanzhi Li, Ruosong Wang, Lin F. Yang
Abstract
Recently there is a surge of interest in under-standing the horizon-dependence of the sample complexity in reinforcement learning (RL). Notably, for an RL environment with horizon length H, previous work have shown that there is a probably approximately correct (PAC) algorithm that learns an O(1)-optimal policy using polylog(H) episodes of environment interactions when the number of states and actions is fixed. It is yet unknown whether the polylog (H) dependence is necessary or not. In this work, we resolve this question by developing an algorithm that achieves the same PAC guarantee while using only O(1) episodes of environment interactions, completely settling the horizon-dependence of the sample complexity in RL. We achieve this bound by (i) establishing a connection between value functions in discounted and finite-horizon Markov decision processes (MDPs) and (ii) a novel perturbation analysis in MDPs. We believe our new techniques are of independent interest and could be applied in related questions in RL.
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 84a20035-24cb-4a97-a585-e8edffe00564Cited by top-tier papers9
- Computationally Efficient Horizon-Free Reinforcement Learning for Linear Mixture MDPsDongruo Zhou, Quanquan GuNeurIPS 2022 · 60 citations
- Provably Feedback-Efficient Reinforcement Learning via Active Reward LearningDingwen Kong, Lin YangNeurIPS 2022 · 19 citations
- Zero-sum Polymatrix Markov Games: Equilibrium Collapse and Efficient Computation of Nash EquilibriaFivos Kalogiannis, Ioannis PanageasNeurIPS 2023 · 10 citations
- On the Power of Pre-training for Generalization in RL: Provable Benefits and HardnessHaotian Ye, Xiaoyu Chen, Liwei Wang, Simon Shaolei DuICML 2023 · 8 citations
- Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPsJunkai Zhang, Weitong Zhang, Quanquan GuICML 2023 · 6 citations
Builds on6
- Almost Optimal Model-Free Reinforcement Learningvia Reference-Advantage DecompositionZihan Zhang, Yuan Zhou, Xiangyang JiNeurIPS 2020 · 183 citations
- Breaking the Sample Size Barrier in Model-Based Reinforcement Learning with a Generative ModelGen Li, Yuting Wei, Yuejie Chi, Yuantao Gu et al.NeurIPS 2020 · 159 citations
- Q-learning with UCB Exploration is Sample Efficient for Infinite-Horizon MDPYuanhao Wang, Kefan Dong, Xiaoyu Chen, Liwei WangICLR 2020 · 107 citations
- A Unifying View of Optimism in Episodic Reinforcement LearningGergely Neu, Ciara Pike-BurkeNeurIPS 2020 · 79 citations
- Nearly Horizon-Free Offline Reinforcement LearningTongzheng Ren, Jialian Li, Bo Dai, Simon S. Du et al.NeurIPS 2021 · 54 citations
Related papers
- Horizon-free Learning for Markov Decision Processes and Games: Stochastically Bounded Rewards and Improved BoundsShengshi Li, Lin YangICML 2023 · 3 citations
- Near Instance-Optimal PAC Reinforcement Learning for Deterministic MDPsAndrea Tirinzoni, Aymen Al Marjani, Emilie KaufmannNeurIPS 2022 · 20 citations
- On the Sample Complexity of Learning Infinite-horizon Discounted Linear Kernel MDPsYuanzhou Chen, Jiafan He, Quanquan GuICML 2022 · 8 citations
- Near-Optimal Regret Bounds for Multi-batch Reinforcement LearningZihan Zhang, Yuhang Jiang, Yuan Zhou, Xiangyang JiNeurIPS 2022 · 16 citations
- Span-Based Optimal Sample Complexity for Weakly Communicating and General Average Reward MDPsMatthew Zurek, Yudong ChenNeurIPS 2024 · 20 citations
