Reinforcement Learning As a Catalyst for Robust and Fair Federated Learning: Deciphering the Dynamics of Client Contributions

Jialuo He,Wei Chen,Xiaojin Zhang
DOI: https://doi.org/10.48550/arxiv.2402.05541
2024-01-01
Abstract:Recent advancements in federated learning (FL) have produced models thatretain user privacy by training across multiple decentralized devices orsystems holding local data samples. However, these strategies often neglect theinherent challenges of statistical heterogeneity and vulnerability toadversarial attacks, which can degrade model robustness and fairness.Personalized FL strategies offer some respite by adjusting models to fitindividual client profiles, yet they tend to neglect server-side aggregationvulnerabilities. To address these issues, we propose Reinforcement FederatedLearning (RFL), a novel framework that leverages deep reinforcement learning toadaptively optimize client contribution during aggregation, thereby enhancingboth model robustness against malicious clients and fairness acrossparticipants under non-identically distributed settings. To achieve this goal,we propose a meticulous approach involving a Deep Deterministic PolicyGradient-based algorithm for continuous control of aggregation weights, aninnovative client selection method based on model parameter distances, and areward mechanism guided by validation set performance. Empirically, extensiveexperiments demonstrate that, in terms of robustness, RFL outperforms thestate-of-the-art methods, while maintaining comparable levels of fairness,offering a promising solution to build resilient and fair federated systems.
What problem does this paper attempt to address?