The O.D.E. Method for Convergence of Stochastic Approximation and Reinforcement Learning II: Mean-Square Error under Bounded StepsizesResearch Paper
Motivation
Stochastic approximation is the family of recursive algorithms that locate a zero of a vector field from noisy evaluations of it. Temporal-difference learning, Q-learning, actor–critic methods and stochastic gradient descent are all instances. In practice these algorithms are often run with a constant or bounded, non-vanishing stepsize: the iterate keeps adapting to new data and never freezes, at the price of never converging exactly. The natural question for such a scheme is quantitative: how far from the target does the iterate stay in the long run, and how does that distance scale with the stepsize?
Borkar and Meyn (SIAM J. Control Optim. 38(2), 2000) answer both halves of the question under one set of hypotheses. First, stability of a "fluid" ODE obtained by scaling at infinity implies that the iterates have bounded second moments, with no a priori boundedness or projection assumption. Second, if the ODE has a globally exponentially stable equilibrium , the asymptotic mean-square error is of the order of the largest stepsize. The first result replaced the usual stochastic-Lyapunov-function verification in reinforcement-learning applications (Section 3 of the paper). This mission formalizes the bounded-stepsize branch of the paper. A companion mission (… I: Stability and Almost-Sure Convergence under Tapering Stepsizes) covers the vanishing-stepsize branch.
Setting
Fix and a Lipschitz vector field . The stochastic approximation recursion (1.1) is
with a deterministic step sequence and a noise sequence on a probability space . The associated ODE (1.2) is .
- The scaled fields are . Assumption (A1) asks that be Lipschitz, that for every as , and that the origin be an asymptotically stable equilibrium of the fluid ODE .
- Let . Assumption (A2) asks that be a martingale difference sequence, , with conditional second moments for a constant .
- Assumption (BS), bounded stepsizes: for all , with .
- The error (2.2) is .
An equilibrium of (1.2) is globally asymptotically stable if it is Lyapunov stable and attracts every solution. It is globally exponentially asymptotically stable if there are and with for every solution.
The proofs compare the iterates with ODE solutions on a time grid: , blocks of length about , the interpolated path of the iterates, its rescaled version with , and the ODE solutions , restarted at each block.
Formalization targets
Goal: Theorem 2.3(ii)
Under (A1), (A2) and (BS), if is a globally asymptotically and globally exponentially asymptotically stable equilibrium of (1.2), there are and such that for all and every initial condition ,
The constant is uniform in the stepsizes and in . No rate constant is fixed: only the linear dependence on is asserted.
Milestones
- Lemma 4.7. For fixed there is , independent of the stepsizes, with and on every block.
- Theorem 2.1(ii). There are and with whenever .
- Lemma 4.8. For , .
Significance
The result. Theorem 2.1(ii) says that a stability property of a deterministic ODE at infinity controls a stochastic recursion in mean square, uniformly over initial conditions. Theorem 2.3(ii) turns this into an error bound: with a non-vanishing stepsize the iterate does not converge, but its mean-square distance from is eventually . This is the quantitative justification for constant-stepsize stochastic approximation in reinforcement learning and adaptive control, and it is the starting point of the trade-off between bias (small stepsize) and speed (large stepsize) discussed in Section 2.2 of the paper.
Formalizing it. The results are proved in the paper, partly by sketch: Lemma 4.8's proof refers to "familiar arguments using the Bellman–Gronwall lemma", and the paper writes several bounds as . A formal proof makes every constant and its dependence explicit, which the paper's quantifier order leaves partly implicit (see the scope section). No machine-checked proof of these results, or of any stochastic-approximation stability theorem in this generality, is known to exist.
Difficulty
The natural first argument, that the iterates track the ODE and the ODE converges, needs the iterates to be bounded. Bounded iterates are exactly what is being proved, so the argument is circular. The paper breaks the circularity by rescaling: on each block the iterate is divided by , so that on large scales it tracks the fluid ODE, whose stability contracts the norm by a fixed factor per block. Making this work in mean square requires conditional second-moment estimates that are uniform in the stepsize and the scale (Lemma 4.7). A second difficulty is that the stepsize does not vanish: the tracking error per block does not go to zero, and the goal's content is precisely that it is of order with a constant that does not depend on .
Formalization scope
- The state space is
EuclideanSpace ℝ (Fin d). ODE solutions are forward solutions: continuous on with right derivatives. Stability notions are the standard ones, with exponential stability in the form the proof of Theorem 2.3(ii) uses. - The filtration is the natural filtration of the iterates. (A2) includes integrability of and so that the conditional expectations are meaningful. Each theorem quantifies over every probability space, every measurable process satisfying (1.1) and (A2), and every deterministic .
- Second moments, suprema and are computed in , so a bound is a genuine finiteness claim.
- Constant placement. , and depend only on , , (and ). depends in addition on , and on and . None depends on the stepsizes. Theorem 2.3's page text puts after . Read literally, that would allow and make the goal a restatement of Theorem 2.1(ii). The formal goal fixes before the stepsizes and the initial condition, as the proof () gives. is existential in Theorem 2.3(ii) and Lemma 4.8 rather than tied to Theorem 2.1(ii)'s witness.
- The proof objects , are characterized by predicates (initial value, continuity, ODE on the block). Lemmas 4.7 and 4.8 hold for every function satisfying them.
- Not included: Theorem 2.3(i) (its proof chooses a radius depending on ), Theorem 2.4 and the Markov-chain results of Section 4.3, and the asynchronous extension (Theorem 2.5). The deterministic ODE lemmas 4.1–4.4 belong to the companion mission.
Useful infrastructure, reusable beyond this mission: Bellman–Gronwall inequalities in discrete time (Lemma 4.3 of the paper), stability of scaled ODEs, and conditional second-moment estimates for martingale-difference-driven recursions. Contributions of any of these as separate lemmas are welcome.
Selected references
- V. S. Borkar and S. P. Meyn, The O.D.E. Method for Convergence of Stochastic Approximation and Reinforcement Learning, SIAM J. Control Optim. 38(2):447–469, 2000. https://doi.org/10.1137/S0363012997331639
- M. Benaïm, Dynamics of stochastic approximation algorithms, Séminaire de Probabilités XXXIII, Lecture Notes in Math. 1709, Springer, 1999. https://doi.org/10.1007/BFb0096509
- H. J. Kushner and G. G. Yin, Stochastic Approximation and Recursive Algorithms and Applications, 2nd ed., Springer, 2003. https://doi.org/10.1007/b97441