federated_l 源码

联合学习网 提供一个中央节点和一个客户端节点实现来构建联合学习网络。 中央节点协调由客户端节点执行的一个或多个模型的训练。 每个客户端节点都有自己的数据集以训练本地模型,并将训练结果发送到中心节点,该中心节点将计算节点客户端模型的所有权重的平均值。 客户将在下一轮培训中使用该平均值。 每个客户端节点的本地数据永远不会与任何其他节点共享。 介绍 有两个选项可用于运行网络中的节点,使用Docker为每种类型的节点创建容器,或使用命令行中的标准本地安装。 当然,两者可以混合使用(某些节点作为容器运行,而其他节点通过命令行运行)。 数据集 目前,有两种训练模型:MNIST和胸部X射线。 对于使用MNIST,您不需要安装其他任何东西,因为客户端节点在运行训练时会下载数据集,但是对于Chest X-Ray模型,您将需要一个数据集来使其工作。 从下载数据集,然后将其解压缩到客户端节点计算机上的任何位置