University of Oulu

S. Liu, G. Yu, X. Chen and M. Bennis, "Joint User Association and Resource Allocation for Wireless Hierarchical Federated Learning with Non-IID Data," ICC 2022 - IEEE International Conference on Communications, Seoul, Korea, Republic of, 2022, pp. 74-79, doi: 10.1109/ICC45855.2022.9839164

Joint user association and resource allocation for wireless hierarchical federated learning with non-IID data

Saved in:
Author: Liu, Shengli1; Yu, Guanding1; Yu, Guanding2;
Organizations: 1College of Information Science and Electronic Engineering, Zhejiang University, Hangzhou, China
2VTT Technical Research Centre of Finland, Finland
3Centre for Wireless Communication, University of Oulu, Finland
Format: article
Version: accepted version
Access: open
Online Access: PDF Full Text (PDF, 1.5 MB)
Persistent link:
Language: English
Published: Institute of Electrical and Electronics Engineers, 2022
Publish Date: 2023-02-10


Wireless hierarchical federated learning (HFL) has been proposed for large-scale model training over multi-cell network while preserving the data privacy. However, the imbalanced data distribution and load have a significant impact on the convergence rate, the learning accuracy, and the learning latency in wireless HFL with non-independent identically distributed training data. To cope with these challenges, we first derive the learning latency and the upper bound of the model error. Then, an optimization problem is formulated to minimize the weighted sum of total data distribution distance and learning latency. Joint user association and wireless resource allocation algorithms are investigated to achieve the optimal learning performance. Finally, the effectiveness of the proposed algorithms are demonstrated by the simulations.

see all

Series: IEEE International Conference on Communications
ISSN: 1550-3607
ISSN-E: 1938-1883
ISSN-L: 1550-3607
ISBN: 978-1-5386-8347-7
ISBN Print: 978-1-5386-8348-4
Pages: 74 - 79
DOI: 10.1109/icc45855.2022.9839164
Host publication: ICC 2022 - IEEE International Conference on Communications
Conference: IEEE International Conference on Communications
Type of Publication: A4 Article in conference proceedings
Field of Science: 213 Electronic, automation and communications engineering, electronics
Funding: The work of Guanding Yu was supported by research grant under Grant GDNRC[2021]32. The work of Xianfu Chen was supported by the Zhejiang Laboratory Open Program under Grant 2021LC0AB06.
Copyright information: © 2022 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.