Provably Efficient Federated Reinforcement Learning with Linear Function Approximation and Logarithmic Communication Cost
stat.ML, cs.AI, cs.LG
Submitted: 2026-08-31
Updated: 2026-08-31
License: http://creativecommons.org/licenses/by/4.0/
The gist: We study federated online reinforcement learning with linear function approximation.
Terminology
Abstract
We study federated online reinforcement learning with linear function approximation. While recent multi-agent reinforcement learning algorithms achieve strong regret guarantees, they typically require sharing raw trajectories. This reliance incurs a communication cost that scales linearly with the number of episodes and violates the privacy constraints of federated settings. To address these limitations, we propose Fed-LSVI, the first provably efficient federated algorithm for online reinforcement learning with linear function approximation in episodic Markov decision processes. By integrating a determinant-based event-triggered synchronization with a stepwise backward update mechanism, Fed-LSVI enables agents to collaboratively learn an optimal policy by exchanging only compressed sufficient statistics. We prove that Fed-LSVI achieves a regret bound of (sqrt Md 3H 4T), where d is the feature dimension, H is the horizon length, M is the number of agents, and T is the number of episodes per agent, matching the best-known regret for multi-agent online reinforcement learning with linear function approximation. Moreover, by following the stringent communication and privacy constraints of the federated setting, Fed-LSVI reduces the communication cost to only logarithmic dependence on T, representing a significant improvement over prior methods.
Sources
- Federated TD Learning over Finite-Rate Erasure Channels: Linear Speedup under Markovian Sampling
- Provably Efficient Cooperative Multi-Agent Reinforcement Learning with Function Approximation
- Understanding Gap-Dependent Regret for Optimism-Based Reinforcement Learning with Linear Function Approximation
Related papers
- Behavior of prediction performance metrics with rare events
- Optimal Estimation of Generic Dynamics by Path-Dependent Neural Jump ODEs
- A Posterior-Dynamics Framework for Imaging Inverse Problems with Pretrained Diffusion Priors
- Efficient quadratic entropy with distance sketches
- Embedding-Bias in Conditional Independence Testing
- From Log-Odds to Shapley Values: An Explanatory Geometry for the Weighted Naive Bayes Classifier