A Short Proof of the Gittins Index Theorem: For Finite-State Semi-Markov Bandits, Every Priority Policy Ordering States by the Index Algorithm's Indices Is OptimalResearch Paper
Motivation
The multi-armed bandit problem with discounted rewards asks how to allocate effort among independent projects, only one of which can be worked on at a time, when the project worked on evolves randomly and the others stay frozen. It models job scheduling, research planning, search, and sequential clinical experimentation. Gittins and Jones showed in 1974 that the problem has an index solution: each state of each project has a number, computed from that project alone, and it is optimal always to work on the project whose current state has the largest number. The result turns an -dimensional dynamic program into one-dimensional computations.
Several proofs followed, each with its own model and its own definition of the index:
- Gittins and Jones (1974) and Gittins (1979): the original interchange proofs, with the index as a supremum over stopping times of a reward-to-time ratio.
- Whittle (1980): a proof through the retirement option and the dynamic-programming equation of the multi-armed problem.
- Varaiya, Walrand and Buyukkoc (1985): the largest-remaining-index algorithm for finite-state Markov bandits, which computes the indices by removing states one at a time.
- Weber (1992): the "prevailing charge" proof.
- Tsitsiklis (1994): the short proof formalized here, for semi-Markov bandits with finite state spaces, by induction on the total number of states, with the indices produced by an index algorithm of the largest-remaining-index type.
Setting
There are bandit processes. Bandit has a finite nonempty state space , and is their disjoint union. When bandit at state is played, a random reward is received, the bandit is busy for a random time , and it then moves to a random state ; the other bandits do not move. The joint law of depends only on , and different plays are independent.
A policy is a map : at time zero and at every play completion time it chooses the bandit to play from the current states. With the start time and the reward of the -th play and a discount rate , the expected discounted reward from the initial joint state is
A policy is optimal if it maximizes for every initial state . A priority rule is a policy for which some ordering of makes it always play the bandit whose current state is ordered highest.
The reward rate of a state is
Reducing bandit by removing a state merges every play that ends in with the following run of plays at into one composite play, whose reward rate is of (2.3). The index algorithm repeatedly picks a state of maximal current rate, sets its index to that rate, and reduces its bandit by removing (or removes the bandit if was its last state).
Formalization targets
Goal: Theorem 2.2
Let be the indices of a run of the index algorithm. Then every priority rule whose ordering satisfies
is optimal. Ties in may be ordered either way, and the run may break ties in its own choices arbitrarily.
Milestones, in the order the proof uses them
- (2.2), the interchange inequality when .
- Lemma 2.1: if , some optimal policy always plays bandit when it is at .
- Reduction step: optimizing within the policies giving top priority is a bandit problem with one state fewer, the reduced bandit with rates (2.3).
- Theorem 2.1: some priority rule is optimal.
- Pick-order optimality: a priority rule that orders the states as a run of the index algorithm picks them is optimal.
- Index monotonicity: along a run, for consecutive picks .
- Picked-before characterization: can be picked before (in a run with the same indices) if and only if .
Significance
The theorem shows that the optimal priority order is determined by numbers that, by the reduction formulas, depend only on the statistics of each bandit separately. It is the basis of index policies in scheduling (the rule and its semi-Markov generalizations), of Klimov's problem and of branching bandits, which the paper's Section 3 obtains by modifying the same argument.
The result has been proved many times, but not in this generality on Prove2Me. The platform has a discrete-time Gittins index theorem for arms sharing one transition kernel and one reward function, with the index defined as a stopping-time supremum. This mission adds semi-Markov dynamics with random play durations and continuous discounting, heterogeneous finite bandits, priority rules, and the index algorithm as the definition of the index. A complete development would also give a machine-checked proof of correctness of a state-elimination algorithm for computing Gittins indices.
Difficulty
Lemma 2.1 compares an optimal stationary policy with a modified policy that is not stationary, and closes the gap by the optimality of stationary policies, which itself requires the dynamic-programming theory of discounted semi-Markov decision problems with finite state spaces. The reduction step requires an exact identification of the values of all top-priority policies with values in a different problem whose plays are composite and may never end. Theorem 2.1 is an induction over a family of problems with shrinking state spaces that are no longer raw semi-Markov bandits. Theorem 2.2 adds ties: a priority order compatible with the indices need not be the order of the given run, and one needs another run with the same indices.
Formalization scope
- Bandits are indexed by
Fin n(), with and each state type finite and nonempty. is the sigma typeΣ i, X i, disjoint by construction. - The law of is a probability measure on with almost surely and integrable. The model also assumes , which the page leaves implicit: it makes the denominator of (2.1) nonzero.
- is computed from the moments and : it is the limit of the expected discounted reward of the first plays, given by the first-step recursion. No path space is constructed. The row sums of are below , so the limit exists; it never mentions rates, indices or reductions.
- Policies are the paper's stationary deterministic maps of the current joint state, and optimality is within that class, for every initial state. An ordering of is an injective ranking into .
- The reduced bandit is given by explicit formulas for its expected discounted reward, discounted duration and discounted transition data. These make the page's "transition probabilities are suitably modified" concrete, and the reduced rate is (2.3). Removing a bandit is the reduction of its last state.
- The index is that of any run of the index algorithm, with arbitrary tie choices.
A formalization in which the value of a policy is defined through rates or indices, or in which is any function compatible with some order rather than the output of the algorithm, would make the goal trivial or empty; both are excluded by the definitions above.
Needed infrastructure: the convergence of the -play values and the first-step (Bellman) equation for ; existence of an optimal stationary policy for finite discounted semi-Markov problems; the value identity for composite plays. The last two are reusable for any finite semi-Markov decision problem. Proofs of individual milestones, alternative arguments (Weber's prevailing charge, the retirement formulation), and the separability statement of p. 198 ("the index algorithm can be carried out separately for each different bandit") are welcome.
Selected references
- J. N. Tsitsiklis, A Short Proof of the Gittins Index Theorem, Annals of Applied Probability 4(1) (1994), 194–199.
- J. C. Gittins and D. M. Jones, A dynamic allocation index for the sequential design of experiments, in Progress in Statistics (J. Gani et al., eds.), North-Holland, 1974, 241–266.
- J. C. Gittins, Bandit processes and dynamic allocation indices, J. Roy. Statist. Soc. Ser. B 41 (1979), 148–177. https://doi.org/10.1111/j.2517-6161.1979.tb01068.x
- P. Whittle, Multi-armed bandits and the Gittins index, J. Roy. Statist. Soc. Ser. B 42 (1980), 143–149. https://doi.org/10.1111/j.2517-6161.1980.tb01111.x
- P. P. Varaiya, J. C. Walrand and C. Buyukkoc, Extensions of the multiarmed bandit problem: the discounted case, IEEE Trans. Automat. Control 30 (1985), 426–439. https://doi.org/10.1109/TAC.1985.1103989
- R. Weber, On the Gittins index for multiarmed bandits, Annals of Applied Probability 2(4) (1992), 1024–1033. https://doi.org/10.1214/aoap/1177005588