Beyond UCB: Optimal and Efficient Contextual Bandits with Regression Oracles
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 195 papers in the map (data through 7 Oct 2026). First posted on arXiv 12 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 32; 2022: 36; 2023: 43; 2024: 31; 2025: 25; 2026: 28.
195 papers.
Sort:
By area:
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 128
- 2021Non-stationary Reinforcement Learning without Prior Knowledge: An Optimal Black-box Approachcited by 98
- 2021Adapting to Misspecification in Contextual Banditscited by 81
- 2021Feel-Good Thompson Sampling for Contextual Bandits and Reinforcement Learningcited by 65
- 2021Efficient First-Order Contextual Bandits: Prediction, Allocation, and Triangular Discriminationcited by 53
- 2021Misspecified Gaussian Process Bandit Optimizationcited by 52
- 2022Smoothed Online Learning is as Easy as Statistical Learningcited by 50
- 2021Efficient and Optimal Algorithms for Contextual Dueling Bandits under Realizabilitycited by 44
- 2021Heuristic-Guided Reinforcement Learningcited by 41
- 2021An Exponential Lower Bound for Linearly-Realizable MDPs with Constant Suboptimality Gapcited by 40
- 2023Oracle Efficient Online Multicalibration and Omnipredictioncited by 36
38 more
- 2022Corruption-Robust Algorithms with Uncertainty Weighting for Nonlinear Contextual Bandits and Markov Decision Processescited by 33
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2022Contextual Bandits with Large Action Spaces: Made Practicalcited by 32
- 2023Multicalibration as Boosting for Regressioncited by 32
- 2022Contextual Bandits with Packing and Covering Constraints: A Modular Lagrangian Approach via Regressioncited by 30
- 2023The Benefits of Being Distributional: Small-Loss Bounds for Reinforcement Learningcited by 27
- 2022Redeeming Intrinsic Rewards via Constrained Optimizationcited by 24
- 2023Improved Regret Bounds of (Multinomial) Logistic Bandits via Regret-to-Confidence-Set Conversioncited by 24
- 2023Optimal Learners for Realizable Regression: PAC Learning and Online Learningcited by 23
- 2024More Benefits of Being Distributional: Second-Order Bounds for Reinforcement Learningcited by 23
- 2021Distribution-free Contextual Dynamic Pricingcited by 22
- 2021Adapting to Misspecification in Contextual Bandits with Offline Regression Oraclescited by 21
- 2022Efficient Active Learning with Abstentioncited by 21
- 2021Information Directed Sampling for Sparse Linear Banditscited by 20
- 2022Multi-task Representation Learning with Stochastic Linear Banditscited by 20
- 2022Oracle-Efficient Online Learning for Beyond Worst-Case Adversariescited by 19
- 2022Lifting the Information Ratio: An Information-Theoretic Analysis of Thompson Sampling for Contextual Banditscited by 19
- 2022Contexts can be Cheap: Solving Stochastic Contextual Bandits with Linear Bandit Algorithmscited by 19
- 2022Optimistic Whittle Index Policy: Online Learning for Restless Banditscited by 18
- 2023Bypassing the Simulator: Near-Optimal Adversarial Linear Contextual Banditscited by 18
- 2021Majorizing Measures, Sequential Complexities, and Online Learningcited by 17
- 2022Contextual Bandits with Smooth Regret: Efficient Learning in Continuous Action Spacescited by 17
- 2023Bayesian Design Principles for Frequentist Sequential Learningcited by 17
- 2021Fast Rates for Nonparametric Online Learning: From Realizability to Learning in Gamescited by 16
- 2022Contextual Information-Directed Samplingcited by 16
- 2022Instance-optimal PAC Algorithms for Contextual Banditscited by 16
- 2024Efficient Sequential Decision Making with Large Language Modelscited by 16
- 2023Adapting to Continuous Covariate Shift via Online Density Ratio Estimationcited by 15
- 2021On component interactions in two-stage recommender systemscited by 14
- 2024On the Performance of Empirical Risk Minimization with Smoothed Datacited by 13
- 2022Optimal Contextual Bandits with Knapsacks under Realizability via Regression Oraclescited by 12
- 2021Model Selection in Batch Policy Optimizationcited by 11
- 2022Optimism in Face of a Context: Regret Guarantees for Stochastic Contextual MDPcited by 11
- 2024Assouad, Fano, and Le Cam with Interaction: A Unifying Lower Bound Framework and Characterization for Bandit Learnabilitycited by 11
- 2021Risk Minimization from Adaptively Collected Data: Guarantees for Supervised and Policy Learningcited by 10
- 2022Efficient and Near-Optimal Smoothed Online Learning for Generalized Linear Functionscited by 10
- 2023Practical Contextual Bandits with Feedback Graphscited by 10
- 2023Tracking Most Significant Shifts in Nonparametric Contextual Banditscited by 10
50 most cited of 195 All 195 (JSON)
- 2026Nearly Minimax-Optimal Regret for Linear Contextual Bandits with Arbitrary Adaptive Action Sets14 Sep 2026
- 2026When Greedy Sampling Explores: KL-Regularized Contextual Bandits without Eluder-Dimension Dependence11 Sep 2026
- 2026CAROL: Context-Aware Online Learning for Fuzzer Scheduling6 Sep 2026
- 2026Learning to Price and Stock Under Contextual and Censored Demand5 Sep 2026
- 2026Optimal and Efficient Contextual Combinatorial Semi-bandits with General Function Approximation15 Jul 2026
- 2026Correlation-Aware Contextual Bandits with Surrogate Rewards for LLM Routing10 Jul 2026
- 2026Balancing Expressivity and Learnability in Quantum Kernel Bandit Optimization1 Jul 2026
- 2026When Does Online Imitation Learning Help in LLM Post-Training? The Role of (Non-)Realizability Beyond Horizon29 Jun 2026
- 2026Stochastic Linear Contextual Bandits with Bounded Noise: A Set-Membership Approach18 Jun 2026
- 2026Policy Regret for Embedding Model Routing: Contextual Bandits with Low-Rank Experts12 Jun 2026
- 2026Online KL-Regularized Reinforcement Learning with Function Approximation under Misspecification4 Jun 2026
- 2026The Sample Complexity of Multiclass and Sparse Contextual Bandits28 May 2026
38 more
- 2026Optimal Regret for Single Index Bandits10 May 2026
- 2026Near-Optimal Last-Iterate Convergence for Zero-Sum Games with Bandit Feedback and Opponent Actions10 May 2026
- 2026Regret-Oracle Complexity Tradeoffs in Agnostic Online Learning8 May 2026
- 2026Constrained Contextual Bandits with Adversarial Contexts7 May 2026
- 2026Replicable Bandits with UCB based Exploration21 Apr 2026
- 2026A Practical Algorithm for Feature-Rich, Non-Stationary Bandit Problems17 Mar 2026
- 2026RIE-Greedy: Regularization-Induced Exploration for Contextual Bandits11 Mar 2026
- 2026Provably Efficient Regularized Online RLHF with Generalized Bilinear Preferences26 Feb 2026
- 2026Interactive Learning of Single-Index Models via Stochastic Gradient Descent19 Feb 2026
- 2026Optimal Regret for Policy Optimization in Contextual Bandits14 Feb 2026
- 2026Taming the Monster Every Context: Complexity Measure and Unified Framework for Offline-Oracle Efficient Contextual Bandits10 Feb 2026
- 2026Interaction-Grounded Learning for Contextual Markov Decision Processes with Personalized Feedback9 Feb 2026
- 2026A Simple Reduction Scheme for Constrained Contextual Bandits with Adversarial Contexts via Regression4 Feb 2026
- 2026Learning-to-Defer in Non-Stationary Time Series via Switching State-Space Models30 Jan 2026
- 2026Group-realizable multi-group learning by minimizing empirical risk23 Jan 2026
- 2026Tractable Multinomial Logit Contextual Bandits with Non-Linear Utilities11 Jan 2026
- 2025Sparse Optimistic Information Directed Sampling28 Oct 2025
- 2025Exploration via Feature Perturbation in Contextual Bandits20 Oct 2025
- 2025Fast Best-in-Class Regret for Contextual Bandits17 Oct 2025
- 2025Regret Bounds for Adversarial Contextual Bandits with General Function Approximation and Delayed Feedback10 Oct 2025
- 2025DyBBT: Dynamic Balance via Bandit-inspired Targeting for Dialog Policy with Cognitive Dual-Systems24 Sep 2025
- 2025One-shot acceleration of transient PDE solvers via online-learned preconditioners10 Sep 2025
- 2025Beyond ATE: Multi-Criteria Design for A/B Testing6 Sep 2025
- 2025Pre-Trained AI Model Assisted Online Decision-Making under Missing Covariates: A Theoretical Perspective10 Jul 2025
- 2025Episodic Contextual Bandits with Knapsacks under Conversion Models9 Jul 2025
- 2025Aligning Learning and Endogenous Decision-Making1 Jul 2025
- 2025Single Index Bandits: Generalized Linear Contextual Bandits with Unknown Reward Functions15 Jun 2025
- 2025Strategic Scaling of Test-Time Compute: A Bandit Learning Approach15 Jun 2025
- 2025Active Learning via Regression Beyond Realizability31 May 2025
- 2025Tradeoffs between Mistakes and ERM Oracle Calls in Online and Transductive Online Learning30 May 2025
- 2025Thompson Sampling in Online RLHF with General Function Approximation29 May 2025
- 2025Improved and Oracle-Efficient Online ℓ1-Multicalibration23 May 2025
- 2025Scalable and Interpretable Contextual Bandits: A Literature Review and Retail Offer Prototype22 May 2025
- 2025Constrained Online Decision-Making: A Unified Framework11 May 2025
- 2025Sparse Nonparametric Contextual Bandits20 Mar 2025
- 2025Greedy Algorithm for Structured Bandits: A Sharp Characterization of Asymptotic Success / Failure6 Mar 2025
- 2025Quantum Non-Linear Bandit Optimization4 Mar 2025
- 2025Parameter-Adaptive Dynamic Pricing2 Mar 2025
50 newest of 195 All 195 (JSON)
- 2026Provably Efficient Regularized Online RLHF with Generalized Bilinear Preferencescited by 5
- 2026Near-Optimal Last-Iterate Convergence for Zero-Sum Games with Bandit Feedback and Opponent Actionscited by 2
- 2022Flexible and Efficient Contextual Bandits with Heterogeneous Treatment Effect Oraclescited by 1
- 2025Thompson Sampling in Online RLHF with General Function Approximationcited by 0
- 2026RIE-Greedy: Regularization-Induced Exploration for Contextual Banditscited by 0
- 2026Near-Optimal Last-Iterate Convergence for Zero-Sum Games with Bandit Feedback and Opponent Actions10 May 2026
- 2026RIE-Greedy: Regularization-Induced Exploration for Contextual Bandits11 Mar 2026
- 2026Provably Efficient Regularized Online RLHF with Generalized Bilinear Preferences26 Feb 2026
- 2025Thompson Sampling in Online RLHF with General Function Approximation29 May 2025
- 2022Flexible and Efficient Contextual Bandits with Heterogeneous Treatment Effect Oracles30 Mar 2022
- 2021Representation Learning for Online and Offline RL in Low-rank MDPscited by 128
- 2024More Benefits of Being Distributional: Second-Order Bounds for Reinforcement Learningcited by 23
- 2026When Does Online Imitation Learning Help in LLM Post-Training? The Role of (Non-)Realizability Beyond Horizoncited by 2
- 2026When Does Online Imitation Learning Help in LLM Post-Training? The Role of (Non-)Realizability Beyond Horizon29 Jun 2026
- 2024More Benefits of Being Distributional: Second-Order Bounds for Reinforcement Learning11 Feb 2024
- 2021Representation Learning for Online and Offline RL in Low-rank MDPs9 Oct 2021
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2021Non-stationary Reinforcement Learning without Prior Knowledge: An Optimal Black-box Approachcited by 98
- 2021Adapting to Misspecification in Contextual Banditscited by 81
- 2021Feel-Good Thompson Sampling for Contextual Bandits and Reinforcement Learningcited by 65
- 2021Efficient First-Order Contextual Bandits: Prediction, Allocation, and Triangular Discriminationcited by 53
- 2021Misspecified Gaussian Process Bandit Optimizationcited by 52
- 2022Smoothed Online Learning is as Easy as Statistical Learningcited by 50
- 2021Efficient and Optimal Algorithms for Contextual Dueling Bandits under Realizabilitycited by 44
- 2021Heuristic-Guided Reinforcement Learningcited by 41
- 2021An Exponential Lower Bound for Linearly-Realizable MDPs with Constant Suboptimality Gapcited by 40
- 2023Oracle Efficient Online Multicalibration and Omnipredictioncited by 36
- 2022Corruption-Robust Algorithms with Uncertainty Weighting for Nonlinear Contextual Bandits and Markov Decision Processescited by 33
12 most cited of 186 All 186 (JSON)
- 2026Nearly Minimax-Optimal Regret for Linear Contextual Bandits with Arbitrary Adaptive Action Sets14 Sep 2026
- 2026When Greedy Sampling Explores: KL-Regularized Contextual Bandits without Eluder-Dimension Dependence11 Sep 2026
- 2026CAROL: Context-Aware Online Learning for Fuzzer Scheduling6 Sep 2026
- 2026Learning to Price and Stock Under Contextual and Censored Demand5 Sep 2026
- 2026Optimal and Efficient Contextual Combinatorial Semi-bandits with General Function Approximation15 Jul 2026
- 2026Correlation-Aware Contextual Bandits with Surrogate Rewards for LLM Routing10 Jul 2026
- 2026Balancing Expressivity and Learnability in Quantum Kernel Bandit Optimization1 Jul 2026
- 2026Stochastic Linear Contextual Bandits with Bounded Noise: A Set-Membership Approach18 Jun 2026
- 2026Policy Regret for Embedding Model Routing: Contextual Bandits with Low-Rank Experts12 Jun 2026
- 2026Online KL-Regularized Reinforcement Learning with Function Approximation under Misspecification4 Jun 2026
- 2026The Sample Complexity of Multiclass and Sparse Contextual Bandits28 May 2026
- 2026Optimal Regret for Single Index Bandits10 May 2026
12 newest of 186 All 186 (JSON)