Personalized Federated Reinforcement Learning via Model-Agnostic Meta-Learning: Convergence of Exact and Hessian-Free Meta-Policy Gradients
On the map: Deep reinforcement learning methods → RL for energy, networks and operations
Abstract (arXiv)
We study personalized federated reinforcement learning, in which n agents, each acting in its own Markov decision process, collaborate through a server to learn a shared MAML-style policy initialization that becomes effective for an individual agent once that agent adapts it with a single local policy-gradient step. We propose Per-FedAvg-PG, in which agents take τ local stochastic meta-policy-gradient steps between communication rounds, and prove that it reaches an varepsilon-approximate first-order stationary point of the personalized objective in K=mathcal O(varepsilon-3/2) rounds with τ=Theta(varepsilon-1/2) local steps. The analysis rests on a structural feature of the reinforcement learning setting: under standard policy-class regularity, the per-agent objectives have uniformly bounded gradients and Hessians with explicit constants, so the bounded-gradient and bounded-heterogeneity conditions imposed by the supervised theory hold automatically and no separate heterogeneity assumption is needed. The exact meta-gradient requires the inner-loop policy Hessian, which our experiments identify as the practical bottleneck. We therefore analyze the Hessian-free variant, bound its bias, and exhibit fixed points at which the meta-gradient is nonzero and of order α, showing that the resulting stationarity floor is a property of the method rather than of the bound. Experiments on tabular and neural navigation confirm the predicted behavior and show transfer to unseen agents at an order of magnitude lower sample cost than independent training. Together these results identify the adaptation step size as a tunable personalization knob and the curvature estimate as the quantity that governs whether exact meta-gradients are affordable.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in RL for energy, networks and operations, one of 15 directions in Deep reinforcement learning methods. Builds on 23 of its 41 references that are in the map. First posted on arXiv 19 Sep 2026.
Builds on
23 of 41 references.
Closest ancestors · 7
Same direction, newest first
- 2025On Global Convergence Rates for Federated Softmax Policy Gradient under Heterogeneous Environments29 May
- 2025Collaborative Value Function Estimation Under Model Mismatch: A Federated Temporal Difference Analysis21 Mar
- 2024On the Linear Speedup of Personalized Federated Reinforcement Learning with Shared Representations22 Nov
- 2024Towards Fast Rates for Federated and Multi-Task Reinforcement Learning9 Sep
- 2024Asynchronous Federated Reinforcement Learning with Policy Gradient Updates: Algorithm Design and Convergence Analysis9 Apr
- 2024Compressed Federated Reinforcement Learning with a Generative Model26 Mar
- 2024Finite-Time Analysis of On-Policy Heterogeneous Federated Reinforcement Learning27 Jan
Other references · 16
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,258
- 2020Personalized Federated Learning with Moreau Envelopescited by 618
- 2021Exploiting Shared Representations for Personalized Federated Learningcited by 541
- 2019On the Convergence Theory of Gradient-Based Model-Agnostic Meta-Learning Algorithmscited by 132
- 2018Stochastic Variance-Reduced Policy Gradientcited by 106
- 2019Lifelong Federated Reinforcement Learning: A Learning Architecture for Navigation in Cloud Robotic Systemscited by 78
- 2022Federated Reinforcement Learning with Environment Heterogeneitycited by 63
- 2023The Blessing of Heterogeneity in Federated Q-Learning: Linear Speedup and Beyondcited by 36
- 2020On the Convergence Theory of Debiased Model-Agnostic Meta-Reinforcement Learningcited by 19
- 2024Meta-Learning Linear Quadratic Regulators: A Policy Gradient MAML Approach for Model-free LQRcited by 17
4 more
- 2023Federated Natural Policy Gradient and Actor Critic Methods for Multi-task Reinforcement Learningcited by 11
- 2023Convergence of First-Order Algorithms for Meta-Learning with Moreau Envelopescited by 8
- 2022PersA-FL: Personalized Asynchronous Federated Learningcited by 5
- 2025A Generalized Meta Federated Learning Framework with Theoretical Convergence Guaranteescited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Collaborative Yet Personalized Policy Training: Single-Timescale Federated Actor-Critic
- 2024Federated Reinforcement Learning with Constraint Heterogeneity
- 2024Momentum for the Win: Collaborative Federated Reinforcement Learning across Heterogeneous Environments
- 2026Personalized Observation Normalization for Federated Reinforcement Learning in Simulation Environments with Heterogeneity
- 2020Characterizing Policy Divergence for Personalized Meta-Reinforcement Learning ↗
- 2025Achieving Tighter Finite-Time Rates for Heterogeneous Federated Stochastic Approximation under Markovian Sampling
- 2026Exploration-Driven Personalized Federated Reinforcement Learning via Intrinsic Motivation
- 2024Momentum-Based Federated Reinforcement Learning with Interaction and Communication Efficiency