Note: This repo is still in progress.
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.
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} } - PFLlib repo - Some Pytorch codes are adapted from this repo.
- @Jingyi Chai - Transfer codes from Pytorch to TensorFlow, PaddlePaddle and MindSpore.