TensorFlow.js中的联合学习
这不是TensorFlow的官方联合学习框架。 这是TensorFlow.js的实验性库,目前尚未维护。 如果您想使用官方的联合学习库,请查看 。
这是中(仅用于实验和演示)实现的父存储库。 联合学习是一种以分布式方式训练机器学习模型的方法。 尽管它涉及一台中央服务器,但该服务器无需查看任何数据甚至无需计算梯度。 相反,客户端在本地执行所有推理和训练(他们已经在Tensorflow.js中进行了训练),并且只是定期向服务器发送更新后的权重(而不是数据)。 服务器的唯一工作就是聚合和重新分发它们,这意味着它可以非常轻巧!
基本用法
在服务器(NodeJS)端:
import * as http from 'http' ;
import * as federated from 'federated-learning-server' ;
const I
1