
基于PyTorch的pFedMe个性化联邦学习方法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究提出了一种基于PyTorch框架的pFedMe算法,旨在通过个性化联邦学习提升模型在不同数据分布下的适应性与性能。
该存储库实现了论文《使用Moreau信封进行个性化联合学习》(NeurIPS 2020)中的所有实验内容。作者为Canh T. Dinh、Nguyen H. Tran 和Tuan Dung Nguyen。
除了实现pFedMe算法外,本项目还实现了FedAvg和Per-FedAvg算法,并使用Pytorch进行联合学习。
软件需求包括:numpy, scipy, torch, pillow以及matplotlib。要下载所需依赖项,请运行命令`pip3 install -r requirements.txt`。
数据集方面,我们采用了两个数据集:MNIST 和 Synthetic。为了生成非均匀分布的MNIST数据,请访问data/Mnist文件夹并执行以下操作:python3 generate_niid_20users.py
全部评论 (0)
还没有任何评论哟~


