Sharp asymptotic theory for Q-learning with LD2Z learning rate and its generalization
Soham Bonnerjee, Zhipeng Lou, Wei Biao Wu
摘要
Despite the sustained popularity of Q-learning as a practical tool for policy determination, a majority of relevant theoretical literature deals with either constant () or polynomially decaying () learning schedules. However, it is well known the these choices suffer from either persistent bias or prohibitively slow convergence. In contrast, the recently proposed linear decay to zero (LD2Z: ) schedule has shown appreciable empirical performance, but its theoretical and statistical properties remain largely unexplored, especially in the Q-learning setting. We address this gap in the literature by first considering a general class of power-law decay to zero (PD2Z-: ). Proceeding step-by-step, we present a sharp non-asymptotic error bound for Q-learning with PD2Z- schedule, which then is used to derive a central limit theory for a new tail Polyak-Ruppert averaging estimator. Finally, we also provide a novel time-uniform Gaussian approximation (also known as strong invariance principle) for the partial sum process of Q-learning iterates, which facilitates bootstrap-based inference. All our theoretical results are complemented by extensive numerical experiments. Beyond being new theoretical and statistical contributions to the Q-learning literature, our results definitively establish that LD2Z and in general PD2Z- achieve a best-of-both-worlds property: they inherit the rapid decay from initialization (characteristic of constant step-sizes) while retaining the asymptotic convergence guarantees (characteristic of polynomially decaying schedules). This dual advantage explains the empirical success of LD2Z while providing practical guidelines for inference through our results.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper13
- An empirical analysis of compute-optimal large language model trainingJordan Hoffmann, Sebastian Borgeaud, Arthur Mensch, Elena Buchatskaya 等NeurIPS 2022 · 被引用 566 次
- Scaling Laws and Compute-Optimal Training Beyond Fixed Training DurationsAlexander Hägele, Elie Bakouch, Atli Kosson, Loubna Ben Allal 等NeurIPS 2024 · 被引用 168 次
- Sample Complexity of Asynchronous Q-Learning: Sharper Analysis and Variance ReductionGen Li, Yuting Wei, Yuejie Chi, Yuantao Gu 等NeurIPS 2020 · 被引用 149 次
- Pessimistic Q-Learning for Offline Reinforcement Learning: Towards Optimal Sample ComplexityLaixi Shi, Gen Li, Yuting Wei, Yuxin Chen 等ICML 2022 · 被引用 110 次
- Finite-Sample Analysis of Contractive Stochastic Approximation Using Smooth Convex EnvelopesZaiwei Chen, Siva Theja Maguluri, Sanjay Shakkottai, Karthikeyan ShanmugamNeurIPS 2020 · 被引用 66 次
相关 Paper
- Parameter-free Optimal Rates for Nonlinear Semi-Norm Contractions with Applications to Q-LearningAnkur Naskar, Gugan Thoppe, Vijay GuptaAAAI 2026 · 被引用 1 次
- 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 等NeurIPS 2024 · 被引用 23 次
- Towards Parameter-Free Temporal Difference LearningYunxiang LI, Mark Schmidt, Reza Babanezhad, Sharan VaswaniICML 2026 · 被引用 2 次
- Statistical inference for Linear Stochastic Approximation with Markovian NoiseSergey Samsonov, Marina Sheshukova, Eric Moulines, Alexey NaumovNeurIPS 2025 · 被引用 12 次
- High-Order Error Bounds for Markovian LSA with Richardson-Romberg ExtrapolationIlya Levin, Alexey Naumov, Sergey SamsonovAAAI 2026
