├── .gitignore ├── LICENSE ├── README.md ├── core ├── Client.py ├── ClientManage.py ├── ClientManage_hr.py ├── ClientManage_hr_joint.py ├── Client_hr.py ├── SGDClient.py ├── SGDClient_hr.py ├── SVRGClient.py ├── SVRGClient_hr.py ├── __init__.py ├── function.py └── test.py ├── data ├── README.md ├── __init__.py ├── cifar │ └── .gitkeep └── mnist │ └── .gitkeep ├── figs ├── fig1.png ├── fig2.png ├── fig3.png ├── fig31.png ├── fig4.png └── fig_algo.png ├── main_hr.py ├── main_hr_joint.py ├── main_imbalance.py ├── minmax_synthetic.ipynb ├── models ├── ModelBuilder.py ├── Nets.py ├── SvrgUpdate.py ├── Update.py └── __init__.py ├── output └── .gitkeep ├── reproduce ├── fig2.py ├── fig2.sh ├── fig3.py └── fig3.sh ├── requirements.txt └── utils ├── Fed.py ├── README.md ├── __init__.py ├── dataset.py ├── dataset_normal.py ├── my_logging.py ├── options.py ├── sampling.py └── svrg.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/README.md -------------------------------------------------------------------------------- /core/Client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/Client.py -------------------------------------------------------------------------------- /core/ClientManage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/ClientManage.py -------------------------------------------------------------------------------- /core/ClientManage_hr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/ClientManage_hr.py -------------------------------------------------------------------------------- /core/ClientManage_hr_joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/ClientManage_hr_joint.py -------------------------------------------------------------------------------- /core/Client_hr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/Client_hr.py -------------------------------------------------------------------------------- /core/SGDClient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/SGDClient.py -------------------------------------------------------------------------------- /core/SGDClient_hr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/SGDClient_hr.py -------------------------------------------------------------------------------- /core/SVRGClient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/SVRGClient.py -------------------------------------------------------------------------------- /core/SVRGClient_hr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/SVRGClient_hr.py -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/__init__.py -------------------------------------------------------------------------------- /core/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/function.py -------------------------------------------------------------------------------- /core/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/core/test.py -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- 1 | # Data 2 | 3 | MNIST & CIFAR-10 datasets will be downloaded automatically by the torchvision package. 4 | -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/cifar/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/mnist/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /figs/fig1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/figs/fig1.png -------------------------------------------------------------------------------- /figs/fig2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/figs/fig2.png -------------------------------------------------------------------------------- /figs/fig3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/figs/fig3.png -------------------------------------------------------------------------------- /figs/fig31.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/figs/fig31.png -------------------------------------------------------------------------------- /figs/fig4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/figs/fig4.png -------------------------------------------------------------------------------- /figs/fig_algo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/figs/fig_algo.png -------------------------------------------------------------------------------- /main_hr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/main_hr.py -------------------------------------------------------------------------------- /main_hr_joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/main_hr_joint.py -------------------------------------------------------------------------------- /main_imbalance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/main_imbalance.py -------------------------------------------------------------------------------- /minmax_synthetic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/minmax_synthetic.ipynb -------------------------------------------------------------------------------- /models/ModelBuilder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/models/ModelBuilder.py -------------------------------------------------------------------------------- /models/Nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/models/Nets.py -------------------------------------------------------------------------------- /models/SvrgUpdate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/models/SvrgUpdate.py -------------------------------------------------------------------------------- /models/Update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/models/Update.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/models/__init__.py -------------------------------------------------------------------------------- /output/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reproduce/fig2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/reproduce/fig2.py -------------------------------------------------------------------------------- /reproduce/fig2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/reproduce/fig2.sh -------------------------------------------------------------------------------- /reproduce/fig3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/reproduce/fig3.py -------------------------------------------------------------------------------- /reproduce/fig3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/reproduce/fig3.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/Fed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/Fed.py -------------------------------------------------------------------------------- /utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/README.md -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/dataset_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/dataset_normal.py -------------------------------------------------------------------------------- /utils/my_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/my_logging.py -------------------------------------------------------------------------------- /utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/options.py -------------------------------------------------------------------------------- /utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/sampling.py -------------------------------------------------------------------------------- /utils/svrg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ucr-optml/FedNest/HEAD/utils/svrg.py --------------------------------------------------------------------------------