Explore First, Exploit Next: The True Shape of Regret in Bandit Problems IV: Under Pairwise Symmetry a Suboptimal Arm Is Drawn T/K Times or Almost as Often as an Optimal ArmResearch Paper
Motivation
Lower bounds for stochastic multi-armed bandits are usually stated for a large horizon: Lai and Robbins (1985) and Burnetas and Katehakis (1996) show that a consistent strategy pulls a suboptimal arm at least about times. These bounds say nothing about the small-horizon regime, the first rounds in which a strategy has not yet had time to separate the arms. Garivier, Ménard and Stoltz (arXiv:1602.07182v3, Mathematics of Operations Research, 2019) describe that regime through one information-theoretic inequality, their (6), and derive several lower bounds from it. This mission formalizes their relative lower bound, Theorem 3 (p. 11): under a mild symmetry assumption on the strategy, a suboptimal arm is either pulled at the uniform rate or, on average, almost as often as an optimal arm. It is the statement behind the title's claim that bandit strategies must first explore.
Setting
There are arms. A bandit problem assigns to each arm a probability distribution on with expectation . The optimal mean is , the gap of arm is ; arm is optimal if and suboptimal if .
A model is a set of distributions on , each with an expectation; a bandit problem is in when every belongs to .
At each round a strategy picks an arm , possibly at random, as a function of the past arms and rewards, and receives a reward drawn from . The number of pulls of arm in the first rounds is
and denotes expectation when plays against . Write .
Kullback–Leibler divergence. is the Kullback–Leibler divergence between distributions, and the divergence between Bernoulli laws of parameters and .
Pairwise symmetry (Definition 3, p. 11). A strategy is pairwise symmetric for optimal arms on if for every bandit problem in , every pair of optimal arms with and every , the pairs and have the same distribution. A strategy that uses only the observed payoffs, and not the labels of the arms, has this property.
Formalization targets
Goal: Theorem 3 (p. 11)
For every model , every strategy pairwise symmetric for optimal arms on , every bandit problem in , every suboptimal arm , every optimal arm with and every ,
Companion: the "in particular" clause
For the second alternative becomes .
Milestones (p. 12)
- The symmetry identity: for the alternative problem obtained by replacing by , .
- Inequality (12): .
- The Jensen display: with .
- The bound in the case , .
Significance
The theorem shows that in the initial phase a symmetric strategy cannot discard a suboptimal arm quickly: unless the arm is already sampled at the uniform rate, it is sampled, in ratio, nearly as often as an optimal arm, as long as is small compared with . Together with Theorem 2 (absolute lower bound) and Theorem 4 (collective lower bound) it describes the regret curve before the logarithmic regime: roughly linear growth, then a transition. It requires no assumption on the model beyond the existence of expectations, and it involves the divergence between the two arms themselves rather than the model-dependent quantity .
The result is proved in the paper; to our knowledge it has no machine-checked proof. The mission produces a Lean statement of Theorem 3 on the canonical bandit model already used on the platform (StochasticBandit, BanditPolicy), a formal version of the symmetry assumption that sibling missions reuse (mission V, the collective lower bound, uses the same Definition 3), and the four intermediate steps of the proof as separate targets.
Difficulty
The proof compares with an alternative problem that differs from it only in arm , and transfers information between them through the fundamental inequality (6) of §2, a data-processing argument on the law of the whole history. The step specific to Theorem 3 is the choice of the test variable: the expectation of is known exactly under only because the strategy is pairwise symmetric, and the symmetry is a statement about joint laws of pull counts, not about their expectations. A version of the assumption that only equates expected pull counts does not determine the expectation of this ratio. The ratio expectation then has to be extracted from a bound on , which needs Jensen's inequality in the right direction and a case split on whether .
Formalization scope
Arms are Fin K (0-based). The horizon is a natural number. A strategy is a published BanditPolicy: a family of Markov kernels from the observed history of (arm, reward) pairs to the next arm. Every strategy of the paper, which may use auxiliary uniform random variables, induces such a kernel with the same law of arms and rewards. The law of the history is the published banditMeasure, and is the published armPullCount. The model is a set of measures on , and every statement assumes the paper's standing hypothesis (§1.1, p. 3) that each member is a probability measure with an integrable identity, together with in .
Committed conventions:
- and take values in ; the Bernoulli divergence is the Kullback–Leibler divergence of two Bernoulli laws on , not a real formula with finite values at .
- Theorem 3 adds the hypothesis . When the divergence is infinite the paper's bound reads and is empty; reading as the real number would make the statement false.
- The "in particular" clause is stated as in , which needs no finiteness assumption.
- Definition 3 is an equality of joint laws (push-forwards of the history law), for every , only for optimal arms of problems in with equal distributions. Invariance under all permutations of the arms would strengthen the hypothesis and weaken the theorem.
- Ratios use as real numbers, so no division by zero occurs; all integrands are bounded by .
A formalization in which the symmetry assumption equates only expected pull counts, in which the divergence is a real number with junk value at , or in which the disjunction is replaced by its second alternative, is not this theorem.
The mission builds on the published definitions StochasticBandit and BanditPolicy. A complete development needs the fundamental inequality (6) for kernel policies (the divergence decomposition, published as BanditAlgorithm.bandit_divergence_decomposition, plus data processing for the Bernoulli test), Pinsker's inequality for Bernoulli laws (proved on the platform as BanditAlgorithm.bernoulli_relative_entropy_pinsker), Jensen's inequality for concave functions (Mathlib), and a lemma that the law of the history is unchanged in distribution when two identically distributed optimal arms are relabelled. The measurability of pull counts and the relation between the extended-real and real Bernoulli divergences are reusable beyond this mission. Proofs of any milestone, and of the general fundamental inequality, are welcome.
Selected references
- A. Garivier, P. Ménard, G. Stoltz, Explore First, Exploit Next: The True Shape of Regret in Bandit Problems, Mathematics of Operations Research 44(2), 2019. arXiv:1602.07182v3, doi:10.1287/moor.2017.0928
- T. L. Lai, H. Robbins, Asymptotically efficient adaptive allocation rules, Advances in Applied Mathematics 6(1), 1985. doi:10.1016/0196-8858(85)90002-8
- A. N. Burnetas, M. N. Katehakis, Optimal adaptive policies for sequential allocation problems, Advances in Applied Mathematics 17(2), 1996. doi:10.1006/aama.1996.0007
- T. Lattimore, C. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020. doi:10.1017/9781108571401