A First-Order Approach to Accelerated Value Iteration 1: For a Reversible Policy Chain, Accelerated Value Computation Converges at Rate 1 − √((1−λ)/(1+λ)) + η for Every η > 0Research Paper
Motivation
Value iteration is the standard method for solving a discounted Markov decision process (MDP), and its policy-evaluation variant, value computation (VC), computes the value of a fixed policy by iterating an affine contraction. Both converge geometrically at rate , the discount factor. When is close to one, as in long-horizon reinforcement-learning benchmarks ( or ), the number of iterations grows like , and each iteration costs a full sweep over states and actions.
In convex optimization, gradient descent on a -strongly convex, -smooth function has the same kind of rate, and Nesterov's acceleration improves the dependence on the condition number from to . Goyal and Grand-Clément (arXiv:1905.09963, Operations Research 2023) read as a gradient with "strong convexity" and "smoothness" , transfer Nesterov's and Polyak's step sizes to value iteration, and ask when the transferred methods are actually faster. This mission covers the first answer they give: for policy evaluation on a reversible chain, the accelerated scheme converges at a rate of order instead of .
Setting
An MDP has a finite set of states with , a finite set of actions , transition probabilities (from state to state under action ; each is a probability distribution), real rewards , and a discount factor . A policy assigns to every state a probability distribution over actions. It induces the reward vector , the transition matrix , the policy operator , and the value function
the expected discounted reward when actions are drawn from . VC iterates .
Accelerated Value Computation (A-VC) starts from arbitrary and sets, for ,
with the constant step sizes and . Write .
A transition matrix is reversible if it is irreducible (every state reaches every other with positive probability in some number of steps) and has a stationary probability distribution satisfying detailed balance, for all .
The analysis runs through the matrix
whose powers propagate the stacked error , where and . Its spectral radius is the largest modulus of a complex eigenvalue.
Formalization targets
Goal: Theorem 5.1
If is irreducible and reversible, then , and for every
The goal is stated with the slack because the iteration matrix has a defective leading eigenvalue: the error is of exact order in general, so a clean bound is false.
Milestones
- The step-size identities of the proof of Lemma 5.2: , , .
- Lemma 5.2: for all .
- The eigenvalues of a reversible transition matrix are real (§2, p. 4).
- for a reversible policy chain.
- (5.3): each eigenvalue of solves for some .
- For : .
- Lemma 5.3: .
- (5.6): , with a positive last term.
Significance
The result. Since , A-VC reaches an -approximation of in iterations against for VC, at the same per-iteration cost. This matches the rate of Nesterov's method for strongly convex functions with the paper's identification of the constants. The same paper shows that this improvement does not extend to arbitrary MDPs: Theorem 6.3 gives a lower bound for every first-order method on a chain MDP, which is why reversibility (or a real spectrum, Remark 5.4) is needed here.
Formalizing it. The result is proved in the paper; no machine-checked proof exists. A formal proof needs the reality of the spectrum of reversible chains, the localization of the spectrum of a structured block matrix, and the passage from the spectral radius to a little-o rate for a matrix with a nontrivial Jordan block. The paper's printed statement puts under the square root, . That rate is faster than and is false (a one-state chain is a counterexample). The mission poses the rate that the paper's own proof concludes on p. 9. This is one of four missions from this paper; the other three pose Theorem 5.5 (momentum value computation), Theorem 6.3 (the first-order lower bound) and Theorem 7.1 (safe accelerated value iteration).
Difficulty
The obvious argument says ", so at rate ". That argument breaks twice. First, computing requires knowing where the eigenvalues of lie. For a general stochastic matrix they fill the unit disk, and then the roots of (5.3) can exceed in modulus. Reversibility is what confines them to , and the reality of the spectrum of a reversible chain is not in Mathlib. Second, the eigenvalue of (coming from ) is a double root of (5.3) with a single eigenvector, so is not ; only the -relaxed little-o holds. The eigenvalues of are non-real for , so the whole computation must be done over .
Formalization scope
- Data. and are finite types. The MDP is the data . Validity (stochastic , ) and the policy property are hypotheses. Rewards are arbitrary reals. The initial-state distribution of the paper's tuple is omitted, since nothing here uses it.
- Reused definitions. , , , policies and the kernel condition are the published
FoundationsML.ReinforcementLearningdefinitions; irreducibility, stationarity and detailed balance are the publishedmm_basic. is the discounted reward series, not an arbitrary fixed point of . - Reversibility is irreducibility plus a stationary probability vector in detailed balance. Detailed balance alone would be satisfied by and is not used.
- Norms and rates. Vectors are functions with the sup norm ; "" is Mathlib's little-o along .
- Spectra are complex: is the spectral radius of the complexified matrix, valued in . Using the real spectrum would make Lemma 5.3 a different, weaker statement.
- Indexing. A-VC is a sequence on with given; is used only for .
- Pinned relative to the page. In the goal, is outside the square root, as in the proof. In milestone 6, is written as , its value for .
Ruling out trivial readings: the hypotheses are satisfiable (a two-state chain with is valid, reversible and has a policy), the rate is stated with rather than for some , and is a fixed object rather than a variable constrained by hypotheses.
Needed infrastructure, reusable beyond this mission: the reality and location of the spectrum of reversible stochastic matrices (also used by the sibling mission on momentum value computation); spectra of block matrices with commuting blocks; and the bound from Gelfand's formula. Contributions of any of these as stand-alone lemmas are welcome.
Selected references
- V. Goyal, J. Grand-Clément, A First-Order Approach to Accelerated Value Iteration, Operations Research 71(2), 2023; arXiv:1905.09963v7 (2021). https://arxiv.org/abs/1905.09963
- Y. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Springer, 2004 (reprint 2013). https://doi.org/10.1007/978-1-4419-8853-9
- P. Brémaud, Markov Chains: Gibbs Fields, Monte Carlo Simulation, and Queues, Springer, 1999 (reprint 2013), Ch. 6. https://doi.org/10.1007/978-1-4757-3124-8
- M. L. Puterman, Markov Decision Processes: Discrete Stochastic Dynamic Programming, Wiley, 1994 (reprint 2014). https://doi.org/10.1002/9780470316887