Skip to content

Official code implementation for "Personalized Federated Learning using Hypernetworks" [ICML 2021]

Notifications You must be signed in to change notification settings

AvivSham/pFedHN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Personalized Federated Learning using Hypernetworks [ICML 2021]

This is an official implementation of Personalized Federated Learning using Hypernetworks paper. [Link]

Installation

  • Create a virtual environment with conda/virtualenv
  • Clone the repo
  • Run: cd <PATH_TO_THE_CLONED_REPO>
  • Run: pip install -e . to install necessary packages and path links.

Reproduce Paper Results


PfedHN Results on CIFAR10
  • Run: cd experiments/pfedhn
  • Run: python trainer.py

PfedHN-PC Results on CIFAR10
  • Run: cd experiments/pfedhn_pc
  • Run: python trainer.py

Citation

If you find pFedHN to be useful in your own research, please consider citing the following paper:

@inproceedings{shamsian2021personalized,
  title={Personalized federated learning using hypernetworks},
  author={Shamsian, Aviv and Navon, Aviv and Fetaya, Ethan and Chechik, Gal},
  booktitle={International Conference on Machine Learning},
  pages={9489--9502},
  year={2021},
  organization={PMLR}
}

About

Official code implementation for "Personalized Federated Learning using Hypernetworks" [ICML 2021]

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages