A First-Order Approach to Accelerated Value Iteration 3: On the n-State Chain MDP, Every First-Order Method Has ‖v_s − v*‖∞ ≥ λ^s/(1+λ) for s = 1, …, n − 1Research Paper
Motivation
Value iteration (VI) is the basic algorithm for computing the optimal value function of a discounted Markov decision process (MDP). Its error contracts by the discount factor per step, so it needs on the order of iterations, which is slow when is close to , the regime of interest in long-horizon planning and reinforcement learning. Goyal and Grand-Clément (arXiv:1905.09963, Operations Research 2023) read VI as gradient descent on the residual map and import Nesterov's acceleration and heavy-ball momentum from convex optimization. In convex optimization the corresponding lower bounds of Nesterov (2004/2013) show that acceleration is optimal among first-order methods. The question this mission formalizes is the MDP counterpart: can any method that only combines past iterates and their Bellman images beat the rate of plain VI on every MDP? Theorem 6.3 of the paper answers no.
Setting
A finite MDP consists of a state set with , a finite nonempty action set , transition probabilities with , rewards , and a discount factor . Its Bellman operator is
a -contraction for the sup norm . Its unique fixed point is the optimal value function.
A sequence in is a first-order sequence for if
VI from () and relaxed VI () from produce such sequences.
The chain MDP of Figure 1 has states and a single action. State is absorbing with reward ; every state moves deterministically to and earns nothing. In Lean (namespace AccelVI.LowerBound) the MDP is MDP, validity is MDP.Valid, is bellman, the chain is chainMDP n lam, and the first-order condition is IsFirstOrder T v.
Formalization targets
Goal: Theorem 6.3 (p. 12)
For every , every , the fixed point of the Bellman operator of the -state chain, every first-order sequence for , and every ,
Milestones
- is a -contraction: (§2, p. 3).
- (3.5): (p. 6).
- (3.6): (p. 6).
- On the chain, the value function is , and it is the only fixed point of (p. 12).
- On the chain, every first-order sequence satisfies for (p. 12).
- (6.2)–(6.4): on the chain, for (p. 12).
Significance
The result. Theorem 6.3 shows that, measured by the number of Bellman evaluations, no first-order method improves on the worst-case rate of value iteration uniformly over MDPs, as long as the number of steps is below the number of states. In convex optimization the analogous class admits acceleration from to ; with , , the analogue for MDPs would be a rate of about , and this theorem rules it out. It is the reason the paper's accelerated schemes are analysed under extra structure, such as reversible policy-evaluation chains, or are safeguarded.
Formalizing it. The result is proved in the paper; no machine-checked version exists. The mission produces a checked lower bound for fixed-point iterations of Bellman operators, the contraction inequalities (3.5)–(3.6) for a general finite MDP, and an explicit value function for a hard instance. The paper's other main results, Theorems 5.1, 5.5 and 7.1, are posed as the sibling missions 1, 2 and 4.
Difficulty
The obvious argument that the error is at least the contribution of states never reached fails without control of the span: the first-order class allows arbitrary real coefficients, including ones that change with and depend on all past iterates, so one has to show that no such combination puts mass on states beyond index . The support claim has to be carried through the nonlinear max in , which is benign here only because the chain has a single action. The second ingredient converts a bound on a few coordinates of into a bound in sup norm with the exact constant , which requires the two-sided inequalities (3.5)–(3.6) for and the explicit value .
Formalization scope
- States are
Fin nand are 0-based: the paper's state is index , so islam ^ k / (1 - lam)at index and "" is . The action set of the chain isUnit. - is Mathlib's sup norm on
Fin n → ℝ. "span" is the real linear spanSubmodule.span ℝ, with both lists running to index inclusive. - enters the goal through its defining equation ; it is not replaced by the explicit vector. The initial distribution of the paper's MDP tuple is omitted, since no result here uses it.
- Pinned quantifiers. As printed, Theorem 6.3 asserts that there exists an MDP instance, with its own number of states; read literally, an instance with has no admissible . The proof fixes an arbitrary and and uses the chain. The goal states that claim for every and on the explicit chain, which implies the printed form.
- is natural-number subtraction, so the goal has content exactly when .
- The goal's constant is as printed; the sharper bound , which the proof's chain also gives, is not the target.
- The Bellman operator and the kernel condition are the published platform definitions
FoundationsML_ReinforcementLearning_BellmanOperatorandFoundationsML_ReinforcementLearning_IsTransitionKernel. - A trivializing formalization would take an unsatisfiable fixed-point hypothesis or an empty first-order class. Both are satisfiable: the explicit vector is a fixed point (milestone 4), and VI from is first-order.
Contributions welcome: proofs of the contraction inequalities for general finite MDPs (reusable by every Bellman-operator mission), of the support recursion, and of the final chain of inequalities.
Selected references
- V. Goyal and J. Grand-Clément, A First-Order Approach to Accelerated Value Iteration, Operations Research 71(2), 2023; preprint arXiv:1905.09963v7, 2021. https://arxiv.org/abs/1905.09963
- Yu. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Springer, 2004. https://doi.org/10.1007/978-1-4419-8853-9
- M. L. Puterman, Markov Decision Processes: Discrete Stochastic Dynamic Programming, Wiley, 1994/2014. https://doi.org/10.1002/9780470316887