Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Bridging Model Heterogeneity in Federated Learning via Uncertainty-based Asymmetrical Reciprocity Learning

About

This paper presents FedType, a simple yet pioneering framework designed to fill research gaps in heterogeneous model aggregation within federated learning (FL). FedType introduces small identical proxy models for clients, serving as agents for information exchange, ensuring model security, and achieving efficient communication simultaneously. To transfer knowledge between large private and small proxy models on clients, we propose a novel uncertainty-based asymmetrical reciprocity learning method, eliminating the need for any public data. Comprehensive experiments conducted on benchmark datasets demonstrate the efficacy and generalization ability of FedType across diverse settings. Our approach redefines federated learning paradigms by bridging model heterogeneity, eliminating reliance on public data, prioritizing client privacy, and reducing communication costs.

Jiaqi Wang, Chenxu Zhao, Lingjuan Lyu, Quanzeng You, Mengdi Huai, Fenglong Ma• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (val)
Accuracy34.4
781
Image ClassificationTinyImageNet (val)
Accuracy31.6
289
Image ClassificationMNIST (val)
Accuracy86.7
58
Image ClassificationCIFAR10 (val)
Accuracy40.2
51
ClassificationOrganAMNIST
Average Delta0.001
18
ClassificationCIFAR-10
Worst Delta0.0224
18
ClassificationOrganAMNIST
Worst Delta0.0256
18
ClassificationOCTMNIST
Average Delta0.0247
18
ClassificationFashionMNIST
Worst Delta3.56
18
ClassificationOCTMNIST
Worst Delta0.0725
18
Showing 10 of 20 rows

Other info

Follow for update