Optimal and instance-dependent guarantees for Markovian linear stochastic approximation
Abstract
We study stochastic approximation procedures for approximately solving a -dimensional linear fixed point equation based on observing a trajectory of length from an ergodic Markov chain. We first exhibit a non-asymptotic bound of the order on the squared error of the last iterate of a standard scheme, where is a mixing time. We then prove a non-asymptotic instance-dependent bound on a suitably averaged sequence of iterates, with a leading term that matches the local asymptotic minimax limit, including sharp dependence on the parameters in the higher order terms. We complement these upper bounds with a non-asymptotic minimax lower bound that establishes the instance-optimality of the averaged SA estimator. We derive corollaries of these results for policy evaluation with Markov noise -- covering the TD() family of algorithms for all -- and linear autoregressive models. Our instance-dependent characterizations open the door to the design of fine-grained model selection procedures for hyperparameter tuning (e.g., choosing the value of when running the TD() algorithm).
Cite
@article{arxiv.2112.12770,
title = {Optimal and instance-dependent guarantees for Markovian linear stochastic approximation},
author = {Wenlong Mou and Ashwin Pananjady and Martin J. Wainwright and Peter L. Bartlett},
journal= {arXiv preprint arXiv:2112.12770},
year = {2024}
}
Comments
Published at Mathematical Statistics and Learning