A General-Purpose Theorem for High-Probability Bounds of Stochastic Approximation with Polyak Averaging
Sajad Khodadadian, Martin Zubeldia
Abstract
Polyak-Ruppert averaging is a widely used technique to achieve the optimal asymptotic variance of stochastic approximation (SA) algorithms, yet its high-probability performance guarantees remain underexplored in general settings. In this paper, we present a general framework for establishing non-asymptotic concentration bounds for the error of averaged SA iterates. Our approach assumes access to individual concentration bounds for the unaveraged iterates and yields a sharp bound on the averaged iterates. We also construct an example, showing the tightness of our result up to constant multiplicative factors. As direct applications, we derive tight concentration bounds for contractive SA algorithms and for algorithms such as temporal difference learning and Q-learning with averaging, obtaining new bounds in settings where traditional analysis is challenging.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on6
- Sample Complexity of Asynchronous Q-Learning: Sharper Analysis and Variance ReductionGen Li, Yuting Wei, Yuejie Chi, Yuantao Gu et al.NeurIPS 2020 · 149 citations
- Finite-Sample Analysis of Contractive Stochastic Approximation Using Smooth Convex EnvelopesZaiwei Chen, Siva Theja Maguluri, Sanjay Shakkottai, Karthikeyan ShanmugamNeurIPS 2020 · 66 citations
- Adapting to Mixing Time in Stochastic Optimization with Markovian DataRon Dorfman, Kfir Yehuda LevyICML 2022 · 41 citations
- Tight High Probability Bounds for Linear Stochastic Approximation with Fixed StepsizeAlain Durmus, Eric Moulines, Alexey Naumov, Sergey Samsonov et al.NeurIPS 2021 · 36 citations
- Tightening the Dependence on Horizon in the Sample Complexity of Q-LearningGen Li, Changxiao Cai, Yuxin Chen, Yuantao Gu et al.ICML 2021 · 19 citations
Related papers
- Gaussian Approximation and Multiplier Bootstrap for Polyak-Ruppert Averaged Linear Stochastic Approximation with Applications to TD LearningSergey Samsonov, Eric Moulines, Qi-Man Shao, Zhuo-Song Zhang et al.NeurIPS 2024 · 23 citations
- High-Order Error Bounds for Markovian LSA with Richardson-Romberg ExtrapolationIlya Levin, Alexey Naumov, Sergey SamsonovAAAI 2026
- Gaussian Approximation for Two-Timescale Linear Stochastic ApproximationBogdan Butyrin, Artemy Rubtsov, Alexey Naumov, Vladimir V. Ulyanov et al.AAAI 2026 · 2 citations
- Statistical inference for Linear Stochastic Approximation with Markovian NoiseSergey Samsonov, Marina Sheshukova, Eric Moulines, Alexey NaumovNeurIPS 2025 · 12 citations
- Parameter-free Optimal Rates for Nonlinear Semi-Norm Contractions with Applications to Q-LearningAnkur Naskar, Gugan Thoppe, Vijay GuptaAAAI 2026 · 1 citation
