Residential College | false |
Status | 已發表Published |
FedGosp: A Novel Framework of Gossip Federated Learning for Data Heterogeneity | |
Li, Guanghao1; Hu, Yue1; Zhang, Miao1; Li, Li2; Chang, Tao3; Yin, Quanjun1 | |
2022 | |
Conference Name | 2022 IEEE International Conference on Systems, Man, and Cybernetics, SMC 2022 |
Source Publication | Conference Proceedings - IEEE International Conference on Systems, Man and Cybernetics |
Volume | 2022-October |
Pages | 840-845 |
Conference Date | 9 October 2022through 12 October 2022 |
Conference Place | Prague |
Abstract | Federated learning (FL) provides the possibility to solve the problem of data privacy, but it suffers much from the data heterogeneity among different participants. Currently, some promising FL algorithms improve the effectiveness of learning under the non independent-and-identically-distributed (Non-IID) data settings. However, they require a large number of communication rounds between the server and clients for an acceptable accuracy. Inspired by the training paradigm of gossip learning, this paper proposes a new FL framework, named FedGosp. It first classifies the clients into different categories based on the model weights trained by the locally stored data. Then FedGosp utilizes the communication not only between clients and the server, but also between different classes of clients themselves. This training process enables instilling knowledge about various data distributions in the passed models. We evaluate the performance of FedGosp in multiple Non-IID settings on CIFAR10 and MNIST datasets, and compare it with the recently popular algorithms such as SCAFFOLD, FedAvg and FedProx. The experimental results show that FedGosp can improve the model accuracy by 6.53% and save 5.6 × communication costs at best compared to the second-ranked baseline. |
Keyword | Clustering Algorithm Data Heterogeneity Federated Learning Prediction Accuracy |
DOI | 10.1109/SMC53654.2022.9945192 |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85142707816 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) |
Corresponding Author | Yin, Quanjun |
Affiliation | 1.National University of Defense Technology, College of Systems Engineering, Changsha, China 2.University of Macau, Iotsc, Macao 3.National University of Defense Technology, College of Computer, Changsha, China |
Recommended Citation GB/T 7714 | Li, Guanghao,Hu, Yue,Zhang, Miao,et al. FedGosp: A Novel Framework of Gossip Federated Learning for Data Heterogeneity[C], 2022, 840-845. |
APA | Li, Guanghao., Hu, Yue., Zhang, Miao., Li, Li., Chang, Tao., & Yin, Quanjun (2022). FedGosp: A Novel Framework of Gossip Federated Learning for Data Heterogeneity. Conference Proceedings - IEEE International Conference on Systems, Man and Cybernetics, 2022-October, 840-845. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment