Federated Learning for Energy-balanced Client Selection in Mobile Edge Computing

• Autor(a) principal: Zheng, Jingjing
• Data de Publicação: 2021
• Outros Autores: Li, Kai, Tovar, Eduardo, Guizani, Mohsen
• Tipo de documento: Artigo
• Idioma: eng
• Título da fonte: Repositório Científico de Acesso Aberto de Portugal (Repositórios Cientìficos)
• Texto Completo: http://hdl.handle.net/10400.22/18259
• topic: Client selection; Mobile edge computing; Federated learning; Heuristic algorithm

Mobile edge computing (MEC) has been considered as a promising technology to provide seamless integration of multiple application services. Federated learning (FL) is carried out at edge clients in MEC for privacy-preserving training of data processing models. Despite that the edge clients with small data payloads consume less energy on FL training, the small data payload gives rise to a low learning accuracy due to insufficient input to the FL training. Inadequate selection of the edge clients can result in a large energy consumption at the edge clients, or a low learning accuracy of the FL training. In this paper, a new FL-based client selection optimization is proposed to balance the trade-off between energy consumption of the edge clients and the learning accuracy of FL. We first show that this optimization problem is NP-complete. Next, we propose a FL-based energy-accuracy balancing heuristic algorithm to approximate the optimal client selection in polynomial time. The numerical results show the advantage of our proposed algorithm.