pFedMe:使用Pytorch(pFedMe)进行个性化联合学习

时间:2024-05-23 20:02:53
【文件属性】:

文件名称:pFedMe:使用Pytorch(pFedMe)进行个性化联合学习

文件大小:10.89MB

文件格式:ZIP

更新时间:2024-05-23 20:02:53

pytorch federated-learning federated-meta-learning personalized-federated-learning Python

使用Moreau信封进行个性化联合学习(NeurIPS 2020) 该存储库实施了用Moreau信封进行个性化联合学习一文中的所有实验。 作者:Canh T. Dinh,Nguyen H. Tran,Tuan Dung Nguyen 全文: : 纸张已被NeurIPS 2020接受。 该存储库不仅实现pFedMe,还实现FedAvg和Per-FedAvg算法。 (使用Pytorch进行联合学习) 软件要求: numpy,scipy,火炬,枕头,matplotlib。 要下载依赖项: pip3 install -r requirements.txt 数据集:我们使用2个数据集:MNIST和Synthetic 要生成非空闲MNIST数据,请执行以下操作: 访问数据/ Mnist并运行:“ python3 generate_niid_20users.py” 我们可以使用


【文件预览】:
pFedMe-master
----main.py(5KB)
----FLAlgorithms()
--------optimizers()
--------servers()
--------users()
--------trainmodel()
----data()
--------Synthetic()
--------Mnist()
--------Cifar10()
----.github()
--------ISSUE_TEMPLATE()
----main_plot.py(2KB)
----results()
--------Mnist_pFedMe_0.005_1.0_15_5u_20b_10_5_0.09_avg.h5(21KB)
--------Mnist_pFedMe_p_0.005_1_15_5u_20b_20_5_0.1_avg.h5(21KB)
----requirements.txt(52B)
----.gitignore(2KB)
----README.md(10KB)
----utils()
--------model_utils.py(14KB)
--------plot_utils.py(34KB)
--------__init__.py(0B)

网友评论