AceFL: Federated Learning Accelerating in 6G-Enabled Mobile Edge Computing Networks
Jing He, Songtao Guo, Mingyan Li, Yongdong Zhu
Abstract
6G is envisioned to achieve ubiquitous Artificial Intelligence (AI) in heterogeneous and massive-scale networks, where FEderated Edge Learning (FEEL) is an effective way to assist AI technology to be implemented at the network edge. FEEL enables training a machine learning model through communication among multiple edge devices based on data from a large number of users. However, the heterogeneity among distributed edge devices and the limitation of resources may degrade the training efficiency of FEEL over 6G-enabled mobile edge computing (MEC) networks. Taking this challenge into account, a novel federated learning scheme is proposed in this paper to accelerate the training process. In particular, we formulate a training efficiency maximization problem, where a novel analytical relation among the training loss, resource consumption and heterogeneity is inferred by convergence analysis. Then, we propose a searching algorithm named IFBA to obtain the optimal inexactness of local models and frequency band allocation for edge devices in order to mitigate the straggler effect caused by the heterogeneity and resource limitation of devices. As a result, the training efficiency can be improved by adapting the inexactness of local models and frequency band allocation for edge devices on-demand. Simulation results validate that AceFL has advantages in training efficiency improvement and heterogeneity adaption, e.g., accelerating the training by 5% - 15% under different heterogeneity.