학술논문

Efficient Federated Learning in Resource-Constrained Edge Intelligence Networks Using Model Compression
Document Type
Periodical
Source
IEEE Transactions on Vehicular Technology IEEE Trans. Veh. Technol. Vehicular Technology, IEEE Transactions on. 73(2):2643-2655 Feb, 2024
Subject
Transportation
Aerospace
Training
Convergence
Computational modeling
Servers
Optimization
Neural networks
Scheduling
Federated learning
model compression
energy efficiency
device selection
resource allocation
Language
ISSN
0018-9545
1939-9359
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.