拆分学习#

拆分学习#

拆分学习的概念#

拆分学习的核心思想是拆分网络结构。每一个参与方拥有模型结构的一部分,所有参与方的模型合在一起形成一个完整的模型。训练过程中,不同参与方只对本地模型进行正向或者反向传播计算,并将计算结果传递给下一个参与方。多个参与方通过联合模型进行训练直至最终收敛。

一个典型的拆分学习例子:

Alice持有数据和基础模型。Bob只有数据、基础模型和fuse模型。

Alice使用自己的数据和基础模型得到 hidden0,然后发送给Bob。

Bob使用自己的数据和基础模型得到 hidden1。

Agg Layer使用 hidden_0 和 hidden_1 作为输入,并输出聚合后的隐层。

Bob把聚合后的隐层作为fuse模型的输入,计算得到梯度。

梯度被拆分成两部分,分别返回给Alice和Bob。

Alice和Bob使用各自收到的梯度更新基础模型。

相关推荐

征途LandCross品牌介绍
365bet365体育在线

征途LandCross品牌介绍

09-07 阅读 5901
在线调整图像色调
日博官网365bet

在线调整图像色调

10-05 阅读 1554