Skip to content

Latest commit

 

History

History
43 lines (26 loc) · 1.3 KB

File metadata and controls

43 lines (26 loc) · 1.3 KB

End2end-Federated-Learning

介绍

联邦学习虽然受到了大量关注,但已有工作基本上基于仿真测试,没有一个完整的系统供大家研究。为此我们搭了一个端到端的跨终端联邦学习系统,并在20台真实安卓设备上做了个验证(demo如下),代码将积极维护和优化,希望可以对社区有所帮助。

demo演示

动画

运行

首先保证服务端和客户端在同一个局域网下:

服务端

1.安装python3.6、pip3,并运行以下依赖包

pip3 install -U MNN
pip3 install cherrypy
pip3 install ws4py

2.运行server.py

客户端

1.数据和初始化模型通过命令行工具adb下载到Android设备本地

cd End2end-Federated-Learning/data
adb push mnist.snapshot.mnn /data/local/tmp/mnn/mnist.snapshot.mnn
adb push mnist_data /data/local/tmp/mnist_data

2.修改app/src/main/java/com/demo/MainActivity.java中的SERVER_URL为服务端的ip地址

3.连接android设备,并运行项目(必须和服务端在同一个局域网下才能正常运行)

联系我们

徐梦炜 (mwx@bupt.edu.cn), 王尚广 (sgwang@bupt.edu.cn)