Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Journal articles
  4. pFedKT: Personalized federated learning with dual knowledge transfer
 
research article

pFedKT: Personalized federated learning with dual knowledge transfer

Yi, Liping
•
Shi, Xiaorong
•
Wang, Nan
Show more
May 23, 2024
Knowledge-Based Systems

Federated learning (FL) has been widely studied as an emerging privacy-preserving machine learning paradigm for achieving multi-party collaborative model training on decentralized data. In practice, such data tend to follow non-independent and identically distributed (non-IID) data distributions. Thus, the performance of models obtained through vanilla horizontal FL tends to vary significantly across FL clients. To tackle this challenge, a new subfield of FL – personalized federated learning (PFL) – has emerged for producing personalized FL models that can perform well on diverse local datasets. Existing PFL approaches are limited in terms of effectively transferring knowledge among clients to improve model generalization while achieving good performance on diverse local datasets. To bridge this important gap, we propose the personalized Federated Knowledge Transfer (pFedKT) approach. It involves dual knowledge transfer: (1) transferring historical local knowledge to local models via local hypernetworks; and (2) transferring latest global knowledge to local models through contrastive learning. By fusing historical local knowledge and the latest global knowledge, the personalization and generalization of individual models for FL clients can be simultaneously enhanced. We provide theoretical analysis on the generalization and convergence of pFedKT. Extensive experiments on 3 real-world datasets demonstrate that pFedKT achieves 0.74%–1.62% higher test accuracy compared to 14 state-of-the-art baselines.

  • Details
  • Metrics
Type
research article
DOI
10.1016/j.knosys.2024.111633
Scopus ID

2-s2.0-85187792396

Author(s)
Yi, Liping
Shi, Xiaorong
Wang, Nan
Wang, Gang
Liu, Xiaoguang
Shi, Zhuan  

École Polytechnique Fédérale de Lausanne

Yu, Han
Date Issued

2024-05-23

Published in
Knowledge-Based Systems
Volume

292

Article Number

111633

Subjects

Contrastive learning

•

Hypernetwork

•

Knowledge transfer

•

Personalized federated learning

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
LIA  
FunderFunding(s)Grant NumberGrant URL

National Research Foundation Singapore

RIE 2020 Advanced Manufacturing and Engineering

Fundamental Research Funds for the Central Universities

Show more
Available on Infoscience
January 16, 2025
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/242916
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés