Training efficiency optimization algorithm of wireless federated learning based on processor performance and network condition awareness

Abstract

Abstract With the explosive growth of smart mobile devices in wireless networks, the increasing computational power of mobile chips and the growing concern for personal privacy, a decentralized deep learning framework at the mobile terminal layer has emerged called federated learning (FL) to enhance user experience. This paper studies the training efficiency optimization problem of wireless FL that jointly considers processor performance, channel conditions and terminals’ power in a non-independent identically distribution (non-IID) scenario. And, the training efficiency optimization problem is mathematically modeled and then decomposed into several sub-problems based on the independence and decoupling of the variables involved. To enhance the training efficiency of wireless FL, a comprehensive scheduling strategy encompassing computational and communication aspects is proposed. Simulation results show that the proposed scheduling strategy for wireless FL achieves superior learning performance with reduced training latency.

Keywords

• Federated learning
• Wireless communication
• Scheduling policies
• Parallel and distributed algorithms
• Resource allocation