Energy-Efficient Client Sampling for Federated Learning in Heterogeneous Mobile Edge Computing Networks
Jian Xin Tang, Xiuhua Li, Hui Li, Min Xiong, Xiaofei Wang, Victor C. M. Leung · 2024
To address network congestion and data privacy concerns, federated learning (FL) that combines multiple clients and a parameter server has been widely used in mobile edge computing (MEC) networks to process the abundant data generated by mobile clients. However, the existing client sampling methods do not adequately consider the data heterogeneity and system heterogeneity. Parameter server selects inappropriate clients to participate in the FL training process. This inevitably leads to slower convergence of the global model and higher energy consumption. In this paper, we design a client sampling model with the goal of selecting suitable clients to improve the energy efficiency of FL in heterogeneous MEC networks. Then we propose an energy-efficient client sampling strategy by quantifying the communication capability, computation capability and data quality of clients. Based on the quantization results, clients are assigned with a corresponding sampled probability. Simulation results show that our proposed strategy can effectively accelerate the convergence of the global model and reduce the energy consumption compared with the baseline schemes.