Joint User Selection and Resource Allocation for Fast Federated Edge Learning

JIANG Zhihui,HE Yinghui,YU Guanding
DOI: https://doi.org/10.12142/ztecom.202002004
2020-01-01
Abstract:By periodically aggregating local learning updates from edge users, federated edge learning(FEEL) is envisioned as a promising means to reap the benefit of local rich data and protect users’ privacy. However, the scarce wireless communication resource greatly limits the number of participated users and is regarded as the main bottleneck which hinders the development of FEEL. To tackle this issue, we propose a user selection policy based on data importance for FEEL system. In order to quantify the data importance of each user, we first analyze the relationship between the loss decay and the squared norm of gradient. Then, we formulate a combinatorial optimization problem to maximize the learning efficiency by jointly considering user selection and communication resource allocation. By problem transformation and relaxation, the optimal user selection policy and resource allocation are derived, and a polynomial-time optimal algorithm is developed. Finally, we deploy two commonly used deep neural network(DNN) models for simulation. The results validate that our proposed algorithm has strong generalization ability and can attain higher learning efficiency compared with other traditional algorithms.
What problem does this paper attempt to address?