arXiv · 2312.14638
Balancing Energy Efficiency and Distributional Robustness in Over-the-Air Federated Learning
Abstract
The growing number of wireless edge devices has magnified challenges concerning energy, bandwidth, latency, and data heterogeneity. These challenges have become bottlenecks for distributed learning. To address these issues, this paper presents a novel approach that ensures energy efficiency for distributionally robust federated learning (FL) with over air computation (AirComp). In this context, to effectively balance robustness with energy efficiency, we introduce a novel client selection method that integrates two complementary insights: a deterministic one that is designed for energy efficiency, and a probabilistic one designed for distributional robustness. Simulation results underscore the efficacy of the proposed algorithm, revealing its superior performance compared to baselines from both robustness and energy efficiency perspectives, achieving more than 3-fold energy savings compared to the considered baselines.
Explore related subjects
Keep this discovery
Mohamed Badi, Chaouki Ben Issaid, Anis Elgabli, Mehdi Bennis. 2023-12-22. Balancing Energy Efficiency and Distributional Robustness in Over-the-Air Federated Learning. https://arxiv.org/abs/2312.14638
Cite the original work for its findings. Save a collection to share your selection of sources.