Improved Communication Efficiency in Federated Natural Policy Gradient via ADMM-based Gradient Updates
Not on the map. Why?
Abstract (arXiv)
Federated reinforcement learning (FedRL) enables agents to collaboratively train a global policy without sharing their individual data. However, high communication overhead remains a critical bottleneck, particularly for natural policy gradient (NPG) methods, which are second-order. To address this issue, we propose the FedNPG-ADMM framework, which leverages the alternating direction method of multipliers (ADMM) to approximate global NPG directions efficiently. We theoretically demonstrate that using ADMM-based gradient updates reduces communication complexity from O(d2) to O(d) at each iteration, where d is the number of model parameters. Furthermore, we show that achieving an ε-error stationary convergence requires O(frac1(1-γ)2ε) iterations for discount factor γ, demonstrating that FedNPG-ADMM maintains the same convergence rate as the standard FedNPG. Through evaluation of the proposed algorithms in MuJoCo environments, we demonstrate that FedNPG-ADMM maintains the reward performance of standard FedNPG, and that its convergence rate improves when the number of federated agents increases.
Key facts. Cited by 21 papers in the map (data through 14 Oct 2026). Builds on 36 of its 51 references that are in the map. First posted on arXiv 9 Oct 2023.
Builds on
36 of 51 references.
References · 36
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2023GPT-4 Technical Reportcited by 18,396
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2019Advances and Open Problems in Federated Learningcited by 3,046
- 2015Trust Region Policy Optimizationcited by 2,580
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 2,026
- 2016Safe, Multi-Agent, Reinforcement Learning for Autonomous Drivingcited by 311
- 2021Federated Learning for Internet of Things: A Comprehensive Surveycited by 252
- 2020Minibatch vs Local SGD for Heterogeneous Distributed Learningcited by 147
24 more
- 2013Communication Efficient Distributed Optimization using an Approximate Newton-type Methodcited by 132
- 2019Federated Learning for Wireless Communications: Motivation, Opportunities and Challengescited by 129
- 2018Stochastic Variance-Reduced Policy Gradientcited by 106
- 2017Towards Generalization and Simplicity in Continuous Controlcited by 68
- 2022An Improved Analysis of (Variance-Reduced) Policy Gradient and Natural Policy Gradient Methodscited by 68
- 2022Federated Reinforcement Learning with Environment Heterogeneitycited by 63
- 2019Sample Efficient Policy Gradient Methods with Recursive Variance Reductioncited by 60
- 2019DeepPool: Distributed Model-free Algorithm for Ride-sharing using Deep Reinforcement Learningcited by 54
- 2021FedNL: Making Newton-Type Methods Applicable to Federated Learningcited by 52
- 2022FedADMM: A Federated Primal-Dual Algorithm Allowing Partial Participationcited by 36
- 2019GADMM: Fast and Communication Efficient Framework for Distributed Machine Learningcited by 30
- 2022FedKL: Tackling Data Heterogeneity in Federated Reinforcement Learning by Penalizing KL Divergencecited by 29
- 2022Federated Stochastic Approximation under Markov Noise and Heterogeneity: Applications in Reinforcement Learningcited by 27
- 2023Federated Temporal Difference Learning with Linear Function Approximation under Environmental Heterogeneitycited by 27
- 2021Distributed Second Order Methods with Fast Rates and Compressed Communicationcited by 25
- 2017Bandit Convex Optimization for Scalable and Dynamic IoT Managementcited by 24
- 2022FedNew: A Communication-Efficient and Privacy-Preserving Newton-Type Method for Federated Learningcited by 23
- 2019Q-GADMM: Quantized Group ADMM for Communication Efficient Decentralized Machine Learningcited by 17
- 2020A Distributed Model-Free Ride-Sharing Approach for Joint Matching, Pricing, and Dispatching using Deep Reinforcement Learningcited by 17
- 2022FedSysID: A Federated Approach to Sample-Efficient System Identificationcited by 17
- 2016Convergence Rate of Distributed ADMM over Networkscited by 16
- 2021Basis Matters: Better Communication-Efficient Second Order Methods for Federated Learningcited by 15
- 2021Communication Efficient Parallel Reinforcement Learningcited by 10
- 2021ElegantRL-Podracer: Scalable and Elastic Library for Cloud-Native Deep Reinforcement Learningcited by 9
Built on by
Citing papers per year: 2023: 1; 2024: 14; 2025: 5; 2026: 1.
21 papers.
Sort:
By area:
- 2024Credit Card Fraud Detection Using Advanced Transformer Modelcited by 26
- 2024Adaptive Ensembles of Fine-Tuned Transformers for LLM-Generated Text Detectioncited by 22
- 2024Asynchronous Federated Reinforcement Learning with Policy Gradient Updates: Algorithm Design and Convergence Analysiscited by 21
- 2024Real-Time Pill Identification for the Visually Impaired Using Deep Learningcited by 21
- 2024Efficiency optimization of large-scale language models based on deep learning in natural language processing taskscited by 14
- 2024Exploration of Multi-Scale Image Fusion Systems in Intelligent Medical Image Analysiscited by 10
- 2024Momentum for the Win: Collaborative Federated Reinforcement Learning across Heterogeneous Environmentscited by 8
- 2024Advanced Payment Security System:XGBoost, LightGBM and SMOTE Integratedcited by 8
- 2024Advancements in Feature Extraction Recognition of Medical Imaging Systems Through Deep Learning Techniquecited by 7
- 2024In-Trajectory Inverse Reinforcement Learning: Learn Incrementally Before An Ongoing Trajectory Terminatescited by 7
- 2025One-Shot Federated Learning with Classifier-Free Diffusion Modelscited by 7
- 2024Global Convergence Guarantees for Federated Policy Gradient Methods with Adversariescited by 6
9 more
- 2024Advanced Multimodal Deep Learning Architecture for Image-Text Matchingcited by 4
- 2024Towards Fast Rates for Federated and Multi-Task Reinforcement Learningcited by 4
- 2025Gap-Dependent Bounds for Federated Q-learningcited by 4
- 2025Achieving Tighter Finite-Time Rates for Heterogeneous Federated Stochastic Approximation under Markovian Samplingcited by 2
- 2025Regret-Optimal Q-Learning with Low Cost for Single-Agent and Federated Reinforcement Learningcited by 2
- 2025Convergence Guarantees for Federated SARSA with Local Training and Heterogeneous Agentscited by 2
- 2023Communication Efficient and Privacy-Preserving Federated Learning Based on Evolution Strategiescited by 0
- 2024StraightLine: An End-to-End Resource-Aware Scheduler for Machine Learning Application Requestscited by 0
- 2026Learning from Unreliable Trajectories: Adversarially-Robust Federated Q-Learningcited by 0
- 2026Learning from Unreliable Trajectories: Adversarially-Robust Federated Q-Learning2 Oct 2026
- 2025Convergence Guarantees for Federated SARSA with Local Training and Heterogeneous Agents19 Dec 2025
- 2025Regret-Optimal Q-Learning with Low Cost for Single-Agent and Federated Reinforcement Learning5 Jun 2025
- 2025Achieving Tighter Finite-Time Rates for Heterogeneous Federated Stochastic Approximation under Markovian Sampling15 Apr 2025
- 2025One-Shot Federated Learning with Classifier-Free Diffusion Models12 Feb 2025
- 2025Gap-Dependent Bounds for Federated Q-learning5 Feb 2025
- 2024In-Trajectory Inverse Reinforcement Learning: Learn Incrementally Before An Ongoing Trajectory Terminates21 Oct 2024
- 2024Towards Fast Rates for Federated and Multi-Task Reinforcement Learning9 Sep 2024
- 2024StraightLine: An End-to-End Resource-Aware Scheduler for Machine Learning Application Requests25 Jul 2024
- 2024Advanced Multimodal Deep Learning Architecture for Image-Text Matching13 Jun 2024
- 2024Advanced Payment Security System:XGBoost, LightGBM and SMOTE Integrated7 Jun 2024
- 2024Credit Card Fraud Detection Using Advanced Transformer Model6 Jun 2024
9 more
- 2024Momentum for the Win: Collaborative Federated Reinforcement Learning across Heterogeneous Environments29 May 2024
- 2024Advancements in Feature Extraction Recognition of Medical Imaging Systems Through Deep Learning Technique23 May 2024
- 2024Exploration of Multi-Scale Image Fusion Systems in Intelligent Medical Image Analysis23 May 2024
- 2024Efficiency optimization of large-scale language models based on deep learning in natural language processing tasks20 May 2024
- 2024Real-Time Pill Identification for the Visually Impaired Using Deep Learning8 May 2024
- 2024Asynchronous Federated Reinforcement Learning with Policy Gradient Updates: Algorithm Design and Convergence Analysis9 Apr 2024
- 2024Adaptive Ensembles of Fine-Tuned Transformers for LLM-Generated Text Detection20 Mar 2024
- 2024Global Convergence Guarantees for Federated Policy Gradient Methods with Adversaries15 Mar 2024
- 2023Communication Efficient and Privacy-Preserving Federated Learning Based on Evolution Strategies5 Nov 2023
- 2024Asynchronous Federated Reinforcement Learning with Policy Gradient Updates: Algorithm Design and Convergence Analysiscited by 21
- 2024Momentum for the Win: Collaborative Federated Reinforcement Learning across Heterogeneous Environmentscited by 8
- 2024Global Convergence Guarantees for Federated Policy Gradient Methods with Adversariescited by 6
- 2024Towards Fast Rates for Federated and Multi-Task Reinforcement Learningcited by 4
- 2025Achieving Tighter Finite-Time Rates for Heterogeneous Federated Stochastic Approximation under Markovian Samplingcited by 2
- 2025Regret-Optimal Q-Learning with Low Cost for Single-Agent and Federated Reinforcement Learningcited by 2
- 2026Learning from Unreliable Trajectories: Adversarially-Robust Federated Q-Learningcited by 0
- 2026Learning from Unreliable Trajectories: Adversarially-Robust Federated Q-Learning2 Oct 2026
- 2025Regret-Optimal Q-Learning with Low Cost for Single-Agent and Federated Reinforcement Learning5 Jun 2025
- 2025Achieving Tighter Finite-Time Rates for Heterogeneous Federated Stochastic Approximation under Markovian Sampling15 Apr 2025
- 2024Towards Fast Rates for Federated and Multi-Task Reinforcement Learning9 Sep 2024
- 2024Momentum for the Win: Collaborative Federated Reinforcement Learning across Heterogeneous Environments29 May 2024
- 2024Asynchronous Federated Reinforcement Learning with Policy Gradient Updates: Algorithm Design and Convergence Analysis9 Apr 2024
- 2024Global Convergence Guarantees for Federated Policy Gradient Methods with Adversaries15 Mar 2024
- 2024Credit Card Fraud Detection Using Advanced Transformer Modelcited by 26
- 2024Adaptive Ensembles of Fine-Tuned Transformers for LLM-Generated Text Detectioncited by 22
- 2024Real-Time Pill Identification for the Visually Impaired Using Deep Learningcited by 21
- 2024Efficiency optimization of large-scale language models based on deep learning in natural language processing taskscited by 14
- 2024Exploration of Multi-Scale Image Fusion Systems in Intelligent Medical Image Analysiscited by 10
- 2024Advanced Payment Security System:XGBoost, LightGBM and SMOTE Integratedcited by 8
- 2024Advancements in Feature Extraction Recognition of Medical Imaging Systems Through Deep Learning Techniquecited by 7
- 2024Advanced Multimodal Deep Learning Architecture for Image-Text Matchingcited by 4
- 2025Gap-Dependent Bounds for Federated Q-learningcited by 4
- 2025Convergence Guarantees for Federated SARSA with Local Training and Heterogeneous Agentscited by 2
- 2023Communication Efficient and Privacy-Preserving Federated Learning Based on Evolution Strategiescited by 0
- 2024StraightLine: An End-to-End Resource-Aware Scheduler for Machine Learning Application Requestscited by 0
- 2025Convergence Guarantees for Federated SARSA with Local Training and Heterogeneous Agents19 Dec 2025
- 2025Gap-Dependent Bounds for Federated Q-learning5 Feb 2025
- 2024StraightLine: An End-to-End Resource-Aware Scheduler for Machine Learning Application Requests25 Jul 2024
- 2024Advanced Multimodal Deep Learning Architecture for Image-Text Matching13 Jun 2024
- 2024Advanced Payment Security System:XGBoost, LightGBM and SMOTE Integrated7 Jun 2024
- 2024Credit Card Fraud Detection Using Advanced Transformer Model6 Jun 2024
- 2024Advancements in Feature Extraction Recognition of Medical Imaging Systems Through Deep Learning Technique23 May 2024
- 2024Exploration of Multi-Scale Image Fusion Systems in Intelligent Medical Image Analysis23 May 2024
- 2024Efficiency optimization of large-scale language models based on deep learning in natural language processing tasks20 May 2024
- 2024Real-Time Pill Identification for the Visually Impaired Using Deep Learning8 May 2024
- 2024Adaptive Ensembles of Fine-Tuned Transformers for LLM-Generated Text Detection20 Mar 2024
- 2023Communication Efficient and Privacy-Preserving Federated Learning Based on Evolution Strategies5 Nov 2023