On the Gittins Index for Multiarmed Bandits 3: The Optimal Expected Discounted Reward Is a Submodular Function of the Set of Available BanditsResearch Paper
Motivation
The multiarmed bandit problem asks how a gambler should allocate successive plays among independent projects, each of which changes state only when it is played, so as to maximize the expected total-discounted reward. Gittins and Jones (1974) showed that it is optimal to play, at every epoch, a project of greatest Gittins index, a number computed from that project alone. The result underlies job scheduling on a single machine, sequential clinical trials and many stochastic resource-allocation models.
Weber (1992) gave a short proof of the index theorem through prevailing charges, and used the same idea to prove a result that was new: the optimal value is a submodular function of the set of projects that are available. Submodularity of the value says that the marginal worth of an extra project shrinks as more projects are on hand. Two consequences appear in the paper: for statistically identical projects started in a common state, the optimal value is concave in their number; and a bound of Tsitsiklis (1986) for a problem with a retirement option is a special case.
Timeline. Gittins and Jones (1974): the index theorem, by an interchange argument (book-length treatment in Gittins (1989)). Whittle (1980): a proof through a retirement option. Varaiya, Walrand and Buyukkoc (1985): the interchange argument in a general setting. Tsitsiklis (1986): a short proof of the index theorem, including the bound for a retirement bandit. Weber (1992): the prevailing-charge proof and the submodularity of the value (Theorem 4).
Setting
There are bandits . Bandit is a Markov chain on a state space with transition kernel ; at each epoch the gambler plays one bandit , receives the reward , the played bandit moves one step of its chain and the others stay where they are. Rewards are nonnegative and uniformly bounded, and is the discount factor. The initial states are fixed. A policy chooses from the history; its value is .
For a set of bandits , the restricted problem is the same game played with the bandits of only, each started in its state in , and
The fair charge of a state is the largest ratio of expected discounted reward to expected discounted time over stopping times of a single bandit started at (Weber's (4)). On a fixed realisation, where bandit passes through states on its successive plays, the prevailing charge after plays is , a nonincreasing sequence. is the sum of the first terms of the interleaving of the sequences , , into nonincreasing order.
Formalization targets
Goal: Theorem 4, eq. (9)
This holds for arbitrary bandits (no identical-bandit hypothesis) under the standing assumptions of the paper's Section 1.
Milestones on the path
- Eq. (10): for nonincreasing nonnegative sequences and every , . A deterministic inequality.
- The representation used in the last line of the proof:
the expectation over independent realisations of the bandits' state sequences.
Consequences stated in the paper
- Identical bandits started in a common state: of the -bandit problem is nondecreasing and concave.
- Tsitsiklis's bound: if bandit always pays , then for .
Significance
Submodularity of the value is a structural fact about the bandit problem that the index theorem alone does not give: the index theorem identifies an optimal policy for a fixed set of projects, while (9) compares optimal values across sets. It yields diminishing returns from adding projects, the concavity of the value in the number of identical projects, and Tsitsiklis's retirement bound, which separates one project from the rest. Bounds of this kind are used when choosing which projects to make available before allocation begins.
Theorem 1 of the paper, the optimality of the Gittins index policy, is already on Prove2Me as BanditAlgorithm.gittins_index_theorem (Proved, Shuze Chen), together with the discounted Markov bandit model and the Gittins index (GittinsIndex, Shuze Chen) and a finite-horizon form of the prevailing-charge coupling (BanditAlgorithm.gittins_finite_prevailing_charge_eq_stack_coupling_value, Harry_Xu). This mission builds on those items. Theorem 4, inequality (10) and the infinite-horizon representation of have no machine-checked proof that we know of. The other two missions of this series formalize Weber's Theorem 2 (mission 1, a suboptimality bound by prevailing charges) and Theorem 3 (mission 2, a suboptimality bound by index gaps); this mission does not depend on them.
Difficulty
Inequality (10) concerns sums of largest terms of merged sequences; it fails without nonnegativity when and are disjoint, and a proof must use both the monotonicity of each sequence and the sign of the terms. The representation of requires the fair-game property of prevailing charges in infinite horizon: that the optimal policy of collects in expectation exactly the discounted charges it pays, and that, on a common probability space of independent bandit trajectories, the charges paid by the optimal policy are the nonincreasing interleaving. The finite-horizon coupling is on the platform; passing to infinite horizon, identifying the interleaving with the play of an index policy, and handling ties and measurability are the remaining work. Finally, (10) is an undiscounted, pathwise statement while (9) compares discounted expectations, so the two must be connected through series of increments of and an exchange of expectation and series.
Formalization scope
The model is the published BanditAlgorithm discounted Markov bandit: one standard Borel state space , one kernel and one measurable reward shared by all bandits; bandits with different dynamics are encoded by a disjoint union of state spaces. is the mean reward. The value of a policy is the series of per-round expected rewards (markovBanditDiscountedValue), equal to the expectation of the discounted sum for bounded rewards. The fair charge is the published gittinsIndex.
Explicit readings of the paper's phrases:
- "the restriction of the problem to bandits in ": the -armed game whose arms are the bandits of in increasing order, started at their states in (
restrictedValue), with set explicitly; - "the maximal expected total-discounted reward": the supremum over all history-dependent randomized policies;
- "a fixed realisation of the states through which bandit evolves": a point of the product of independent Markov chain laws
Measure.pi (fun j => markovChainMeasure P (x j)); - "interleaving into nonincreasing order … summing the first elements":
partialInterleavedSum, the maximum over allocations of plays of the sum of initial segments, equal to the interleaving sum for nonincreasing sequences, with ; the sequences start at (the printed is a misprint); - "multiplying (10) by , summing": the discounted charge is , the increments of ;
- "rewards are nonnegative and uniformly bounded", : binders of every probabilistic statement, together with
StandardBorelSpace SandMeasurable r; - "concave increasing": nondecreasing and with nonincreasing increments, including ;
- "bandit 1 always pays ": an absorbing initial state with reward .
Trivializing formalizations are ruled out: and are defined as rather than obtained from a supremum over an empty type, the supremum defining is over a nonempty family bounded by , is a maximum over a finite nonempty set, and the hypotheses are satisfiable (checked on a one-bandit instance).
A complete development needs: the infinite-horizon prevailing-charge identity for index policies, measurability of the charge stacks (the measurability of gittinsIndex is on the platform), and a combinatorial lemma on merges of nonincreasing sequences. Inequality (10) and the merge lemmas are reusable beyond bandits, for sums of largest elements and weighted rank functions of uniform matroids.
Selected references
- R. Weber, On the Gittins index for multiarmed bandits, Ann. Appl. Probab. 2(4), 1024–1033, 1992. https://doi.org/10.1214/aoap/1177005588
- J. C. Gittins, D. M. Jones, A dynamic allocation index for the sequential design of experiments, in Progress in Statistics (J. Gani, ed.), North-Holland, 241–266, 1974.
- J. C. Gittins, Bandit Processes and Dynamic Allocation Indices, Wiley, 1989.
- P. Whittle, Multiarmed bandits and the Gittins index, J. Roy. Statist. Soc. Ser. B 42, 143–149, 1980.
- P. Varaiya, J. Walrand, C. Buyukkoc, Extensions of the multiarmed bandit problem: the discounted case, IEEE Trans. Automat. Control 30, 426–439, 1985.
- J. N. Tsitsiklis, A lemma on the multiarmed bandit problem, IEEE Trans. Automat. Control 31, 576–577, 1986.
- T. Lattimore, Cs. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020, Chapter 35. https://doi.org/10.1017/9781108571401