Optimal and instance-dependent guarantees for Markovian linear stochastic approximation
Wenlong Mou, Ashwin Pananjady, Martin J. Wainwright, Peter L. Bartlett
TL;DR
This work provides non-asymptotic, instance-dependent guarantees for Markovian linear stochastic approximation, showing that averaged SA with Polyak–Ruppert averaging achieves an optimal $O\left( \tfrac{t_{\text{mix}} d}{n} \right)$-type rate up to polylog factors, while detailing how the local structure of the problem governs higher-order terms. It derives a local minimax lower bound and demonstrates that the averaged estimator attains instance-optimal performance, highlighting the practical relevance for hyperparameter tuning in TD$(\lambda)$ and related linear Z-estimation problems. The results apply to TD$\{0,\lambda\}$, sieve estimators, and autoregressive models, providing a principled framework for adaptive model selection and parameter tuning in Markovian settings. The paper also introduces bootstrapping techniques and surrogate processes to achieve sharp dimension-dependent bounds, contributing new analytic tools for stochastic approximation with dependent data. Overall, the findings offer both theoretical tightness and actionable guidance for policy evaluation and time-series estimation in stochastic, Markov-driven environments.
Abstract
We study stochastic approximation procedures for approximately solving a $d$-dimensional linear fixed point equation based on observing a trajectory of length $n$ from an ergodic Markov chain. We first exhibit a non-asymptotic bound of the order $t_{\mathrm{mix}} \tfrac{d}{n}$ on the squared error of the last iterate of a standard scheme, where $t_{\mathrm{mix}}$ 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 $(d, t_{\mathrm{mix}})$ 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 $λ\in [0, 1)$ -- 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).
