Leveraging Offline Data in Linear Latent Contextual Bandits
Chinmaya Kausik, Kevin Tan, Ambuj Tewari
Abstract
Leveraging offline data is an attractive way to accelerate online sequential decision-making. However, it is crucial to account for latent states in users or environments in the offline data, and latent bandits form a compelling model for doing so. In this light, we design end-to-end latent bandit algorithms capable of handing uncountably many latent states. We focus on a linear latent contextual bandit -a linear bandit where each user has its own high-dimensional reward parameter in R d A , but reward parameters across users lie in a low-rank latent subspace of dimension d K ≪ d A . First, we provide an offline algorithm to learn this subspace with provable guarantees. We then present two online algorithms that utilize the output of this offline algorithm to accelerate online learning. The first enjoys ) regret guarantees, so that the effective dimension is lower when the size N of the offline dataset is larger. We prove a matching lower bound on regret, showing that our algorithm is minimax optimal up to coverage terms. The second is a practical algorithm that enjoys only a slightly weaker guarantee, but is computationally efficient. We also establish the efficacy of our methods using experiments on both synthetic data and real-life movie recommendation data from MovieLens. Finally, we theoretically establish the generality of the latent bandit model by proving a de Finetti theorem for stateless decision processes.
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 4edde24b-abbb-463d-8567-60156a7e108dBuilds on9
- Is Pessimism Provably Efficient for Offline RL?Ying Jin, Zhuoran Yang, Zhaoran WangICML 2021 · 419 citations
- Provable Meta-Learning of Linear RepresentationsNilesh Tripuraneni, Chi Jin, Michael I. JordanICML 2021 · 218 citations
- Minimax-Optimal Off-Policy Evaluation with Linear Function ApproximationYaqi Duan, Zeyu Jia, Mengdi WangICML 2020 · 161 citations
- Meta-learning for Mixed Linear RegressionWeihao Kong, Raghav Somani, Zhao Song, Sham M. Kakade et al.ICML 2020 · 70 citations
- Leveraging Offline Data in Online Reinforcement LearningAndrew Wagenmaker, Aldo PacchianoICML 2023 · 47 citations
Related papers
- Latent Bandits RevisitedJoey Hong, Branislav Kveton, Manzil Zaheer, Yinlam Chow et al.NeurIPS 2020 · 55 citations
- Beyond UCB: Optimal and Efficient Contextual Bandits with Regression OraclesDylan J. Foster, Alexander RakhlinICML 2020 · 241 citations
- Impact of Representation Learning in Linear BanditsJiaqi Yang, Wei Hu, Jason D. Lee, Simon Shaolei DuICLR 2021 · 58 citations
- Fast and Sample Efficient Multi-Task Representation Learning in Stochastic Contextual BanditsJiabin Lin, Shana Moothedath, Namrata VaswaniICML 2024 · 9 citations
- Symmetric Linear Bandits with Hidden SymmetryNam Phuong Tran, The-Anh Ta, Debmalya Mandal, Long Tran-ThanhNeurIPS 2024 · 1 citation
