代码之家 › 专栏 › 技术社区 › Amit Portnoy

我可以在tensorflow federated中聚合渐变吗?

tensorflow-federated

0

Amit Portnoy · 技术社区 · 7 年前

目前,tensorflow的联邦_学习似乎只包括对模型可训练变量起作用的联邦_平均。我将如何实现需要在服务器上进行聚合的梯度算法?

谢谢

0 回复 | 直到 7 年前

1

2

Zachary Garrett 7 年前

释放时 0.2.0 ,TensorFlow Federated包含FedSGD的一个实现( tff.learning.build_federated_sgd_process() )如论文所述:

从分散的数据中高效地学习深度网络 布伦丹·麦克马汉、艾德·摩尔、丹尼尔·拉梅、塞斯·汉普森、布莱斯·阿奎拉和阿尔卡斯。2017年美国统计局。 https://arxiv.org/abs/1602.05629

代码可以在中找到 tensorflow_federated/python/learning/federated_sgd.py ,它显示在客户端聚合渐变,并将聚合的渐变发送回服务器。