Leveraging Demonstrations to Improve Online Learning: Quality Matters
Botao Hao, Rahul Jain, Tor Lattimore, Benjamin Van Roy, Zheng Wen
Abstract
We investigate the extent to which offline demonstration data can improve online learning. It is natural to expect some improvement, but the question is how, and by how much? We show that the degree of improvement must depend on the quality of the demonstration data. To generate portable insights, we focus on Thompson sampling (TS) applied to a multi-armed bandit as a prototypical online learning algorithm and model. The demonstration data is generated by an expert with a given competence level, a notion we introduce. We propose an informed TS algorithm that utilizes the demonstration data in a coherent way through Bayes' rule and derive a prior-dependent Bayesian regret bound. This offers insight into how pretraining can greatly improve online performance and how the degree of improvement increases with the expert's competence level. We also develop a practical, approximate informed TS algorithm through Bayesian bootstrapping and show substantial empirical regret reduction through experiments. * Equal contribution 1 Deepmind 2 University of Southern California, work done by Rahul Jain while at DeepMind.
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 32f107ac-97ca-43d1-ae60-e6b2da592425Cited by top-tier papers3
- Learning Across the Gap: Hybrid Multi-armed Bandits with Heterogeneous Offline and Online DataQijia He, Minghan Wang, Xutong Liu, Zhiyong Wang et al.NeurIPS 2025 · 6 citations
- Leveraging (Biased) Information: Multi-armed Bandits with Offline DataWang Chi Cheung, Lixing LyuICML 2024 · 3 citations
- Sequential Decision Making with Expert Demonstrations under Unobserved HeterogeneityVahid Balazadeh Meresht, Keertana Chidambaram, Viet Nguyen, Rahul G. Krishnan et al.NeurIPS 2024 · 3 citations
Builds on13
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 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
- Policy Finetuning: Bridging Sample-Efficient Offline and Online Reinforcement LearningTengyang Xie, Nan Jiang, Huan Wang, Caiming Xiong et al.NeurIPS 2021 · 207 citations
- Epistemic Neural NetworksIan Osband, Zheng Wen, Seyed Mohammad Asghari, Vikranth Dwaracherla et al.NeurIPS 2023 · 142 citations
- Meta-Thompson SamplingBranislav Kveton, Mikhail Konobeev, Manzil Zaheer, Chih-Wei Hsu et al.ICML 2021 · 74 citations
Related papers
- Thompson Sampling for Robust Transfer in Multi-Task BanditsZhi Wang, Chicheng Zhang, Kamalika ChaudhuriICML 2022 · 7 citations
- Supervised Pretraining Can Learn In-Context Reinforcement LearningJonathan Lee, Annie Xie, Aldo Pacchiano, Yash Chandak et al.NeurIPS 2023 · 170 citations
- Contextual Thompson Sampling via Generation of Missing DataKelly W. Zhang, Tiffany Tianhui Cai, Hongseok Namkoong, Daniel RussoNeurIPS 2025 · 5 citations
- No Regrets for Learning the Prior in BanditsSoumya Basu, Branislav Kveton, Manzil Zaheer, Csaba SzepesváriNeurIPS 2021 · 39 citations
- GuideBoot: Guided Bootstrap for Deep Contextual Banditsin Online AdvertisingFeiyang Pan, Haoming Li, Xiang Ao, Wei Wang et al.WWW 2021
