Adaptive Communication Resource Allocation for Federated Learning with UEP Strategies

Muhang Lan,Song Xiao,Wenyi Zhang
DOI: https://doi.org/10.1109/vtc2024-spring62846.2024.10683286
2024-01-01
Abstract:This paper proposes an adaptive communication resource allocation algorithm to address the model transfer challenge in federated learning (FL). Our approach dynamically adjusts transmission power and channel coding rates based on FL training states, utilizing a bit-position-aware transmission scheme rooted in unequal error protection (UEP) principles. The system's performance is analytically assessed through a derived upper bound on the FL convergence rate, considering model transmission accuracy and client gradient diversities in heterogeneous data distributions. Then a resource allocation problem, aimed at minimizing the proposed upper bound, is decomposed into two sub-problems and solved. The first sub-problem, minimizing a derived upper bound on model transmission error, is tackled using a proposed algorithm for adaptive bit power allocation. The second sub-problem, minimizing cumulative gradient diversities, is formulated as a Markov decision process (MDP) and solved using deep Q-learning. Numerical evaluations show that our method outperforms vanilla and existing UEP-coded algorithms.
What problem does this paper attempt to address?