Regret-Optimal Q-Learning with Low Cost for Single-Agent and Federated Reinforcement Learning
Haochen Zhang, Zhong Zheng, Lingzhou Xue
Abstract
Motivated by real-world settings where data collection and policy deployment -- whether for a single agent or across multiple agents -- are costly, we study the problem of on-policy single-agent reinforcement learning (RL) and federated RL (FRL) with a focus on minimizing burn-in costs (the sample sizes needed to reach near-optimal regret) and policy switching or communication costs. In parallel finite-horizon episodic Markov Decision Processes (MDPs) with states and actions, existing methods either require superlinear burn-in costs in and or fail to achieve logarithmic switching or communication costs. We propose two novel model-free RL algorithms -- Q-EarlySettled-LowCost and FedQ-EarlySettled-LowCost -- that are the first in the literature to simultaneously achieve: (i) the best near-optimal regret among all known model-free RL or FRL algorithms, (ii) low burn-in cost that scales linearly with and , and (iii) logarithmic policy switching cost for single-agent RL or communication cost for FRL. Additionally, we establish gap-dependent theoretical guarantees for both regret and switching/communication costs, improving or matching the best-known gap-dependent bounds.
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.
Cited by top-tier papers1
Ask how each one uses itBuilds on42
- Almost Optimal Model-Free Reinforcement Learningvia Reference-Advantage DecompositionZihan Zhang, Yuan Zhou, Xiangyang JiNeurIPS 2020 · 183 citations
- Provably Efficient Reinforcement Learning with Linear Function Approximation under Adaptivity ConstraintsTianhao Wang, Dongruo Zhou, Quanquan GuNeurIPS 2021 · 169 citations
- Sample Complexity of Asynchronous Q-Learning: Sharper Analysis and Variance ReductionGen Li, Yuting Wei, Yuejie Chi, Yuantao Gu et al.NeurIPS 2020 · 149 citations
- Pessimistic Q-Learning for Offline Reinforcement Learning: Towards Optimal Sample ComplexityLaixi Shi, Gen Li, Yuting Wei, Yuxin Chen et al.ICML 2022 · 110 citations
- Logarithmic Regret for Reinforcement Learning with Linear Function ApproximationJiafan He, Dongruo Zhou, Quanquan GuICML 2021 · 108 citations
Related papers
- Federated Q-Learning with Reference-Advantage Decomposition: Almost Optimal Regret and Logarithmic Communication CostZhong Zheng, Haochen Zhang, Lingzhou XueICLR 2025
- Federated Q-Learning: Linear Regret Speedup with Low Communication CostZhong Zheng, Fengyu Gao, Lingzhou Xue, Jing YangICLR 2024 · 21 citations
- Sample-Efficient Reinforcement Learning with loglog(T) Switching CostDan Qiao, Ming Yin, Ming Min, Yu-Xiang WangICML 2022 · 35 citations
- Breaking the Sample Complexity Barrier to Regret-Optimal Model-Free Reinforcement LearningGen Li, Laixi Shi, Yuxin Chen, Yuantao Gu et al.NeurIPS 2021 · 71 citations
- Regret-Optimal Model-Free Reinforcement Learning for Discounted MDPs with Short Burn-In TimeXiang Ji, Gen LiNeurIPS 2023 · 11 citations
