├── .gitignore ├── README.md ├── experiments ├── FedAvg │ ├── Models.py │ ├── __init__.py │ └── fedAvg_trainer.py ├── STC │ ├── __init__.py │ ├── main.py │ ├── models.py │ └── utils.py ├── __init__.py ├── dataset.py ├── lstm_shakespeare │ ├── __init__.py │ ├── main.py │ ├── models.py │ └── utils.py ├── pfedhn │ ├── __init__.py │ ├── models.py │ ├── node.py │ └── trainer.py ├── pfedhn_pc │ ├── __init__.py │ ├── models.py │ ├── node.py │ └── trainer.py └── utils.py ├── note ├── FedAvg算法+LSTM模型+ Shakespeare数据集——字符预测任务.pdf ├── FedAvg算法复现.pdf ├── STC稀疏三元压缩算法复现.pdf ├── 《Communication-Efficient Learning of Deep Networks from Decentralized Data》论文阅读报告.pdf └── 《Robust and Communication-Efficient Federated Learning From Non-i.i.d. Data》阅读报告.pdf ├── requirements.txt ├── resources ├── pfedhn.png └── pfedhn_arch.png └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/README.md -------------------------------------------------------------------------------- /experiments/FedAvg/Models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/FedAvg/Models.py -------------------------------------------------------------------------------- /experiments/FedAvg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/FedAvg/fedAvg_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/FedAvg/fedAvg_trainer.py -------------------------------------------------------------------------------- /experiments/STC/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/STC/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/STC/main.py -------------------------------------------------------------------------------- /experiments/STC/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/STC/models.py -------------------------------------------------------------------------------- /experiments/STC/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/STC/utils.py -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/dataset.py -------------------------------------------------------------------------------- /experiments/lstm_shakespeare/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/lstm_shakespeare/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/lstm_shakespeare/main.py -------------------------------------------------------------------------------- /experiments/lstm_shakespeare/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/lstm_shakespeare/models.py -------------------------------------------------------------------------------- /experiments/lstm_shakespeare/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/lstm_shakespeare/utils.py -------------------------------------------------------------------------------- /experiments/pfedhn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/pfedhn/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/pfedhn/models.py -------------------------------------------------------------------------------- /experiments/pfedhn/node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/pfedhn/node.py -------------------------------------------------------------------------------- /experiments/pfedhn/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/pfedhn/trainer.py -------------------------------------------------------------------------------- /experiments/pfedhn_pc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/pfedhn_pc/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/pfedhn_pc/models.py -------------------------------------------------------------------------------- /experiments/pfedhn_pc/node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/pfedhn_pc/node.py -------------------------------------------------------------------------------- /experiments/pfedhn_pc/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/pfedhn_pc/trainer.py -------------------------------------------------------------------------------- /experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/experiments/utils.py -------------------------------------------------------------------------------- /note/FedAvg算法+LSTM模型+ Shakespeare数据集——字符预测任务.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/note/FedAvg算法+LSTM模型+ Shakespeare数据集——字符预测任务.pdf -------------------------------------------------------------------------------- /note/FedAvg算法复现.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/note/FedAvg算法复现.pdf -------------------------------------------------------------------------------- /note/STC稀疏三元压缩算法复现.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/note/STC稀疏三元压缩算法复现.pdf -------------------------------------------------------------------------------- /note/《Communication-Efficient Learning of Deep Networks from Decentralized Data》论文阅读报告.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/note/《Communication-Efficient Learning of Deep Networks from Decentralized Data》论文阅读报告.pdf -------------------------------------------------------------------------------- /note/《Robust and Communication-Efficient Federated Learning From Non-i.i.d. Data》阅读报告.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/note/《Robust and Communication-Efficient Federated Learning From Non-i.i.d. Data》阅读报告.pdf -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/requirements.txt -------------------------------------------------------------------------------- /resources/pfedhn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/resources/pfedhn.png -------------------------------------------------------------------------------- /resources/pfedhn_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/resources/pfedhn_arch.png -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhongjian-zhang/FederalLearning/HEAD/setup.py --------------------------------------------------------------------------------