Efficient Federated Learning in Resource-Constrained Edge Intelligence Networks using Model Compression

Chao Chen,Bohang Jiang,Shengli Liu,Chuanhuang Li,Celimuge Wu,Rui Yin
DOI: https://doi.org/10.1109/tvt.2023.3318080
IF: 6.8
2023-01-01
IEEE Transactions on Vehicular Technology
Abstract:We investigate energy-efficient federated learning (FL) in computation and communication resource-constrained edge intelligence networks using model compression. An edge device selection strategy is designed to select appropriate edge devices for participating in FL at the beginning of each training iteration. An optimization problem is then formulated to jointly optimize the pruning ratio, CPU frequency, uplink power, and bandwidth allocation for the selected edge devices. Due to the non-convexity of the optimization problem, it is decomposed into three subproblems, and closed-form solutions or efficient algorithms are developed for each subproblem. Based on these solutions, an alternating optimization algorithm is constructed to solve the original problem. Simulation results show that the proposed scheme outperforms baseline schemes in improving the energy efficiency of FL.
telecommunications,engineering, electrical & electronic,transportation science & technology
What problem does this paper attempt to address?