English

MetaFed: Federated Learning among Federations with Cyclic Knowledge Distillation for Personalized Healthcare

Machine Learning 2023-07-11 v3 Computers and Society

Abstract

Federated learning has attracted increasing attention to building models without accessing the raw user data, especially in healthcare. In real applications, different federations can seldom work together due to possible reasons such as data heterogeneity and distrust/inexistence of the central server. In this paper, we propose a novel framework called MetaFed to facilitate trustworthy FL between different federations. MetaFed obtains a personalized model for each federation without a central server via the proposed Cyclic Knowledge Distillation. Specifically, MetaFed treats each federation as a meta distribution and aggregates knowledge of each federation in a cyclic manner. The training is split into two parts: common knowledge accumulation and personalization. Comprehensive experiments on three benchmarks demonstrate that MetaFed without a server achieves better accuracy compared to state-of-the-art methods (e.g., 10%+ accuracy improvement compared to the baseline for PAMAP2) with fewer communication costs.

Keywords

Cite

@article{arxiv.2206.08516,
  title  = {MetaFed: Federated Learning among Federations with Cyclic Knowledge Distillation for Personalized Healthcare},
  author = {Yiqiang Chen and Wang Lu and Xin Qin and Jindong Wang and Xing Xie},
  journal= {arXiv preprint arXiv:2206.08516},
  year   = {2023}
}

Comments

Accepted by IEEE Trans. on Neural Networks and Learning Systems (TNNLS); IJCAI'22 FTL workshop innovation award; code at https://github.com/microsoft/PersonalizedFL