Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
liyu98 committed May 7, 2022
1 parent 5a6c874 commit ec07df1
Show file tree
Hide file tree
Showing 2 changed files with 35 additions and 9 deletions.
44 changes: 35 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,35 +2,61 @@
A simple version of the horizontal federated learning project to achieve image classification learning. This project is simulated locally, and does not involve network communication details and failure processing. It only involves model aggregation functions, which can quickly verify the relevant algorithms and capabilities of federated learning.

### Operating Manual

#### Simulation Ensemble Run(fl_integration)

```shell
python3 fl_integration.py
```

#### Simulation Ensemble Run(fl_integration)

It will define a server object and multiple client objects respectively to simulate horizontal federation training scenarios.


#### Server(fl_server)

Perform model aggregation on the local model uploaded by the selected client.



#### Client(fl_client)

Receive commands and global models from the server, and use local data to train local models.


##### Functional screenshot

<img src="./doc/img/20220507-024348.jpg" width="930">

<img src="./doc/image/20220508-022016.jpg" width="930">

##### Results screenshot

<img src="./doc/img/acc.jpg" width="930">
<img src="./doc/image/acc.jpg" width="930">

<img src="./doc/image/loss.jpg" width="930">

_References: Federated Learning in Action_

---

### 操作手册

#### 联邦学习模拟运行集成(fl_integration)
定义一个服务端对象和多个客户端对象,用来模拟横向联邦训练场景。
```shell
python3 fl_integration.py
```

#### 联邦学习服务端(fl_server)
将被选择的客户端上传的本地模型进行模型聚合。

#### Client(fl_client)
接收服务端的下发指令和全局模型,并利用本地数 据进行局部模型训练。

##### 功能展示

<img src="./doc/image/20220508-022016.jpg" width="930">

##### 结果展示

<img src="./doc/image/acc.jpg" width="930">

<img src="./doc/img/loss.jpg" width="930">
<img src="./doc/image/loss.jpg" width="930">

---
_参考资料:联邦学习实战_
Binary file added doc/image/20220508-022016.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit ec07df1

Please sign in to comment.