Clustered Federated Learning with Model Integration for Non-IID Data in Wireless Networks

Jingyi Wang, Zhongyuan Zhao, Wei Hong, Tony Q.S. Quek, Zhiguo Ding

研究成果: Conference contribution

2 引文 斯高帕斯(Scopus)

摘要

As a typical distributed learning paradigm, federated learning has enabled network edge intelligence by making full use of the local data and the computing resources at edge devices without privacy leakage. However, due to the non-IID characteristics of data samples and the unreliability of transmission circumstances, deployment of federated learning in edge networks cannot be well guaranteed. To tackle these challenges, in this paper, a clustered federated learning paradigm with model integration is proposed. First, the detailed framework of our paradigm is introduced. The key idea is to divide the users into multiple individual user clusters by managing the scale and participants of each cluster, and the distribution divergence can be mitigated via cluster-based federated learning. Then, all the learning models are ensembled by model integration to generalize on various target tasks. Second, an upper bound on the accuracy loss of our proposed paradigm is derived, which provides some insights for the impact of data distributions and channel qualities on model performance. To further improve the accuracy performance in wireless networks, a user clustering algorithm is sophisticatedly designed. Finally, the simulation results are provided to verify the significant performance gains of our proposed framework.

原文English
主出版物標題2022 IEEE GLOBECOM Workshops, GC Wkshps 2022 - Proceedings
發行者Institute of Electrical and Electronics Engineers Inc.
頁面1634-1639
頁數6
ISBN(電子)9781665459754
DOIs
出版狀態Published - 2022
事件2022 IEEE GLOBECOM Workshops, GC Wkshps 2022 - Virtual, Online, Brazil
持續時間: 2022 12月 42022 12月 8

出版系列

名字2022 IEEE GLOBECOM Workshops, GC Wkshps 2022 - Proceedings

Conference

Conference2022 IEEE GLOBECOM Workshops, GC Wkshps 2022
國家/地區Brazil
城市Virtual, Online
期間22-12-0422-12-08

All Science Journal Classification (ASJC) codes

  • 儀器
  • 電腦網路與通信
  • 控制和優化

指紋

深入研究「Clustered Federated Learning with Model Integration for Non-IID Data in Wireless Networks」主題。共同形成了獨特的指紋。

引用此