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

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...

Full description

Saved in:
Bibliographic Details
Main Authors: Guohao Pang, Xiaorong Zhu
Format: Article
Language:English
Published: SpringerOpen 2024-11-01
Series:EURASIP Journal on Advances in Signal Processing
Subjects:
Online Access:https://doi.org/10.1186/s13634-024-01192-6
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary: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.
ISSN:1687-6180