Two-Timescale Linear Stochastic Approximation: Constant Stepsizes Go a Long Way

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kwon, Jeongyeol, Dotson, Luke, Chen, Yudong, Xie, Qiaomin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913702906691584
author Kwon, Jeongyeol
Dotson, Luke
Chen, Yudong
Xie, Qiaomin
author_facet Kwon, Jeongyeol
Dotson, Luke
Chen, Yudong
Xie, Qiaomin
contents Previous studies on two-timescale stochastic approximation (SA) mainly focused on bounding mean-squared errors under diminishing stepsize schemes. In this work, we investigate {\it constant} stpesize schemes through the lens of Markov processes, proving that the iterates of both timescales converge to a unique joint stationary distribution in Wasserstein metric. We derive explicit geometric and non-asymptotic convergence rates, as well as the variance and bias introduced by constant stepsizes in the presence of Markovian noise. Specifically, with two constant stepsizes $α< β$, we show that the biases scale linearly with both stepsizes as $Θ(α)+Θ(β)$ up to higher-order terms, while the variance of the slower iterate (resp., faster iterate) scales only with its own stepsize as $O(α)$ (resp., $O(β)$). Unlike previous work, our results require no additional assumptions such as $β^2 \ll α$ nor extra dependence on dimensions. These fine-grained characterizations allow tail-averaging and extrapolation techniques to reduce variance and bias, improving mean-squared error bound to $O(β^4 + \frac{1}{t})$ for both iterates.
format Preprint
id arxiv_https___arxiv_org_abs_2410_13067
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Two-Timescale Linear Stochastic Approximation: Constant Stepsizes Go a Long Way
Kwon, Jeongyeol
Dotson, Luke
Chen, Yudong
Xie, Qiaomin
Systems and Control
Machine Learning
Optimization and Control
Previous studies on two-timescale stochastic approximation (SA) mainly focused on bounding mean-squared errors under diminishing stepsize schemes. In this work, we investigate {\it constant} stpesize schemes through the lens of Markov processes, proving that the iterates of both timescales converge to a unique joint stationary distribution in Wasserstein metric. We derive explicit geometric and non-asymptotic convergence rates, as well as the variance and bias introduced by constant stepsizes in the presence of Markovian noise. Specifically, with two constant stepsizes $α< β$, we show that the biases scale linearly with both stepsizes as $Θ(α)+Θ(β)$ up to higher-order terms, while the variance of the slower iterate (resp., faster iterate) scales only with its own stepsize as $O(α)$ (resp., $O(β)$). Unlike previous work, our results require no additional assumptions such as $β^2 \ll α$ nor extra dependence on dimensions. These fine-grained characterizations allow tail-averaging and extrapolation techniques to reduce variance and bias, improving mean-squared error bound to $O(β^4 + \frac{1}{t})$ for both iterates.
title Two-Timescale Linear Stochastic Approximation: Constant Stepsizes Go a Long Way
topic Systems and Control
Machine Learning
Optimization and Control
url https://arxiv.org/abs/2410.13067