Federated Q-Learning with Reference-Advantage Decomposition: Almost Optimal Regret and Logarithmic Communication Cost
Zhong Zheng, Haochen Zhang, Lingzhou Xue
Abstract
In this paper, we consider model-free federated reinforcement learning for tabular episodic Markov decision processes. Under the coordination of a central server, multiple agents collaboratively explore the environment and learn an optimal policy without sharing their raw data. Despite recent advances in federated Q-learning algorithms achieving near-linear regret speedup with low communication cost, existing algorithms only attain suboptimal regrets compared to the information bound. We propose a novel model-free federated Q-learning algorithm, termed FedQ-Advantage. Our algorithm leverages reference-advantage decomposition for variance reduction and adopts three novel designs: separate event-triggered communication and policy switching, heterogeneous communication triggering conditions, and optional forced synchronization. We prove that our algorithm not only requires a lower logarithmic communication cost but also achieves an almost optimal regret, reaching the information bound up to a logarithmic factor and near-linear regret speedup compared to its single-agent counterpart when the time horizon is sufficiently large.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c62ff31d-9256-4e7b-bcbf-858d877056c3Cited by top-tier papers7
- In-Trajectory Inverse Reinforcement Learning: Learn Incrementally Before an Ongoing Trajectory TerminatesShicheng Liu, Minghui ZhuNeurIPS 2024 · 11 citations
- Regret-Optimal Q-Learning with Low Cost for Single-Agent and Federated Reinforcement LearningHaochen Zhang, Zhong Zheng, Lingzhou XueNeurIPS 2025 · 3 citations
- Q-Learning with Fine-Grained Gap-Dependent RegretHaochen Zhang, Zhong Zheng, Lingzhou XueICLR 2026 · 3 citations
- Explainable Reinforcement Learning from Human Feedback to Improve AlignmentShicheng Liu, Siyuan Xu, Wenjie Qiu, Hangfan Zhang et al.NeurIPS 2025 · 2 citations
- Gap-Dependent Bounds for Q-Learning using Reference-Advantage DecompositionZhong Zheng, Haochen Zhang, Lingzhou XueICLR 2025
Builds on27
- 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
- Near-Optimal Offline Reinforcement Learning via Double Variance ReductionMing Yin, Yu Bai, Yu-Xiang WangNeurIPS 2021 · 72 citations
Related papers
- Federated Q-Learning: Linear Regret Speedup with Low Communication CostZhong Zheng, Fengyu Gao, Lingzhou Xue, Jing YangICLR 2024 · 21 citations
- Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage SufficesJiin Woo, Laixi Shi, Gauri Joshi, Yuejie ChiICML 2024 · 9 citations
- The Sample-Communication Complexity Trade-off in Federated Q-LearningSudeep Salgia, Yuejie ChiNeurIPS 2024 · 10 citations
- The Blessing of Heterogeneity in Federated Q-Learning: Linear Speedup and BeyondJiin Woo, Gauri Joshi, Yuejie ChiICML 2023 · 36 citations
- Gap-Dependent Bounds for Federated Q-LearningHaochen Zhang, Zhong Zheng, Lingzhou XueICML 2025
