Adaptive Asynchronous Federated Learning in Resource-Constrained Edge Computing

Jianchun Liu,Hongli Xu,Lun Wang,Yang Xu,Chen Qian,Jinyang Huang,He Huang
DOI: https://doi.org/10.1109/tmc.2021.3096846
IF: 6.075
2023-01-01
IEEE Transactions on Mobile Computing
Abstract:Federated learning (FL) has been widely used to train machine learning models over massive data in edge computing. However, the existing FL solutions may cause long training time and/or high resource ( e.g. , bandwidth) cost, and thus cannot be directly applied for resource-constrained edge nodes, such as base stations and access points. In this paper, we propose a novel communication-efficient asynchronous federated learning (CE-AFL) mechanism, in which the parameter server will aggregate the local model updates only from a certain fraction α, with 0 < α < 1, of all edge nodes by their arrival order in each epoch. As a case study, we design efficient algorithms to determine the optimal value of α for two cases of CE-AFL, single learning task and multiple learning tasks, under bandwidth constraints. We formally prove the convergence of the proposed algorithm. We evaluate the performance of our algorithm with experiments on Jetson TX2, deep learning workstation and extensive simulations. Both experimental results and simulation results on the classical models and datasets show the effectiveness of our proposed mechanism and algorithms. For example, CE-AFL can reduce the training time by about 69% while achieving similar accuracy, and improve the accuracy of the trained models by about 18% under resource constraints, compared with the state-of-the-art solutions.
What problem does this paper attempt to address?