FlexiFed-repeat-student

2023-7月，FlexiFed联邦学习实验复现源码-更新中
论文原文：3543507.3583347
复现实验作者：周臻鹏
HUST-iot2001
查看运行方式

实验说明

本仓库复现了论文"FlexiFed: Personalized Federated Learning for Edge Clients with Heterogeneous Model Architectures"中Table1部分的实验结果

配置

硬件环境

• GPU RTX 3080 10G
• CPU 12核
• 40G内存

软件环境

• ubuntu==9.4.1
• jupyter lab
• torch=1.11.0+cu113
• torchvision=0.12.0+cu113
• python==3.8
• h5py=3.9.0
• librosa=0.10.0.post2
• matplotlib=3.5.2

实验对象

数据集

• 图像-CIFAR-10
• 图像CINIC-10
• 音频-Speech_Commands
• 文本-Ag News

模型

• VGG Family
• ResNet Family
• CharCNN Family
• VDCNN Family

FlexiFed框架

• Basic-Common
• Clustered-Common
• Max-Common

实验运行方式

参考仓库目录结构如下图,按步骤运行:

• 根据requirements.txt下载所需模块
• 下载所需数据集到本地
• 运行fed-main.py,通过运行目录名判断当前训练的数据集,模型和聚合方法
• 运行完成之后,得到table.txt用于绘图
• 将运行目录作为输入运行view_results/drawer.py,得到acc曲线图像
• 可以在三种策略均训练完成之后运行drawer_compare.py比较不同方法之间的效果差异

目录结构

FlexiFed-repeat-student
│ README.md
│ 目录结构.txt
│
├─FlexiFed-main
│ │ fed-main.py
│ │ requirements.txt
│ │ utils.py
│ │ 任务说明.txt
│ │
│ ├─datasets
│ │ deldataset.py
│ │
│ ├─model_family
│ │ CharCNN.py
│ │ ResNet.py
│ │ VDCNN.py
│ │ vgg.py
│ │
│ └─view_results
│ drawer.py
│ drawer_compare.py
│ view_model.py
│
└─数据&报告
 FlexiFed 实验复现结果数据汇总.xlsx
 FlexiFed 实验复现结果曲线图汇总.pdf
 实验结果曲线图汇总-8.1.pdf
 生产实习汇报小结-7.3-7.13-周臻鹏.pdf
 生产实习汇报小结-第二、三周-周臻鹏.pdf
 生产实习汇报小结-第四周&总结-周臻鹏.docx
 生产实习汇报小结-第四周&总结-周臻鹏.pdf
 第三周实验复现结果曲线图汇总-共12项.pdf