Beyond UCB: Optimal and Efficient Contextual Bandits with Regression Oracles
Dylan J. Foster, Alexander Rakhlin
Abstract
A fundamental challenge in contextual bandits is to develop flexible, general-purpose algorithms with computational requirements no worse than classical supervised learning tasks such as classification and regression. Algorithms based on regression have shown promising empirical success, but theoretical guarantees have remained elusive except in special cases. We provide the first universal and optimal reduction from contextual bandits to online regression. We show how to transform any oracle for online regression with a given value function class into an algorithm for contextual bandits with the induced policy class, with no overhead in runtime or memory requirements. We characterize the minimax rates for contextual bandits with general, potentially nonparametric function classes, and show that our algorithm is minimax optimal whenever the oracle obtains the optimal rate for regression. Compared to previous results, our algorithm requires no distributional assumptions beyond realizability, and works even when contexts are chosen adversarially.
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 35ff330a-ba90-480d-b702-29d8291d7e29Cited by top-tier papers114
- Neural Contextual Bandits with UCB-based ExplorationDongruo Zhou, Lihong Li, Quanquan GuICML 2020 · 329 citations
- Bilinear Classes: A Structural Framework for Provable Generalization in RLSimon S. Du, Sham M. Kakade, Jason D. Lee, Shachar Lovett et al.ICML 2021 · 207 citations
- Neural Thompson SamplingWeitong Zhang, Dongruo Zhou, Lihong Li, Quanquan GuICLR 2021 · 152 citations
- Representation Learning for Online and Offline RL in Low-rank MDPsMasatoshi Uehara, Xuezhou Zhang, Wen SunICLR 2022 · 138 citations
- Adapting to Misspecification in Contextual BanditsDylan J. Foster, Claudio Gentile, Mehryar Mohri, Julian ZimmertNeurIPS 2020 · 111 citations
Builds on1
Related papers
- Efficient First-Order Contextual Bandits: Prediction, Allocation, and Triangular DiscriminationDylan J. Foster, Akshay KrishnamurthyNeurIPS 2021 · 62 citations
- Adapting to misspecification in contextual bandits with offline regression oraclesSanath Kumar Krishnamurthy, Vitor Hadad, Susan AtheyICML 2021 · 27 citations
- Contextual Multinomial Logit Bandits with General Value FunctionsMengxiao Zhang, Haipeng LuoNeurIPS 2024 · 5 citations
- An Improved Relaxation for Oracle-Efficient Adversarial Contextual BanditsKiarash Banihashem, MohammadTaghi Hajiaghayi, Suho Shin, Max SpringerNeurIPS 2023 · 3 citations
- Efficient Contextual Bandits with Continuous ActionsMaryam Majzoubi, Chicheng Zhang, Rajan Chari, Akshay Krishnamurthy et al.NeurIPS 2020 · 39 citations
