A First-Order Approach to Accelerated Value Iteration 2: For a Reversible Policy Chain, Momentum Value Computation Converges at Rate (1 − √(1/κ))/(1 + √(1/κ)) + η for Every η > 0Research Paper
Why accelerate value computation
Value iteration and its policy-evaluation special case, value computation, are the basic algorithms of dynamic programming for Markov decision processes (MDPs). Both are fixed-point iterations of a -contraction, so they converge at the geometric rate , the discount factor. When is close to , which is the regime of long effective horizons in reinforcement learning ( or ) and in healthcare and operations applications, the number of iterations grows like .
Goyal and Grand-Clément (arXiv:1905.09963, Operations Research 2023) observe that behaves like the gradient of a function with smoothness constant and strong-convexity constant . This suggests importing the accelerated schemes of convex optimization: Nesterov's acceleration and Polyak's momentum (heavy-ball) method (Polyak 1964; Ghadimi, Feyzmahdavian, Johansson 2015). For the affine operator of a fixed policy, they prove that both schemes converge at a rate governed by rather than , with , provided the Markov chain of the policy is irreducible and reversible. This mission poses the momentum result, Theorem 5.5.
Setting
A finite MDP has a state set with , a finite action set , transition probabilities (from state to state under action ; each is a probability vector), rewards , and a discount factor . A policy is a matrix with nonnegative rows summing to ; it induces
The value function of is , the expected discounted reward from each starting state; it is the unique fixed point of .
Reversibility (Definition 2.1): the chain with transition matrix is irreducible, and it has a stationary probability distribution with for all .
Momentum Value Computation (M-VC) starts from arbitrary and iterates
with the constant step sizes (4.2)
The analysis runs through the matrix
which propagates the error of the pair , and its spectral radius , the largest modulus of a complex eigenvalue.
Formalization targets
Goal: Theorem 5.5
For a valid MDP and a policy whose chain is irreducible and reversible, the M-VC iterates with step sizes (4.2) converge to , and for every
The goal states the paper's three clauses as printed: convergence, the little-o rate for every , and the comparison with .
Milestones
- The step-size identity (proof of Lemma B.1).
- Lemma B.1: for all , with .
- The eigenvalues of an irreducible, reversible transition matrix are real (§2).
- (B.1): every eigenvalue of satisfies for an eigenvalue of .
- For , the roots of (B.1) have modulus .
- The identity .
- Lemma B.2: for a reversible chain, .
- for .
Significance
The rate is the optimal rate of the heavy-ball method on strongly convex quadratics with condition number . Theorem 5.5 shows that policy evaluation for reversible chains enjoys the same improvement over , so that an -approximation of needs a number of iterations scaling like rather than . The paper also shows that the reversibility assumption cannot simply be dropped: on a cyclic chain both accelerated schemes may diverge. Lemma B.2 is an exact spectral computation for a structured block matrix that reappears in any heavy-ball analysis driven by a matrix with real spectrum in .
The paper is a sibling series of four missions. This one poses Theorem 5.5; the others pose Theorem 5.1 (Accelerated Value Computation), Theorem 6.3 (a lower bound for first-order methods on a chain MDP) and Theorem 7.1 (Safe Accelerated Value Iteration). The results are proved on paper; none of them has a machine-checked proof. The work here is to formalize the known argument: the linear-recursion representation, the spectrum of , and the passage from spectral radius to a little-o rate.
Difficulty
The obvious argument bounds in some norm and iterates. No such bound by the rate holds in general: is not normal, and for eigenvalues of the two roots of (B.1) coincide, producing Jordan blocks. The rate must come from the spectral radius, and the gap between and is exactly what the absorbs. Computing needs two separate facts: the eigenvalues of are tied to those of through the quadratic (B.1), which needs a determinant identity for block matrices with commuting blocks; and the eigenvalues of are real and lie in , which needs reversibility together with irreducibility, used to make the stationary distribution positive. For non-real the modulus computation fails, which is why the hypothesis cannot be dropped.
Formalization scope
States and actions are finite types. The MDP data is a plain structure, with validity (stochastic , ) a separate hypothesis; the paper's initial distribution plays no role and is omitted. The value function is the Neumann series . The norm on is the sup norm, and is with respect to it as . Spectra and spectral radii are those of the complexified matrices, so that the non-real eigenvalues of are seen. Reversibility uses the published mm_basic notions: irreducibility, a stationary probability distribution, and detailed balance with respect to it; detailed balance alone would be trivially satisfied by and is not used.
Three printed slips are corrected, and each milestone keeps the printed text verbatim. Lemma B.1 is printed with , but is undefined; the statement uses . Lemma B.2 prints , a negative number; its proof computes , the rate of Theorem 5.5, which is stated. On p. 22 the constant is printed as . The second step size, printed in (4.2), is the of (M-VC) and App. B.
A statement that holds only vacuously, for example one in which reversibility is replaced by detailed balance with an arbitrary vector or in which is unconstrained, is ruled out by the hypotheses above. Lemma B.2 is an equality and must not be weakened to an upper bound.
A complete development needs the determinant of a block matrix with commuting blocks, the reality of the spectrum of a reversible chain (implied by the published MarkovMixing.spectral_representation once the stationary distribution is shown positive), Gelfand's formula for matrices, and the convergence of the Neumann series. The spectral facts about reversible chains and the block-matrix spectrum are reusable beyond this mission. Proofs of any milestone are welcome independently.
Selected references
- V. Goyal and J. Grand-Clément, A First-Order Approach to Accelerated Value Iteration, Operations Research, 2023; arXiv:1905.09963v7 (2021). https://arxiv.org/abs/1905.09963
- B. T. Polyak, Some methods of speeding up the convergence of iteration methods, USSR Computational Mathematics and Mathematical Physics 4(5), 1964. https://doi.org/10.1016/0041-5553(64)90137-5
- E. Ghadimi, H. R. Feyzmahdavian, M. Johansson, Global convergence of the Heavy-ball method for convex optimization, European Control Conference, 2015. https://arxiv.org/abs/1412.7457
- P. Brémaud, Markov Chains: Gibbs Fields, Monte Carlo Simulation, and Queues, Springer, 2nd ed. 2020 (cited in the paper as 2013), Chapter 6. https://doi.org/10.1007/978-3-030-45982-6
- M. L. Puterman, Markov Decision Processes: Discrete Stochastic Dynamic Programming, Wiley, 2014. https://doi.org/10.1002/9780470316887