Contextual Bandits with Large Action Spaces: Made Practical
Yinglun Zhu, Dylan J. Foster, John Langford, Paul Mineiro
Abstract
A central problem in sequential decision making is to develop algorithms that are practical and computationally efficient, yet support the use of flexible, general-purpose models. Focusing on the contextual bandit problem, recent progress provides provably efficient algorithms with strong empirical performance when the number of possible alternatives ("actions") is small, but guarantees for decision making in large, continuous action spaces have remained elusive, leading to a significant gap between theory and practice. We present the first efficient, general-purpose algorithm for contextual bandits with continuous, linearly structured action spaces. Our algorithm makes use of computational oracles for (i) supervised learning, and (ii) optimization over the action space, and achieves sample complexity, runtime, and memory independent of the size of the action space. In addition, it is simple and practical. We perform a large-scale empirical evaluation, and show that our approach typically enjoys superior performance and efficiency compared to standard baselines.
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 a23f50d5-1f50-409e-84a1-f89c5e2dd2beCited by top-tier papers16
- Bypassing the Simulator: Near-Optimal Adversarial Linear Contextual BanditsHaolin Liu, Chen-Yu Wei, Julian ZimmertNeurIPS 2023 · 20 citations
- Bayesian Design Principles for Frequentist Sequential LearningYunbei Xu, Assaf ZeeviICML 2023 · 19 citations
- Exponential Smoothing for Off-Policy LearningImad Aouali, Victor-Emmanuel Brunel, David Rohde, Anna KorbaICML 2023 · 17 citations
- Efficient Batched Algorithm for Contextual Linear Bandits with Large Action Space via Soft EliminationOsama A. Hanna, Lin Yang, Christina FragouliNeurIPS 2023 · 12 citations
- Integrating Reward Maximization and Population Estimation: Sequential Decision-Making for Internal Revenue Service Audit SelectionPeter Henderson, Ben Chugg, Brandon R. Anderson, Kristen M. Altenburger et al.AAAI 2023 · 10 citations
Builds on12
- Neural Contextual Bandits with UCB-based ExplorationDongruo Zhou, Lihong Li, Quanquan GuICML 2020 · 329 citations
- Beyond UCB: Optimal and Efficient Contextual Bandits with Regression OraclesDylan J. Foster, Alexander RakhlinICML 2020 · 241 citations
- Learning with Good Feature Representations in Bandits and in RL with a Generative ModelTor Lattimore, Csaba Szepesvári, Gellért WeiszICML 2020 · 181 citations
- Neural Thompson SamplingWeitong Zhang, Dongruo Zhou, Lihong Li, Quanquan GuICLR 2021 · 152 citations
- Adapting to Misspecification in Contextual BanditsDylan J. Foster, Claudio Gentile, Mehryar Mohri, Julian ZimmertNeurIPS 2020 · 111 citations
Related papers
- Efficient Contextual Bandits with Continuous ActionsMaryam Majzoubi, Chicheng Zhang, Rajan Chari, Akshay Krishnamurthy et al.NeurIPS 2020 · 39 citations
- Contextual Bandits with Smooth Regret: Efficient Learning in Continuous Action SpacesYinglun Zhu, Paul MineiroICML 2022 · 19 citations
- Efficient and Near-Optimal Algorithm for Contextual Dueling Bandits with Offline Regression OraclesAadirupa Saha, Robert E. SchapireNeurIPS 2025 · 3 citations
- An Improved Algorithm for Adversarial Linear Contextual Bandits via ReductionTim van Erven, Jack J. Mayo, Julia Olkhovskaya, Chen-Yu WeiNeurIPS 2025 · 4 citations
- Contextual Online Decision Making with Infinite-Dimensional Functional RegressionHaichen Hu, Rui Ai, Stephen Bates, David Simchi-LeviICML 2025
