联邦学习——学习笔记1：FedAvg算法

---

本笔记参考自b站up主：丸一口

原视频链接

如上图，现有6个医院：眼科、儿科、妇科、骨科、综合医院1、综合医院2。中间节点为政府。

现政府要求用各个医院的数据训练某个模型，希望对某些疾病进行一些预测，数据如下图：

如果是传统方法，那就是六个医院直接把数据发送到政府处，然后政府那边做机器学习。

如果用联邦学习，那政府是不能直接接收其他医院的所有数据的，因为如果直接接收数据，会有一个严重问题，就是隐私泄露的问题。因为传数据的过程中可能会被拦截，并且大量数据传送过程中对带宽也是有较高要求。

联邦学习的具体解决方案：中心节点初始化一个模型，把这个初始模型发给其他节点，其他节点在自己的地方做训练，训练结束后，把训练好的模型传到中心节点。