Foundations of Reinforcement Learning XI: Generalized UCB Has Regret Controlled by the Eluder DimensionTextbook
Why the eluder dimension
A structured bandit is a repeated decision problem in which a learner picks a decision from a set and observes a noisy reward. The set can be large or continuous: a dosage, a price vector, a path in a graph. Treating each decision separately costs regret of order , which is useless when is large relative to the horizon . The learner must generalize across decisions through a function class that is assumed to contain the true mean reward. Knowing that the mean reward is in is not enough on its own: in Example 4.1 of the notes, a class of needle functions has , yet regret of order is unavoidable. A guarantee therefore needs a second complexity measure, one that says how much is learned about one decision from others.
The eluder dimension of Russo and Van Roy (NeurIPS 2013) is such a measure. It was introduced to analyse optimistic algorithms and posterior sampling for general function classes, and it remains the standard yardstick for optimism beyond linear models. This mission is number XI in the formalization of Foster and Rakhlin's lecture notes (arXiv:2312.16730), and it covers §4.1.1 and §4.1.2.
Setting
Let be an arbitrary set of decisions. In each round the learner selects and observes , drawn from a fixed model . The mean reward is , and regret is , with a maximizer of . The learner knows a finite class with .
Generalized UCB forms the empirical minimizer of the square loss on the first rounds, keeps every function whose empirical loss is within of it, in a confidence set , and plays a maximizer of the upper confidence bound .
A sequence is -elusive for if for each some has and . The first layer of the dimension is the length of the longest such sequence. The eluder dimension is its monotone version, floored at one, maximized over the class:
Formalization targets
Goal: Proposition 12
With probability at least ,
with one absolute constant. The goal fixes no constant, so it survives any sharpening of the argument.
Intermediate targets
- Lemma 10, structured form: with probability at least , stays in every confidence set and every function in has cumulative squared error at most against on the past.
- (4.7): on that event, regret is at most the upper confidence width.
- The regret-to-width step, p. 61: , for the width over functions close to on the past.
- Lemma 12: for every sequence and every .
- (4.13), (4.14) and the two displays after them: the deterministic reduction of to the eluder dimension.
Significance
The theorem has no dependence on . For linear models the eluder dimension is of order , which recovers the LinUCB rate up to logarithms, and for generalized linear models with link derivative in it gains a factor . It also shows the limits of the approach: a single ReLU neuron has eluder dimension exponential in (Li et al.). The result is the reference point against which the Decision-Estimation Coefficient of the later chapters is compared.
The result is proved in the literature, so nothing here is open. The eluder dimension has no prior formal treatment on the platform. The mission contributes that definition, in a form usable for any class, together with a combinatorial lemma that makes no reference to probability, and a concentration statement for least-squares confidence sets under an arbitrary adaptive algorithm that the contextual and reinforcement-learning chapters can reuse. The platform's FoundationsRL.Bandits.optimism_lemma is stated for finitely many decisions, so it does not carry over to this setting.
Difficulty
The obvious argument stops at (4.7). The confidence set constrains a function on the decisions already played, while the regret is charged at the new decision , and in general good fit on the past says nothing about a new point. The eluder dimension bounds the number of rounds on which the two can disagree, but its definition has a single scale: squared error on the past and a surprise of . The confidence sets have a fixed radius which is much larger than for small widths. The central step is Lemma 12, which must control the count of large widths at every scale at once, and the elliptical-potential argument for linear models has no analogue for a general class.
The confidence sets themselves need a concentration inequality for dependent data: the decisions are chosen adaptively, so the square loss differences do not form an independent sequence.
Formalization scope
Decisions live in an arbitrary type with no finiteness assumption. The class is a Set of real functions, F.Finite in the goal and arbitrary in Lemma 12. The dimension takes values in ℕ∞, with edimRaw for the underlined quantity and edimFstar, edim for the two layers built from it. The bound is stated in ℝ≥0∞, so it is vacuous, not false, when the dimension is infinite. The argmax over is presupposed by the source's notation, so a run of the algorithm has the maximizer as a hypothesis, and the regret comparator is a supremum. The first clause of (4.9) is strict and the second is not. Rounds are numbered from , as in the notes. The width of Lemma 12 is extended-real valued, because a real supremum would return a junk value on an unbounded class.
The conditional law of the reward is stated on measurable sets for an arbitrary filtration. The constant in the goal is existential and quantified before the types, so it cannot depend on , or . A trivializing reading, in which regret is measured against a fixed decision rather than the best one, or in which the supremum in the width collapses, is ruled out by these conventions.
Contributions are welcome on the eluder-dimension combinatorics, which are reusable beyond this mission, and on the concentration lemma, whose martingale argument also serves the contextual form of Lemma 10.
Selected references
- D. J. Foster, A. Rakhlin, Foundations of Reinforcement Learning and Interactive Decision Making, lecture notes, 2023. arXiv:2312.16730v1
- D. Russo, B. Van Roy, Eluder Dimension and the Sample Complexity of Optimistic Exploration, NeurIPS 2013, pp. 2256–2264. Proceedings
- G. Li, P. Kamath, D. J. Foster, N. Srebro, Eluder Dimension and Generalized Rank, 2021. arXiv:2104.06970