An Efficient Asynchronous Federated Learning Protocol for Edge Devices
Qian Li,Ziyi Gao,Yetao Sun,Yan Wang,Rui Wang,Haiyan Zhu
DOI: https://doi.org/10.1109/jiot.2024.3406634
IF: 10.6
2024-08-25
IEEE Internet of Things Journal
Abstract:Recent studies highlight the significant potential of edge computing and federated learning (FL) in advancing artificial intelligence. However, challenges, such as unstable device performance and the heterogeneously distributed feature of local data, pose threats to the efficiency of global model training. To address these issues, we propose an efficient federated edge learning protocol with key innovations: 1) propose a timing query mechanism, which controls the impact of slow clients on interaction time and changes the parameter server from a passive receiver to an active querier to guarantee the aggregated subset size; 2) propose a screening supplementary strategy from the perspective of optimizing the quality of the data combination; and 3) integrate the timing query mechanism and screening supplementary strategy in a flexible manner. This combination improves the interaction efficiency of global model training, focusing on both interaction time and aggregation update quality. Experiments show that while ensuring the convergence efficiency of each round and the global model performance, the interaction efficiency has been improved by 25% to 75%, thus ensuring the efficiency of FL.
computer science, information systems,telecommunications,engineering, electrical & electronic