Nonasymptotic Analysis of Monte Carlo Tree Search: UCB with a Polynomial Bonus on a Non-Stationary Bandit Converges and Concentrates PolynomiallyResearch Paper
Motivation
Monte Carlo tree search repeatedly chooses actions using estimates built from earlier simulations. The rewards attached to an action need not form an independent, identically distributed sequence: deeper parts of the search tree change as the search proceeds. Shah, Xie and Xu isolate this difficulty in a non-stationary multi-armed bandit and study a version of upper confidence bound selection with a polynomial exploration bonus. Their bandit theorem is the concentration result used in their subsequent analysis of tree search; it is a distinct finite-sample result with explicit assumptions and constants. Shah, Xie and Xu, §5–§7
The mission concerns Theorem 3 of arXiv:1902.05213v4, including both its expected-reward bound and its two-sided concentration bound. It provides the bandit foundation for the paper's later Monte Carlo tree search statements. The latter require a separate model of the search tree and are outside this mission. Shah, Xie and Xu, Theorems 1 and 3
Setting
A reward table has arms and random variables , where is the reward obtained on the th play of arm . Every reward lies in for a fixed . The arm's empirical mean after plays is , and the expectations of these means converge to . Rewards may depend on earlier rewards, including rewards from other arms. The model imposes no independence or identical-distribution assumption. Shah, Xie and Xu, §5, equation (12)
The per-arm concentration assumption fixes , , and . For every , both tails of beyond have probability at most . This describes each arm's complete reward sequence without specifying a filtration or a sampling law. The limiting optimal arm has mean ; each other arm has gap . Shah, Xie and Xu, §5, equation (13)
A run chooses arm at round . Its count records how many rounds through chose arm , and its average reward is . The polynomial UCB index is for , with . At round , the run maximizes the index formed using counts and bonus time . An unplayed arm has infinite priority, and ties among eligible arms are arbitrary. Shah, Xie and Xu, §5.1–§5.2, equations (14)–(16)
Formalization targets
Expected-reward convergence
Theorem 3.A bounds the gap between expected run reward and the best limiting arm mean. With a positive lower bound for every , and , the target for is
The first term is the optimal arm's own finite-sample bias. The remaining term is explicit in the bandit parameters. Lemma 2 and the displayed decomposition on page 18 are the curated targets for this part. Shah, Xie and Xu, Theorem 3.A and §6
Polynomial concentration
Theorem 3.B states that one constant , depending only on , bounds both tails for every admissible reward table, measurable UCB run, horizon , and :
The tuning parameter satisfies and . Lemma 1, the event inclusion (34), the play-count tail (36), and Lemma 3 are the curated targets for this part. The same theorem statement carries parts A and B. Shah, Xie and Xu, Theorem 3 and §6.1–§6.3
Significance
The expected-reward bound relates the run's finite-horizon mean to the optimal arm's own mean and to an explicit penalty for sampling other arms. The concentration bound says that the run's cumulative reward inherits polynomial two-sided tails at a derived exponent. Together these claims let the paper apply a bandit guarantee in its analysis of Monte Carlo tree search, where an action's reward process changes as the search continues. Shah, Xie and Xu, §6–§7
The result is proved in the cited paper; the mission seeks a machine-checked Lean development of that known result. Its reward-table model and finite-sample count estimates may be reused when a sequential choice rule is analyzed without independent arm rewards. The mission currently provides compiled definitions and theorem statements as proof targets, not verified proofs of those targets.
Difficulty
A standard stationary-bandit analysis would replace each arm's history by independent draws from a fixed law. Here the assumptions give only convergence of expected arm means and polynomial per-arm concentration. A run also samples arms adaptively, so the random number of observed rewards on an arm cannot simply be inserted into a fixed-sample tail inequality. The finite-sample guarantee has to retain the exact dependence on the bonus, arm gaps, pull counts, and the tuning window. The paper's intermediate results identify the required count and tail statements. Shah, Xie and Xu, §6.1–§6.3
Formalization scope
Lean uses a probability space , a table , and a run . Play numbers begin at one; index zero is unused. The model makes reward measurability, boundedness, limiting expected means, and both concentration inequalities explicit. The run requires measurable arm-selection events so that its probabilities and bounded expectations are genuine. The model does not add independence, an identical-distribution condition, or a filtration.
The real index is compared only when every arm under comparison has a positive count; a separate clause gives unplayed arms the paper's priority. The time argument of the index at round is . The thresholds and use the paper's closed ceiling formulas. The paper's is represented by a positive lower bound on all suboptimality gaps; setting recovers the printed statement when that minimum is defined. Theorem 3 explicitly requires and . The existential precedes the probability space, rewards, run, , and , so it cannot change with an instance or horizon. Both parts of Theorem 3 are present, and the model's hypotheses are satisfiable for ordinary bounded bandits.
The mission needs reusable definitions for reward tables, empirical and run averages, pull counts, measurable runs, UCB selection and polynomial play thresholds. Contributions proving the six curated milestones, connecting them to the goal, or establishing reusable measurability and finite-sum facts are in scope. The pathwise two-sided reward decomposition is also posed as a supporting theorem.
Selected references
- Devavrat Shah, Qiaomin Xie, and Zhi Xu, Non-Asymptotic Analysis of Monte Carlo Tree Search, arXiv:1902.05213v4, 2020; published in Operations Research, 2022. arXiv preprint