Alert button

Towards Communication-Learning Trade-off for Federated Learning at the Network Edge

Add code
Alert button
May 27, 2022
Figure 1 for Towards Communication-Learning Trade-off for Federated Learning at the Network Edge
Figure 2 for Towards Communication-Learning Trade-off for Federated Learning at the Network Edge

Share this with someone who'll enjoy it:

In this letter, we study a wireless federated learning (FL) system where network pruning is applied to local users with limited resources. Although pruning is beneficial to reduce FL latency, it also deteriorates learning performance due to the information loss. Thus, a trade-off problem between communication and learning is raised. To address this challenge, we quantify the effects of network pruning and packet error on the learning performance by deriving the convergence rate of FL with a non-convex loss function. Then, closed-form solutions for pruning control and bandwidth allocation are proposed to minimize the weighted sum of FL latency and FL performance. Finally, numerical results demonstrate that 1) our proposed solution can outperform benchmarks in terms of cost reduction and accuracy guarantee, and 2) a higher pruning rate would bring less communication overhead but also worsen FL accuracy, which is consistent with our theoretical analysis.

* IEEE Communications Letters, 2022  * This paper has been accepted by IEEE Communications Letters  

Share this with someone who'll enjoy it: