Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learning
Gen Li, Wenhao Zhan, Jason D. Lee, Yuejie Chi, Yuxin Chen
Abstract
This paper studies tabular reinforcement learning (RL) in the hybrid setting, which assumes access to both an offline dataset and online interactions with the unknown environment. A central question boils down to how to efficiently utilize online data collection to strengthen and complement the offline dataset and enable effective policy fine-tuning. Leveraging recent advances in reward-agnostic exploration and model-based offline RL, we design a three-stage hybrid RL algorithm that beats the best of both worlds -- pure offline RL and pure online RL -- in terms of sample complexities. The proposed algorithm does not require any reward information during data collection. Our theory is developed based on a new notion called single-policy partial concentrability, which captures the trade-off between distribution mismatch and miscoverage and guides the interplay between offline and online data.
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 3be69e5c-c5fa-4fae-b201-84ba4462e075Cited by top-tier papers8
- Harnessing Density Ratios for Online Reinforcement LearningPhilip Amortila, Dylan J. Foster, Nan Jiang, Ayush Sekhari et al.ICLR 2024 · 14 citations
- Scalable Online Exploration via CoverabilityPhilip Amortila, Dylan J. Foster, Akshay KrishnamurthyICML 2024 · 10 citations
- Hybrid Reinforcement Learning Breaks Sample Size Barriers In Linear MDPsKevin Tan, Wei Fan, Yuting WeiNeurIPS 2024 · 6 citations
- Hybrid Reinforcement Learning from Offline Observation AloneYuda Song, Drew Bagnell, Aarti SinghICML 2024 · 6 citations
- On The Statistical Complexity of Offline Decision-MakingThanh Nguyen-Tang, Raman AroraICML 2024 · 2 citations
Builds on29
- Conservative Q-Learning for Offline Reinforcement LearningAviral Kumar, Aurick Zhou, George Tucker, Sergey LevineNeurIPS 2020 · 2,881 citations
- Is Pessimism Provably Efficient for Offline RL?Ying Jin, Zhuoran Yang, Zhaoran WangICML 2021 · 419 citations
- Bridging Offline Reinforcement Learning and Imitation Learning: A Tale of PessimismParia Rashidinejad, Banghua Zhu, Cong Ma, Jiantao Jiao et al.NeurIPS 2021 · 373 citations
- Bellman-consistent Pessimism for Offline Reinforcement LearningTengyang Xie, Ching-An Cheng, Nan Jiang, Paul Mineiro et al.NeurIPS 2021 · 339 citations
- Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-TuningMitsuhiko Nakamoto, Simon Zhai, Anikait Singh, Max Sobol Mark et al.NeurIPS 2023 · 296 citations
Related papers
- Policy Finetuning: Bridging Sample-Efficient Offline and Online Reinforcement LearningTengyang Xie, Nan Jiang, Huan Wang, Caiming Xiong et al.NeurIPS 2021 · 207 citations
- Hybrid RL: Using both offline and online data can make RL efficientYuda Song, Yifei Zhou, Ayush Sekhari, Drew Bagnell et al.ICLR 2023 · 7 citations
- Offline Meta-Reinforcement Learning with Online Self-SupervisionVitchyr H. Pong, Ashvin Nair, Laura Smith, Catherine Huang et al.ICML 2022 · 78 citations
- Offline Data Enhanced On-Policy Policy Gradient with Provable GuaranteesYifei Zhou, Ayush Sekhari, Yuda Song, Wen SunICLR 2024 · 11 citations
- Policy Finetuning in Reinforcement Learning via Design of Experiments using Offline DataRuiqi Zhang, Andrea ZanetteNeurIPS 2023 · 12 citations
