Personalized Federated Learning with Inferred Collaboration Graphs, ICML2023

Note: This repo is still in progress.

Introduction

This repo is the official implementation of Personalized Federated Learning with Inferred Collaboration Graphs (ICML Link), which has been accepted by ICML, 2023.

We have open-sourced our codes on three libraries, including Pytorch, TensorFlow, PaddlePaddle, and MindSpore! Please refer to the corresponding sub-folder for detailed README.

Citation

Please cite our paper if you find the repository helpful.

@inproceedings{ye2023personalized,
  title={Personalized federated learning with inferred collaboration graphs},
  author={Ye, Rui and Ni, Zhenyang and Wu, Fangzhao and Chen, Siheng and Wang, Yanfeng},
  booktitle={International conference on machine learning},
  pages={39801--39817},
  year={2023},
  organization={PMLR}
}

Acknowledgement

• PFLlib repo - Some Pytorch codes are adapted from this repo.
• @Jingyi Chai - Transfer codes from Pytorch to TensorFlow, PaddlePaddle and MindSpore.