A Communication-efficient Semi Asynchronous Federated Learning Network in Power Grid
Xian Luo, Zehua You, Rongtao Liao, Fen Liu, Dong Liang · 2022
The Electric Internet of Things (IoT) scenario generates a large amount of network edge data. The analysis of these data will significantly enhance the efficiency of IoT. However, traditional mobile edge computing (MEC) can not meet the need for data privacy protection. Currently, a new distributed learning approach, federated learning, has been proposed to solve data security problems and provide efficient model performance through the training of a shared model by multiple devices. Nevertheless, limited wireless communication resources and device heterogeneity are also key factors affecting FL performance Therefore, we propose a novel semi asynchronous federated learning (SAFL) framework, which aims to reduce calculation and transmission latency and improve the learning rate. We formulate a joint terminal device selection and resource allocation problem under the proposed SAFL framework to ensure communication efficiency. To deal with our proposed Markov Decision Process (MDP) problem, we introduce a deep deterministic policy gradient (DDPG) algorithm to find the optimal solution. Numerical results demonstrate the advantages of the proposed algorithm.