machine learning

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

arXiv:2607.13414

summary

The paper analyzes two‑time‑scale stochastic approximation with a non‑expansive slow map, establishes sharp lower bounds on residual decay, and proposes bias‑corrected and single‑loop algorithms that improve convergence rates up to a near‑optimal .

Abstract

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 , the classical KM residual scale is worst-case sharp for the corresponding unregularized KM update. Combined with the raw fast-tracking leakage scale, this explains the previously observed 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 , while the corrected oracle yields . 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 with primitive samples per iteration.

Topics & keywords

#stochastic approximation#two-time-scale algorithms#non-expansive operators#convergence rates#bias correction#accelerated methodsKrasnoselskii-Mann iterationfast-slow dynamicsTikhonov regularizationresidual preconditioningsample complexity
Non-Expansive Two-Time-Scale Stochastic Approximation: A Fixed-Schedule One-Quarter Barrier and Bias-Corrected Acceleration · wovepaper