Residential College | false |
Status | 已發表Published |
FedEKT: Ensemble Knowledge Transfer for Model-Heterogeneous Federated Learning | |
Meihan Wu1; Li Li2; Tao Chang1; Peng Qiao1; Cui Miao1; Jie Zhou1; Jingnan Wang1; Xiaodong Zhang1 | |
2024-05 | |
Conference Name | International Symposium on Quality of Service |
Source Publication | IEEE International Workshop on Quality of Service, IWQoS |
Pages | 202971 |
Conference Date | 2024-06 |
Conference Place | Guangdong |
Publisher | Institute of Electrical and Electronics Engineers Inc. |
Abstract | Federated Learning (FL) enables multiple clients to collaboratively train a shared server model while preserving data privacy. Most existing FL systems rely on the assumption that the server model and client models have homogeneous architecture. However, intensive resource requirements during the training process prevent low-end devices from contributing to the server model with their own data. On the other hand, the resource constraints on participating clients can significantly limit the size of the server model in the model-homogeneous setting, thereby restricting the application scope of FL. In this work, we propose FedEKT, a novel model-heterogeneous FL system designed to obtain a high-performance large server model while benefiting heterogeneous small client models. Specifically, a new aggregation approach is designed to enable the integration of knowledge from heterogeneous client models to a large server model while mitigating the adverse effects of biases stemming from data heterogeneity. Subsequently, to enhance the performance of client models by benefiting from the high-performance server model, FedEKT distills this large server model into multiple heterogeneous client models, facilitating the transfer of integrated knowledge back to the client models. In addition, we design specialized modules within the model and communication strategy to accomplish aggregation and transfer of knowledge in a data-free manner. The evaluation results demonstrate that FedEKT enhances the accuracy of the server model and client models by up to 53.96% and 12.35%, respectively, compared with the state-of-the-art FL approach on CIFAR-100. |
Keyword | Federated Learning Knowledge Transfer Model Heterogeneity |
DOI | 10.1109/IWQoS61813.2024.10682872 |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85206381671 |
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 | Li Li |
Affiliation | 1.National University of Defense Technology, Changsha, China 2.University of Macau, Macao |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Meihan Wu,Li Li,Tao Chang,et al. FedEKT: Ensemble Knowledge Transfer for Model-Heterogeneous Federated Learning[C]:Institute of Electrical and Electronics Engineers Inc., 2024, 202971. |
APA | Meihan Wu., Li Li., Tao Chang., Peng Qiao., Cui Miao., Jie Zhou., Jingnan Wang., & Xiaodong Zhang (2024). FedEKT: Ensemble Knowledge Transfer for Model-Heterogeneous Federated Learning. IEEE International Workshop on Quality of Service, IWQoS, 202971. |
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