The Statistical Complexity of Interactive Decision Making III: E2D with Hellinger Confidence Sets Has Regret at Most Σ_t sup dec_γ(M_εt(M̄), M̄) + γ·Est_H(T,δ)Research Paper
Motivation
Interactive decision making covers bandits, contextual bandits and reinforcement learning: a learner repeatedly chooses a decision, observes a reward and an observation drawn from an unknown model, and tries to accumulate nearly as much reward as the best decision for that model. Foster, Kakade, Qian and Rakhlin (arXiv:2112.13487) introduced the Decision-Estimation Coefficient (DEC), a single quantity that gives lower bounds on the regret of every algorithm and, through the Estimation-to-Decisions (E2D) meta-algorithm, matching upper bounds. This mission formalizes the upper-bound side in its strongest form: E2D run with Hellinger confidence sets (Option II of Algorithm 1) has regret controlled by the localized DEC, the same quantity that appears in the paper's lower bounds (Theorems 3.1 and 3.2).
The basic E2D guarantee (33) in terms of the global DEC is already posed on Prove2Me as an open goal (FoundationsRL.GeneralDM.e2d_general_regret_bound_v2, the restatement in Foster and Rakhlin's lecture notes, arXiv:2312.16730, Proposition 26). It is not posed again here.
Setting
A decision space is a finite set. An outcome consists of a reward and an observation and ranges over a finite set , and is its reward. A model assigns to every decision a probability distribution on . Its mean reward is , and is a decision maximizing . The learner knows a model class that contains the true model . At round it plays a distribution on , and its regret is
Write for the squared Hellinger distance. For a reference model and the DEC is
The localized class keeps only the models whose optimal value exceeds that of by at most .
E2D with Option II receives at each round an estimate from an online estimation oracle. It keeps the confidence set of models whose accumulated Hellinger error against past estimates, , is at most . It plays a minimizer of the DEC problem over at . Assumption D.1 asks that, with probability , the oracle's Hellinger error on the rounds where is at most a known bound . It also asks that .
Formalization targets
Goal: Theorem 4.1, (35) (in the form (137) of Theorem 4.1a)
With , rewards in and , with probability at least ,
Milestones (the steps of the proof in App. D.1)
- (138): the per-round step. At a minimizer , the instantaneous regret minus times the Hellinger error of is at most .
- Lemma D.1: on the event of Assumption D.1, for every , and the total Hellinger error is at most .
- Lemma D.2: if for all , then for any model , is bounded by plus the average over of the DEC values and Hellinger errors.
- Containment: for every round, with the radius derived on p. 99 using .
Companion results
- Theorem 4.3, (47): E2D for a general divergence and randomized estimators satisfies almost surely. With and point estimates it contains (33), so it is a generalization of the open goal above, not the same statement.
- Theorem 8.1, (87): contextual E2D with arbitrary contexts satisfies .
Significance
The bound is the paper's most general upper bound. Together with the lower bounds of §3, it shows that the localized DEC characterizes the regret of interactive decision making up to the estimation term. With an aggregating estimator satisfying (App. A.3), it recovers the finite-class guarantee of Theorem 3.3. The companions extend E2D to general divergences, randomized estimators and adversarially chosen contexts. That last setting contains SquareCB for contextual bandits.
All results of this mission are proved in the paper. None was machine-checked before. Formalizing them produces a checked reduction from decision making to estimation for finite decision and outcome spaces, stated against the published DEC and localized-class definitions, so that later bounds on the DEC of specific classes (bandits, tabular RL) can be combined with it.
Difficulty
The non-localized bound follows from the per-round step (138) by summation. The localized bound does not: the per-round DEC is taken over the confidence set , and replacing it by a localized class requires that every model surviving in have optimal value close to that of the current estimate . The confidence set constrains Hellinger errors against past estimates, not optimal values, and the estimate itself is in general not a member of the class. Bridging these is the content of Lemma D.2 and the containment. The first round is a boundary case: there the confidence set is the whole class, so the containment holds only if the radius is at least the full range of the rewards.
Formalization scope
- Finite alphabets. The paper works over measurable spaces. This mission takes a finite decision set (a
Fintype), a finite outcome set , and modelsm : S → Y → ℝwith everym πa probability vector. Expectations are finite sums. Contexts in Theorem 8.1 range over an arbitrary type. - Published layer. Hellinger distance, , regret, the DEC (
decGf, anddecfor the supremum over ) and the localized class are the publishedFoundationsRL.GeneralDMdefinitions, referenced and not redefined. - Pathwise statements. A run is a realized sequence of estimates, distributions and confidence sets tied to Algorithm 1 by
IsOptionIIRun(lines 7 and 8 and ). "With probability at least " becomes the hypothesisEventD1, the event of Assumption D.1. Line 8's arg min is relaxed to "certifies the minimax value from above", as Remark 4.1 permits. Rounds are 0-based, and is evaluated at the paper's index. - Added hypothesis in the goal and the containment. Without it the goal is false. With , two decisions, two Bernoulli models and , the first round has positive regret while the right side is . In applications .
- Radius. The goal uses the printed theorem radius with . The containment milestone uses the p. 99 proof radius with . Averaging the DEC terms over the earlier rounds gives the goal's sharper factor.
- Junk values. When a localized class is empty, its Lean DEC is instead of , which only weakens the bound. All other suprema are over nonempty bounded sets.
- Ruled out. Leaving the confidence sets, estimates or distributions unconstrained would make the goal false or trivial; they are tied to the algorithm by
IsOptionIIRun. The open bound (33) is not restated. - Welcome contributions. Convexity and the triangle inequality for , the bound for rewards in , and monotonicity of
decGfin the class are reusable beyond this mission.
Selected references
- D. J. Foster, S. M. Kakade, J. Qian, A. Rakhlin, The Statistical Complexity of Interactive Decision Making, arXiv preprint, v3, 2023. https://arxiv.org/abs/2112.13487
- D. J. Foster, A. Rakhlin, Foundations of Reinforcement Learning and Interactive Decision Making, lecture notes, 2023. https://arxiv.org/abs/2312.16730
- D. J. Foster, A. Rakhlin, Beyond UCB: Optimal and Efficient Contextual Bandits with Regression Oracles, ICML 2020. https://arxiv.org/abs/2002.04926