Federated Learning for Cellular Networks: Joint User Association and Resource Allocation

Abstract

Recent years have shown a remarkable interest in federated learning from researchers to make several Internet of Things applications smart. Although, federated learning offers users' privacy preservation, it has communication resources optimization challenge. In this paper, we consider federated learning for cellular networks. We formulate an optimization problem to jointly minimizes latency and effect of loss in federated learning model accuracy due to channel uncertainties. We decompose the main optimization problem into two sub-problems: resource allocation and device association sub-problems, due to the NP-hard nature of the main optimization problem. To solve these sub-problems, we propose an iterative approach which further uses efficient heuristic algorithms for resource blocks allocation and device association. Finally, we provide numerical results for the validation of our proposed scheme.