Nearly Horizon-Free Offline Reinforcement Learning
Tongzheng Ren, Jialian Li, Bo Dai, Simon S. Du, Sujay Sanghavi
Abstract
We revisit offline reinforcement learning on episodic time-homogeneous Markov Decision Processes (MDP). For tabular MDP with S states and A actions, or linear MDP with anchor points and feature dimension d, given the collected K episodes data with minimum visiting probability of (anchor) stateaction pairs d m , we obtain nearly horizon H-free sample complexity bounds for offline reinforcement learning when the total reward is upper bounded by 1. Specifically: • For offline policy evaluation, we obtain an Õ 1 Kdm error bound for the plug-in estimator, which matches the lower bound up to logarithmic factors and does not have additional dependency on poly (H, S, A, d) in higher-order term. • For offline policy optimization, we obtain an Õ 1 Kdm + min(S,d) Kdm sub-optimality gap for the empirical optimal policy, which approaches the lower bound up to logarithmic factors and a high-order term, improving upon the best known result by Cui and Yang [2020] that has additional poly (H, S, d) factors in the main term. To the best of our knowledge, these are the first set of nearly horizon-free bounds for episodic timehomogeneous offline tabular MDP and linear MDP with anchor points. Central to our analysis is a simple yet effective recursion based method to bound a "total variance" term in the offline scenarios, which could be of individual interest. Offline Policy Evaluation. For OPE in infinite horizon tabular MDP, [Li et al., 2020] showed that plugin estimator can achieve the error of O 1 dm(1-γ) under Assumption 1, which matches the lower bound in [Pananjady and Wainwright, 2020] up to logarithmic factors. For OPE in finite horizon time-inhomogeneous tabular MDP, [Yin and Wang, 2020, Yin et al., 2020] provided an error bound of O 1 Kdm + √ SA Kdm under the uniform reward assumption, which matches the lower bound [Jiang and Li, 2016] up to logarithmic fac-
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 9ab39e3e-210b-4c2e-be16-ba54931244f4Cited by top-tier papers28
- Policy Finetuning: Bridging Sample-Efficient Offline and Online Reinforcement LearningTengyang Xie, Nan Jiang, Huan Wang, Caiming Xiong et al.NeurIPS 2021 · 207 citations
- Towards Instance-Optimal Offline Reinforcement Learning with PessimismMing Yin, Yu-Xiang WangNeurIPS 2021 · 93 citations
- Should I Run Offline Reinforcement Learning or Behavioral Cloning?Aviral Kumar, Joey Hong, Anikait Singh, Sergey LevineICLR 2022 · 84 citations
- Hierarchical Diffusion for Offline Decision MakingWenhao Li, Xiangfeng Wang, Bo Jin, Hongyuan ZhaICML 2023 · 80 citations
- Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with PessimismMing Yin, Yaqi Duan, Mengdi Wang, Yu-Xiang WangICLR 2022 · 74 citations
Builds on8
- Conservative Q-Learning for Offline Reinforcement LearningAviral Kumar, Aurick Zhou, George Tucker, Sergey LevineNeurIPS 2020 · 2,881 citations
- Minimax Weight and Q-Function Learning for Off-Policy EvaluationMasatoshi Uehara, Jiawei Huang, Nan JiangICML 2020 · 199 citations
- Minimax-Optimal Off-Policy Evaluation with Linear Function ApproximationYaqi Duan, Zeyu Jia, Mengdi WangICML 2020 · 161 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
- Batch Value-function Approximation with Only RealizabilityTengyang Xie, Nan JiangICML 2021 · 131 citations
Related papers
- Near-Optimal Offline Reinforcement Learning via Double Variance ReductionMing Yin, Yu Bai, Yu-Xiang WangNeurIPS 2021 · 72 citations
- Revisiting the Linear-Programming Framework for Offline RL with General Function ApproximationAsuman E. Ozdaglar, Sarath Pattathil, Jiawei Zhang, Kaiqing ZhangICML 2023 · 8 citations
- On the Sample Complexity of Vanilla Model-Based Offline Reinforcement Learning with Dependent SamplesMustafa O. Karabag, Ufuk TopcuAAAI 2023 · 6 citations
- Optimal Uniform OPE and Model-based Offline Reinforcement Learning in Time-Homogeneous, Reward-Free and Task-Agnostic SettingsMing Yin, Yu-Xiang WangNeurIPS 2021 · 19 citations
- Model-based Reinforcement Learning for Confounded POMDPsMao Hong, Zhengling Qi, Yanxun XuICML 2024 · 5 citations
