1.一种联邦学习网络下的模型训练方法,其特征在于,包括下述步骤:建立联邦学习网络,所述联邦学习网络包括中央客户端和多个节点,控制每个所述节点接收中央客户端下发的初始化模型,作为本地模型,每个节点分别对本地模型进行多轮更新训练;
直至更新训练后各节点对应的本地模型收敛,各节点分别获得结果模型;
控制所述节点接收用户数据,并输入到所述节点对应的所述结果模型中,获得所述结果模型输出的推荐信息;所述用户数据包括用户购买力、用户个人喜好和产品特征;
其中,在每轮更新训练中,所述更新训练的过程包括:
控制每个所述节点使用节点对应的本地数据对所述本地模型进行训练,获得每个节点的梯度信息,并将所述梯度信息发送至所述中央客户端;
控制所述中央客户端接收并根据所述梯度信息生成全局信息,将所述全局信息发送至各节点;
控制当前节点接收并根据所述全局信息获得其他节点的梯度信息,分别使用每个节点的梯度信息对当前节点的本地模型进行测试,获得准确率,根据所述准确率调整接收到的全局信息,获得调整后的全局信息,并使用所述调整后的全局信息更新当前节点的本地模型;以及直至完成当前轮全部节点更新训练,判断各节点对应的本地模型是否收敛。
2.根据权利要求1所述的联邦学习网络下的模型训练方法,其特征在于,根据所述准确率调整接收到的全局信息,获得调整后的全局信息的步骤包括:根据所述准确率获得每个节点的梯度信息在全局信息中的权重;
将所述权重和梯度信息进行加权求和,获得调整后的全局信息。
3.根据权利要求2所述的联邦学习网络下的模型训练方法,其特征在于,根据所述准确率获得每个节点的梯度信息在全局信息中的权重的步骤包括:根据所述准确率计算准确率中间值,其中,所述准确率中间值为各准确率的中位数;
通过如下公式计算各节点的梯度信息的权重:
;
其中, 为各节点的梯度信息的权重, 为上一轮的各节点梯度信息的权重,为学习率, 为各节点的准确率, 为准确率中间值。
4.根据权利要求1所述的联邦学习网络下的模型训练方法,其特征在于,所述本地数据由训练数据和验证集数据组成,所述分别使用每个节点的梯度信息对当前节点的本地模型进行测试,获得准确率的步骤包括:分别使用每个节点的梯度信息和验证集对当前节点的本地模型进行测试,获得准确率。
5.根据权利要求1所述的联邦学习网络下的模型训练方法,其特征在于,所述本地数据由训练数据和验证集数据组成,所述控制每个所述节点使用节点对应的本地数据对所述本地模型进行训练,获得每个节点的梯度信息的步骤包括:控制每个所述节点使用训练数据对所述本地模型进行训练,获得每个节点的梯度信息。
6.根据权利要求1至5任意一项所述的联邦学习网络下的模型训练方法,其特征在于,所述将所述梯度信息发送至所述中央客户端的步骤包括:将所述梯度信息使用所述中央客户端预先传输的公钥进行加密;
将所述加密后的梯度信息发送至所述中央客户端;
所述控制所述中央客户端接收并根据所述梯度信息生成全局信息的步骤包括:控制所述中央客户端解密所述加密后的梯度信息,获得梯度信息;
根据所述梯度信息生成全局信息。
7.根据权利要求1至5任意一项所述的联邦学习网络下的模型训练方法,其特征在于,所述将所述梯度信息发送至所述中央客户端的步骤包括:将所述梯度信息使用所述中央客户端预先传输的对称密钥进行加密;
将所述加密后的梯度信息发送至所述中央客户端;
所述控制当前节点接收并根据所述全局信息获得其他节点的梯度信息的步骤包括:控制当前节点接收所述全局信息;
根据所述全局信息获得加密后的梯度信息;
使用对称密钥解密所述加密后的梯度信息,获得梯度信息。
8.一种联邦学习网络下的模型训练装置,其特征在于,包括:建立模块,用于建立联邦学习网络,所述联邦学习网络包括中央客户端和多个节点,控制每个所述节点接收中央客户端下发的初始化模型,作为本地模型,每个节点分别对本地模型进行多轮更新训练;
获得模块,用于直至更新训练后各节点对应的本地模型收敛,各节点分别获得结果模型;
输出模块,用于控制所述节点接收用户数据,并输入到所述节点对应的所述结果模型中,获得所述结果模型输出的推荐信息;所述用户数据包括用户购买力、用户个人喜好和产品特征;
所述建立模块包括训练子模块、生成子模块、调整子模块和判断子模块;
其中,所述训练子模块,用于在每轮更新训练中,控制每个所述节点使用节点对应的本地数据对所述本地模型进行训练,获得每个节点的梯度信息,并将所述梯度信息发送至所述中央客户端;
所述生成子模块,用于在每轮更新训练中,控制所述中央客户端接收并根据所述梯度信息生成全局信息,将所述全局信息发送至各节点;
所述调整子模块,用于在每轮更新训练中,控制当前节点接收并根据所述全局信息获得其他节点的梯度信息,分别使用每个节点的梯度信息对当前节点的本地模型进行测试,获得准确率,根据所述准确率调整每个节点的梯度信息在全局信息中的权重,获得调整后的全局信息,并使用所述调整后的全局信息更新当前节点的本地模型;以及所述判断子模块,用于直至完成当前轮全部节点更新训练,判断各节点对应的本地模型是否收敛。
9.一种计算机设备,其特征在于,包括存储器和处理器,所述存储器中存储有计算机可读指令,所述处理器执行所述计算机可读指令时实现如权利要求1至7中任一项所述的联邦学习网络下的模型训练方法的步骤。
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机可读指令,所述计算机可读指令被处理器执行时实现如权利要求1至7中任一项所述的联邦学习网络下的模型训练方法的步骤。