ProxyFL

Code accompanying the paper "ProxyFL: Decentralized Federated Learning through Proxy Model Sharing"

Authors: Shivam Kalra*, Junfeng Wen*, Jesse C. Cresswell, Maksims Volkovs, Hamid R. Tizhoosh†

• * Denotes equal contribution
• † University of Waterloo / Vector Institute

Prerequisite

• Python 3.9

conda create -n ProxyFL python=3.9
conda activate ProxyFL

• PyTorch 1.9.0

conda install pytorch=1.9.0 torchvision=0.10.0 numpy=1.21.2 -c pytorch

• mpi4py 3.1.2

conda install -c conda-forge mpi4py=3.1.2

• opacus 0.14.0

pip install 'opacus==0.14.0'

• matplotlib 3.4.3

conda install -c conda-forge matplotlib=3.4.3

Run experiment

Download data via

bash download_data.sh

Then run the script

bash run_exp.sh

Citation

If you find this code useful in your research, please cite the following paper:

@inproceedings{kalra2021,
  title={ProxyFL: Decentralized Federated Learning through Proxy Model Sharing},
  author={Shivam Kalra, Junfeng Wen, Jesse C. Cresswell, Maksims Volkovs, Hamid R. Tizhoosh},
  year={2021}
}