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}
}