Learning from Unreliable Trajectories: Adversarially-Robust Federated Q-Learning
cs.LG, cs.SY, eess.SY
Submitted: 2026-10-02
Updated: 2026-10-02
License: http://creativecommons.org/licenses/by/4.0/
The gist: We study federated reinforcement learning in which multiple agents interact with a common Markov decision process and communicate through a central server to collaboratively learn the optimal
Terminology
Abstract
We study federated reinforcement learning in which multiple agents interact with a common Markov decision process and communicate through a central server to collaboratively learn the optimal state-action value function. Our goal is to understand whether the sample-efficiency benefits of collaboration can be retained when a fraction of the agents behave adversarially and transmit arbitrarily corrupted information. To address this problem, we introduce Robust Async-Fed-Q, an epoch-based federated learning algorithm that combines variance-reduced estimation of the Bellman optimality operator at the agents with robust aggregation at the server. We establish high-probability finite-time guarantees showing that the proposed method preserves the statistical gains of collaboration among the honest agents while tolerating adversarial corruption. In particular, the effect of the adversarial agents decreases as the amount of data collected by each honest agent grows and eventually vanishes in the infinite-sample limit. We complement these guarantees with information-theoretic lower bounds that characterize the unavoidable statistical cost of adversarial corruption, leading to the first nearly matching upper and lower bounds for adversarially robust federated reinforcement learning. We further extend our framework to accommodate single-trajectory Markovian sampling and heterogeneous partial coverage, where different agents may explore different regions of the state-action space and learning relies on their collective coverage. Finally, our epoch-based design substantially improves the best known communication complexity for federated Q-learning under asynchronous sampling.
Sources
- Federated Optimization: Distributed Machine Learning for On-Device Intelligence
- Federated Reinforcement Learning: Techniques, Applications, and Open Challenges
- Federated Temporal Difference Learning with Linear Function Approximation under Environmental Heterogeneity
- Improved Communication Efficiency in Federated Natural Policy Gradient via ADMM-based Gradient Updates
- Federated Q-Learning with Reference-Advantage Decomposition: Almost Optimal Regret and Logarithmic Communication Cost
- Federated UCBVI: Communication-Efficient Federated Regret Minimization with Heterogeneous Agents
- Momentum for the Win: Collaborative Federated Reinforcement Learning across Heterogeneous Environments
- Linear Stochastic Approximation: Constant Step-Size and Iterate Averaging
- Private and Byzantine-Proof Cooperative Decision-Making
- Stochastic approximation with cone-contractive operators: Sharp $\ell_\infty$-bounds for $Q$-learning
- Finite-Sample Analysis of Nonlinear Stochastic Approximation with Applications in Reinforcement Learning
- Gymnasium: A Standard Interface for Reinforcement Learning Environments
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks