arrow
Return

Adaptive asynchronous federated learning

delete2024-03-01
delete6
PRE
AI
R
Renhao Lu
张
张伟哲 (Weizhe Zhang) *
李
李琼 (Qiong Li)
何
何慧 (Hui He)
X
Xiaoxiong Zhong
H
Hongwei Yang
王
王德胜 (Desheng Wang)
Z
Zenglin Xu
M
Mamoun Alazab
DOI:10.1016/j.future.2023.11.001delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Federated Learning enables data owners to train an artificial intelligence model collaboratively while keeping all the training data locally, reducing the possibility of personal data breaches. However, the heterogeneity of local resources and dynamic characteristics of federated learning systems bring new challenges hindering the development of federated learning techniques. To this end, we propose an Adaptive Asynchronous Federated Learning scheme with Momentum, called FedAAM, comprising an adaptive weight allocation algorithm and a novel asynchronous federated learning framework. Firstly, we dynamically allocate weights for the global model update using an adaptive weight allocation strategy that can improve the convergence rate of models in asynchronous federated learning systems. Then, targeting the challenges mentioned previously, we proposed two new asynchronous global update rules based on the differentiated strategy, which is an essential component of the proposed novel federated learning framework. Furthermore, our asynchronous federated learning framework introduces the historical global update direction (i.e., global momentum) into the global update operation, aiming at improving training efficiency. Moreover, we prove that the model under the FedAAM scheme can achieve a sublinear convergence rate. Extensive experiments on real-world datasets demonstrate that the FedAAM scheme outperforms representative synchronous and asynchronous federated learning schemes (i.e., FedAvg and FedAsync) regarding the model's convergence rate and capacity to deal with dynamic systems.
Keywords:
Federated learning
Asynchronous aggregation
Distributed machine learning
Momentum

Journal

F
Future Generation Computer Systems-The International Journal of eScience
IF:
6.1
Papers:
6.9K
Citations:
2.3W

Organization

H
harbin institute of technology
Scholars:
8.0W
Papers: 6.6W
Citations: 66
Charles Darwin University cover
Charles Darwin University
Scholars:
3.9K
Papers: 3.7K
Citations: 3.3K
P
Peng Cheng Laboratory
Scholars:
1.7K
Papers: 1.8K
Citations: 2.0K
researcher View more organizations
Cited Papers

Cited Papers

FedSA: A Semi-Asynchronous Federated Learning Mechanism in Heterogeneous Edge Computing
err2021-12-01
err104
errOAAI
errMa, Qianpiao; Xu, Yang; Xu, Hongli; Jiang, Zhida; Huang, Liusheng; Huang, He
errShare
errSave
Convergence Time Optimization for Federated Learning Over Wireless Networks
err2021-04-01
err222
errOAAI
errChen, Mingzhe; Poor, H. Vincent; Saad, Walid; Cui, Shuguang
errShare
errSave
Scheduling Policies for Federated Learning in Wireless Networks
err2020-01-01
err445
errOAAI
errYang, Howard H.; Liu, Zuozhu; Quek, Tony Q. S.; Poor, H. Vincent
errShare
errSave
FedProc: Prototypical contrastive federated learning on non-IID data
err2023-06-01
err74
errOAAI
errMu, Xutong; Shen, Yulong; Cheng, Ke; Geng, Xueli; Fu, Jiaxuan; Zhang, Tao; Zhang, Zhiwei
errShare
errSave
errShare
errSave
researcher View more