cs.ITJun 12, 2026

Nonlinear Two-Time-Scale Stochastic Approximation: A Sharp Phase Transition and How to Beat It

Authors: Dhruv SarkarVaneet Aggarwal

Organizations: 1Indian Institute of Technology Kharagpur · 2Mohamed bin Zayed University of Artificial Intelligence · 3Purdue University

Abstract

Recent finite-time analyses of nonlinear two-time-scale stochastic approximation show that under contractive assumptions the slow iterate YkY_k with stepsizes βk=Θ(k1)β_k=Θ(k^{-1}) and αk=Θ(ka)α_k=Θ(k^{-a}), a(1/2,1)a\in(1/2,1), generally satisfies a mean-square rate of order kak^{-a}; decoupled k1k^{-1} rates require strong local linearity. We identify a sharp regularity-dependent boundary. In a rate-determining normal form where the slow drift contains a locally linear leakage and a nonlinear remainder of order 1+ρ1+ρ (ρ[0,1]ρ\in[0,1]), the uncorrected recursion satisfies

EYk2C(k1+ka(1+ρ)),\mathbb{E}\|Y_k\|^2 \le C\bigl(k^{-1}+k^{-a(1+ρ)}\bigr),

and a matching scalar Gaussian lower bound shows that the slower term is unavoidable without modifying the update. Thus the decoupled k1k^{-1} rate is guaranteed for the uncorrected recursion exactly when a(1+ρ)1a(1+ρ)\ge 1. This lower bound concerns only the naive update; it is not an information-theoretic obstruction. We demonstrate this by equipping the normal-form recursion with an auxiliary online bias estimator

Mk+1=Mk+γk(R(Xk)Mk),βkγkαk,M_{k+1}=M_k+γ_k(R(X_k)-M_k),\qquad β_k\llγ_k\llα_k,

and subtracting MkM_k from the slow update. Under the same stability, moment, and remainder assumptions, the corrected recursion achieves EY~k2=O(k1)\mathbb{E}\|\widetilde Y_k\|^2=O(k^{-1}) for every ρ[0,1]ρ\in[0,1], including regimes where the uncorrected update provably suffers the slower rate. Finally, we prove localized transfer theorems that extend the phase-transition mechanism to general nonlinear TTSA in fast-manifold coordinates. The proofs are non-asymptotic and rely on two Abel-transform cancellations: one for the locally linear fast-error leakage, and one for the tracked nonlinear bias.

Explore similar work

Sep 17, 2026cs.LG

The Bias of Nonlinear Two-Time-scale Stochastic Approximation under Constant Step-Sizes

Two-timescale stochastic approximation (TTSA) is a fundamental tool for analyzing coupled iterative algorithms in reinforcement learning, optimization, and stochastic control. However, finite-time guarantees for nonlinear two-timescale schemes remain difficult to obtain, especially under constant step-sizes. In this paper, we study nonlinear TTSA with step-sizes αβα\ggβ. Under standard stability, regularity, and Markovian noise assumptions, we upper bound the mean-squared error and the bias of both iterates around their limiting equilibria. Our bounds scale as O(α+β2/α2)O(α+β^2/α^2), which we prove to be tight when βα3/2β\leα^{3/2}. The analysis separates the contributions of initial conditions, fast-timescale tracking error, Markovian dependence, and timescale coupling, thereby clarifying the origin of the β2/α2β^2/α^2 term. Our results reveal qualitative differences from the linear TTSA setting previously studied, showing that nonlinear dynamics introduce additional finite-time effects that are absent in the linear case.
Djamel Rassem Lamouri, Dorian Baudry, Nicolas Gast
Jul 15, 2026stat.ML

Non-Expansive Two-Time-Scale Stochastic Approximation: A Fixed-Schedule One-Quarter Barrier and Bias-Corrected Acceleration

Non-expansive two-time-scale stochastic approximation is governed by a slow stochastic Krasnoselskii--Mann fixed-point iteration rather than by contraction to a unique equilibrium. We study this regime under a contractive fast map and a non-expansive reduced slow map. We first prove a finite-horizon lower bound showing that, for any prescribed slow stepsize schedule (βk)(β_k), the classical KM residual scale (i<Nβi(1βi))1(\sum_{i<N}β_i(1-β_i))^{-1} is worst-case sharp for the corresponding unregularized KM update. Combined with the raw fast-tracking leakage scale, this explains the previously observed k1/4+o(1)k^{-1/4+o(1)} last-iterate mean-square residual exponent. We then introduce a residual-preconditioned slow oracle that cancels the first-order dependence on the fast tracking error. In a nested Tikhonov-KM algorithm, the uncorrected oracle yields total-sample rate T1/4+o(1)T^{-1/4+o(1)}, while the corrected oracle yields T1/3+o(1)T^{-1/3+o(1)}. This improvement comes from changing the slow-oracle bias from first order to second order in the fast error after all inner-loop samples are counted. Finally, we show that the repeated inner-loop cost of the nested method can be avoided in a smooth derivative-oracle model. A single-loop algorithm that tracks both the fast equilibrium and the leakage preconditioner online achieves T1/2+o(1)T^{-1/2+o(1)} with O(1)O(1) primitive samples per iteration.
Dhruv Sarkar, Vaneet Aggarwal
Feb 14, 2025cs.LG

Nonasymptotic CLT and Error Bounds for Linear Two-Time-Scale Stochastic Approximation

We consider linear two-time-scale stochastic approximation algorithms driven by martingale noise. Recent applications in machine learning motivate the need to understand finite-time error rates, but conventional stochastic approximation analyses focus on either asymptotic convergence in distribution or finite-time bounds that are far from optimal. Prior work on asymptotic central limit theorems (CLTs) suggests that two-time-scale algorithms may be able to achieve 1/K1/\sqrt{K} error in expectation, with a constant given by the expected norm of the limiting Gaussian vector. However, the best known finite-time rates are much slower. We derive the first nonasymptotic Wasserstein-1 CLT for linear two-time-scale stochastic approximation with Polyak-Ruppert averaging driven by martingale difference noise. As a corollary, we show that the expected error achieved by Polyak-Ruppert averaging decays at rate 1/K1/\sqrt{K}, which significantly improves on the rates of convergence in prior works.
Seo Taek Kong, Sihan Zeng, Thinh T. Doan +1