Bypassing the Monster: A Faster and Simpler Optimal Algorithm for Contextual Bandits Under Realizability: FALCON+ Has Regret ≤ 14.3·Σₘ√(K𝓔(τₘ₋₁−τₘ₋₂))(τₘ−τₘ₋₁) + τ₁ + √(8T log(2/δ)) w.p. 1−δResearch Paper
Contextual bandits and the reduction to regression
A contextual bandit is a sequential decision problem: in each round a learner sees a context (a user's features, a patient's covariates), picks one of actions, and observes the reward of that action only. Applications include recommendation, dynamic pricing and adaptive clinical decisions. The difficulty is statistical and computational at once: the learner must explore to learn which action is best for which context, and the class of models it learns over is typically huge.
A long-standing question was whether contextual bandits can be solved optimally with nothing more than an offline regression oracle, a black-box supervised learning routine that fits a predictor of rewards to i.i.d. data. Earlier optimal algorithms either required a cost-sensitive classification oracle called many times per round (the "monster" of Agarwal et al., 2014) or an online regression oracle that must handle adversarially generated data (Foster and Rakhlin, 2020). Simchi-Levi and Xu (2022) answered the question with FALCON+, which calls an arbitrary offline regression oracle once per epoch, or times in total, and whose regret is governed by the oracle's estimation error.
Timeline:
- 2014: Agarwal, Hsu, Kale, Langford, Li and Schapire, Taming the Monster: optimal regret with calls to a classification oracle.
- 2020: Foster and Rakhlin, Beyond UCB: SquareCB, optimal regret with an online regression oracle, via inverse gap weighting.
- 2020 (arXiv) / 2022 (journal): Simchi-Levi and Xu, FALCON and FALCON+: optimal regret with an offline regression oracle.
Setting
There is a finite set of actions and a set of contexts. In round nature draws a context and a reward vector from a fixed unknown distribution ; the learner sees , picks and observes . The conditional mean reward is , and is an action maximizing . The regret after rounds is
An action selection kernel gives a probability distribution over actions for each context. The offline regression oracle takes samples and returns a predictor . It satisfies Assumption 2 with error function if, for every kernel , when the samples are i.i.d. with and , then with probability at least ,
FALCON+ runs in epochs given by a schedule . In epoch it sets and for ; it obtains by feeding only the data of epoch to the oracle (); and in every round of the epoch it plays
where maximizes . The epoch of round is .
Formalization targets
Goal: Theorem 2 in explicit form
If and for , then for every and , with probability at least ,
The paper states Theorem 2 as an bound; this is the explicit inequality its proof establishes, with the additive terms kept. It holds for every oracle, error function and schedule satisfying the hypotheses, so it is the shape of the result rather than a rate for a particular class.
Milestones
In attack order: Lemma A.2 (each oracle call is accurate with probability ; the good event has probability ); Lemma A.3 (the kernel is induced by a randomized policy on ); Lemma A.4 (the conditional expected regret of a round is the implicit regret of ); Lemmas A.5 and A.6 (low predicted regret and bounded divergence ); Lemma A.7 (prediction error of implicit rewards on ); Lemma A.8 (true and predicted implicit regret agree up to factor 2 and ); Lemma A.9 (); Lemma A.10 in expectation; the Azuma display (A.10); and Lemma A.10 with high probability.
Significance
Theorem 2 makes contextual bandits reducible to offline regression with no loss in rate: for any function class with an estimation-error bound, plugging that bound into the theorem gives the regret. With empirical entropy this yields , matching the known lower bound, and for linear classes it improves the best known bound when (Examples 1–2 of the paper). The oracle is called once per epoch on i.i.d. data, so any off-the-shelf supervised learner can be used, with a guarantee transferred automatically.
Formalizing the result produces a machine-checked offline-to-online reduction, independent of the function class. The paper's result is proved, but it has no machine-checked proof; the platform has formal statements for the online-oracle algorithm SquareCB and for the classification-oracle algorithm of Agarwal et al., not for this reduction. The milestones isolate reusable facts about inverse-gap-weighted kernels (Lemmas A.3, A.5, A.6) and an epoch-wise induction pattern (Lemma A.8) that recur in later work on bandits and reinforcement learning with regression oracles.
Difficulty
The obvious argument fails at the regression step. A natural approach feeds all past data to the oracle, but those data are adaptively collected and not i.i.d., so an offline guarantee does not apply to them; FALCON+ uses only the previous epoch's data, which are i.i.d. given the past, and the analysis must make this conditional application of Assumption 2 rigorous on a product space (Lemma A.2). The second obstacle is that the oracle's error is measured under the kernel of the previous epoch, while regret is incurred under the current one. Transferring the error across epochs requires the divergence and the two-sided comparison of true and predicted implicit regret in Lemma A.8, proved by induction on epochs with exact constants; a one-sided bound does not close the induction.
Formalization scope
Lean representation and conventions:
- Finite contexts. is a
Fintypewith the discrete σ-algebra. The paper claims Theorem 2 for arbitrary , but its detailed proof (App. A.1–A.6) is "under the condition that "; the uncountable case (App. A.7) is not formalized. No bound depends on . - Rewards in , the model of §1.2 and the assumption of the proof of Theorem 2. Setup 2's heavy-tailed generality (Condition (5)) is not formalized.
- Nature's law is a context marginal with a Markov reward kernel; is the kernel mean.
- The run lives on the canonical space with the i.i.d. product law of ; actions are drawn by the inverse CDF from the uniform coordinate.
- Probabilities in failure form: "with probability at least " is "the failure set has outer measure at most ".
- Added hypotheses (disclosed in every statement): (Algorithm 2 divides by it); measurability of the oracle and of the tie-breaking rule for (App. A.7, p. 1929, assumes such conditions directly); ; .
- The tie-breaking rule for and the maximizer are universally quantified.
- Quantities the paper indexes by a round are indexed by its epoch; is replaced by the coordinate filtration, which contains it.
A trivializing formalization is ruled out: the goal is a bound on the realized regret of the run of Algorithm 2, whose oracle sees only samples of the previous epoch; it is not a bound for an algorithm that is handed or its population error, or that observes the full reward vector.
Infrastructure needed: conditional application of an i.i.d. guarantee inside a product measure (Fubini over Measure.infinitePi), Azuma–Hoeffding for bounded martingale differences (proved on the platform as bounded_diff_martingale_two_sided), and finite-sum algebra for inverse gap weighting. The kernel lemmas A.3, A.5 and A.6 are reusable beyond this mission. Proofs of any milestone are welcome, as are proofs of the goal from the milestones.
Selected references
- D. Simchi-Levi and Y. Xu, Bypassing the Monster: A Faster and Simpler Optimal Algorithm for Contextual Bandits Under Realizability, Mathematics of Operations Research 47(3):1904–1931, 2022. https://doi.org/10.1287/moor.2021.1193
- D. J. Foster and A. Rakhlin, Beyond UCB: Optimal and Efficient Contextual Bandits with Regression Oracles, ICML 2020. https://arxiv.org/abs/2002.04926
- A. Agarwal, D. Hsu, S. Kale, J. Langford, L. Li and R. E. Schapire, Taming the Monster: A Fast and Simple Algorithm for Contextual Bandits, ICML 2014. https://arxiv.org/abs/1402.0555