Asynchronous Stochastic Approximation and Q-Learning 1: Bounded Asynchronous Stochastic Approximation Iterates Converge with Probability 1 to the Unique Fixed Point of a Monotone Continuous MapResearch Paper
Motivation
Q-learning (Watkins, 1989) learns the optimal action values of a Markov decision problem from sampled transitions, without a model of the transition probabilities. At each step only one state–action pair is updated, with a noisy estimate of the Bellman operator evaluated at possibly outdated values of the other pairs. Watkins and Dayan (1992) gave a first convergence proof for discounted problems. Tsitsiklis (1994) and, independently, Jaakkola, Jordan and Singh (1994) placed Q-learning inside the theory of stochastic approximation: the Robbins–Monro scheme of iterating with decreasing stepsizes and zero-mean noise, here in an asynchronous form where different components are updated at different times using delayed information, as in the distributed iterations of Bertsekas and Tsitsiklis (1989).
Tsitsiklis's paper proves convergence with probability 1 under two structural hypotheses on the iteration mapping : a weighted maximum-norm contraction (Theorem 3), which covers discounted Q-learning, and monotonicity (Theorem 2), which covers undiscounted (stochastic shortest path) problems, whose Bellman operator is monotone but in general not a contraction. This mission formalizes the monotone case.
Setting
A mapping with components is given, and the goal is to solve . Time is discrete, . The iterate evolves by
where is a stepsize ( when component is not updated), is noise, and are delays: the update of component may read component as it was at an earlier time. All quantities are random variables on a probability space with an increasing sequence of σ-fields , the history up to the choice of the stepsizes at time .
Vector inequalities are componentwise and is the vector of ones. The paper's assumptions are:
- Assumption 1: every delay index tends to infinity, with probability 1.
- Assumption 2: , , are -measurable and is -measurable; ; and for deterministic constants .
- Assumption 3: and with probability 1, for a deterministic .
- Assumption 4: is monotone (), continuous, has a unique fixed point , and for all and all .
In Lean the algorithm is the structure AsyncSA.Monotone.Algorithm with fields F, x, α, w, τ, and the assumptions are Assumption1–Assumption4.
Formalization targets
Goal: Theorem 2 (p. 189)
Under Assumptions 1–4, if is bounded with probability 1, then
Boundedness is a hypothesis, not a conclusion: for monotone it must be established separately (the paper does so for Q-learning in §7). The bound may differ between sample paths.
Milestones
- Lemma 1 (p. 190): for a scalar recursion with martingale-difference noise whose conditional variance is bounded by an a.s. bounded , and stepsizes with , , with probability 1.
- Lemma 4 (p. 193): the sequences , , and , , satisfy and .
- Lemma 5 (p. 193): and .
- Lemma 6 (p. 194): on a sample path where from time on and all delays exceed from time on, , where relaxes from towards and accumulates the noise from .
- Lemma 7 (p. 195): under the choices of and made on pp. 194–195, for all .
Significance
Theorem 2 is the convergence result for asynchronous stochastic approximation driven by a monotone mapping whose fixed point is unique. Its main application is Q-learning for stochastic shortest path problems with improper policies excluded, and more generally any learning scheme whose expected update is a monotone, sup-norm nonexpansive operator in the sense of (7) (dynamic programming operators are the standard examples). The same pattern, an a.s. bounded iterate plus a monotone mean map, recurs in later analyses of asynchronous and distributed learning algorithms.
The result has a published proof but, as far as the platform's catalog shows, no machine-checked one: there is no formalized asynchronous stochastic approximation theorem, and no formal Lemma 1 for random stepsizes and random conditional-variance bounds. A complete development provides the scalar almost-sure convergence lemma, the deterministic envelope argument for monotone maps, and the pathwise comparison steps, each of which is reusable.
Difficulty
The noise has a conditional variance that grows with the iterate, so it is neither bounded nor integrable a priori, and the stepsizes are random and chosen adaptively. The classical deterministic-bound argument for Lemma 1 does not apply directly. The iteration is not a contraction in any norm, so there is no Lyapunov function giving a geometric decrease, and the delays mean that the mapping is evaluated at a vector no single time index describes. Convergence has to be propagated from the deterministic envelopes to the random iterate through a sequence of random times, one envelope at a time, and each step requires the noise accumulated after a random time to be small uniformly in the remaining horizon.
Formalization scope
- Components are
Fin n(0-based); vectors areFin n → ℝwith the componentwise order, soMonotone Fis Assumption 4(a) exactly. isr • 1. - The update is stated in the paper's unified form for every , with off the update times; the update sets are not represented. The paper sets off ; the formalization leaves free there, which is more general. and hold on every sample path.
- Conditional expectations are generalized, written with set integrals (
CondMeanZero,CondSqLe): for every -set on which is integrable, and in . Mathlib'scondExpis not used: it is for non-integrable functions and would make the noise assumptions vacuous. - Series conditions use partial sums (divergence to ; every partial sum at most ), never
tsum, which is for non-summable sequences. The constants are chosen before the almost-sure quantifier; the bound on is chosen after it. - The paper treats as determined by and uses this implicitly; Theorem 2 assumes it explicitly (
Adapted). - Lemma 1 leaves unconstrained, as on the page. Lemmas 4 and 5 hold for every . Lemmas 6 and 7 are pathwise: they fix one sample path and take the induction's objects (, , , , , ) as given with exactly the properties the paper has established for them; they are not existential statements about these times.
- A trivializing formalization (junk conditional expectations,
tsumfor the series, a deterministic bound on , or a second unrelated ) is ruled out by the choices above.
Contributions welcome: a general almost-sure convergence theorem for Robbins–Monro recursions with generalized conditional moments (Lemma 1 and its tail version ), and lemmas on monotone maps satisfying (7).
Selected references
- J. N. Tsitsiklis, Asynchronous Stochastic Approximation and Q-Learning, Machine Learning 16 (1994) 185–202. https://doi.org/10.1023/A:1022689125041
- C. J. C. H. Watkins and P. Dayan, Q-learning, Machine Learning 8 (1992) 279–292. https://doi.org/10.1007/BF00992698
- T. Jaakkola, M. I. Jordan and S. P. Singh, On the Convergence of Stochastic Iterative Dynamic Programming Algorithms, Neural Computation 6 (1994) 1185–1201. https://doi.org/10.1162/neco.1994.6.6.1185
- D. P. Bertsekas and J. N. Tsitsiklis, Parallel and Distributed Computation: Numerical Methods, Prentice Hall, 1989. https://hdl.handle.net/1721.1/3719
- B. T. Poljak and Ya. Z. Tsypkin, Pseudogradient adaptation and training algorithms, Automation and Remote Control 34 (1973) 377–397.
- H. Robbins and S. Monro, A Stochastic Approximation Method, Annals of Mathematical Statistics 22 (1951) 400–407. https://doi.org/10.1214/aoms/1177729586