Time Efficient Joint Optimization Federated Learning over Wireless Communication Networks

Junshuai Sun,Yingying Wang,Xin Sun,Na Li,Gaofeng Nie
DOI: https://doi.org/10.23919/jcc.2022.06.013
2022-01-01
China Communications
Abstract:Artificial intelligence(AI)has made a pro-found impact on our daily life.The 6th generation mobile networks(6G)should be designed to enable AI services.The native intelligence is introduced as an important feature in 6G.6G native AI network is realized by the philosophy of federated learning(FL)to ensure data security and privacy.Federated learning over wireless communication networks is treated as a potential solution to realize native AI.However,intro-ducing FL in the 6G will lead to expansive communi-cation cost and unstable FL convergence with unreli-able air interface.In this paper,we propose a solution for FL over wireless networks and analyze the training efficiency.To make full use of the advantages of the proposed network,we introduce a communication-FL joint optimization(CFJO)algorithm by jointly con-sidering the effects of uplink resource,energy con-sumption and latency constraints.CFJO derives a transmission strategy with resource allocation and re-transmissions to reduce the wireless transmission in-terruption probability and model upload latency.The simulation results show that CFJO significantly im-proves the model training efficiency and convergence performance with lower interruption probability under the latency constraint.
What problem does this paper attempt to address?