一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

横向联邦学习方法、装置、计算机设备和存储介质与流程

2021-10-20 00:02:00 来源:中国专利 TAG:人工智能 学习方法 横向 联邦 装置

技术特征:
1.一种横向联邦学习方法,其特征在于,所述横向联邦学习方法基于横向联邦学习系统实现,所述横向联邦学习系统包括主节点以及与所述主节点相连接的至少两个参与者节点,包括:所述参与者节点从所述主节点上下载机器学习模型,利用本地数据对所述机器学习模型进行训练,得到训练后的模型参数;所述参与者节点对所述训练后的模型参数进行同态加密,将加密数据发送给所述主节点;所述主节点对所述加密数据进行同态求平均处理,把经过同态求平均处理的数据发送给各所述参与者节点;所述参与者节点对所述经过同态求平均处理的数据进行解密,得到解密后数据;所述参与者节点基于预设规则对所述解密后数据进行运算,并根据运算结果去除干扰数据,得到模型更新参数;所述参与者节点利用所述模型更新参数更新所述机器学习模型。2.如权利要求1所述的横向联邦学习方法,其特征在于,所述机器学习模型为梯度提升决策树模型,所述梯度提升决策树模型包括树节点,所述利用本地数据对所述机器学习模型进行训练包括:每个所述参与者节点分别获取所述树节点的节点训练特征集以及所述梯度提升决策树模型的特征矩阵的梯度数据;每个所述参与者节点分别针对所述树节点,根据所述节点训练特征集和所述梯度数据确定出局部最佳分裂规则并发送给所述主节点。3.如权利要求1所述的横向联邦学习方法,其特征在于,所述利用本地数据对所述机器学习模型进行训练包括:将所述机器学习模型转换为计算图;将所述本地数据输入到所述计算图中进行训练操作;将训练完成的计算图转换为训练完成的机器学习模型。4.如权利要求1所述的横向联邦学习方法,其特征在于,所述参与者节点对训练后的模型参数进行同态加密,包括:所述参与者节点利用自身的随机噪声对需要进行加密的训练后的模型参数进行编码,得到编码数据;其中,所述编码数据中包括所述随机噪声以及添加了干扰数据的训练后的模型参数,所述干扰数据是基于预设规则对所述随机噪声进行运算后获得的。5.如权利要求4所述的横向联邦学习方法,其特征在于,所述参与者节点对训练后的模型参数进行同态加密,包括:所述参与者节点利用自身的随机噪声对需要进行加密的训练后的模型参数进行编码,得到编码数据;其中,所述编码数据中包括所述随机噪声以及添加了干扰数据的训练后的模型参数,所述干扰数据是基于预设规则对所述随机噪声进行运算后获得的;所述参与者节点利用对称密钥对所述编码数据进行加密,得到加密数据。6.如权利要求5所述的横向联邦学习方法,其特征在于,所述干扰数据是基于预设规则对所述随机噪声进行运算后获得的,包括:以各所述参与者节点共享的预设数据对所述随机噪声进行乘运算后获得所述干扰数
据。7.如权利要求5所述的横向联邦学习方法,其特征在于,所述参与者节点利用自身的随机噪声对需要进行加密的训练后的模型参数进行编码,得到编码数据,包括:当所述随机噪声的个数为一个时,将所述预设数据与所述随机噪声相乘得到干扰数据,利用所述干扰数据以及添加了所述干扰数据的训练后的模型参数组成编码数据;当所述随机噪声的个数至少为两个时,将所述预设数据与至少两个所述随机噪声分别相乘,得到至少两个干扰数据;将所述至少两个干扰数据作为第一数据部分;在所述训练后的模型参数中添加所述第一数据部分,得到第二数据部分;利用所述第一数据部分以及所述第二数据部分组成编码数据。8.一种横向联邦学习装置,其特征在于,包括:训练模块,用于参与者节点从主节点上下载机器学习模型,利用本地数据对所述机器学习模型进行训练,得到训练后的模型参数;加密模块,用于所述参与者节点对所述训练后的模型参数进行同态加密,将加密数据发送给所述主节点;计算模块,用于所述主节点对所述加密数据进行同态求平均处理,把经过所述同态求平均处理的数据发送给各所述参与者节点;解密模块,用于所述参与者节点对所述经过同态求平均处理的数据进行解密,得到解密后数据;运算模块,用于所述参与者节点基于预设规则对所述解密后数据进行运算,并根据运算结果去除干扰数据,得到模型更新参数;更新模块,用于所述参与者节点利用所述模型更新参数更新所述机器学习模型。9.一种计算机设备,包括存储器和处理器,所述存储器中存储有计算机可读指令,所述计算机可读指令被所述处理器执行时,使得所述处理器执行如权利要求1至8中任一项权利要求所述横向联邦学习方法的步骤。10.一种存储有计算机可读指令的存储介质,所述计算机可读指令被一个或多个处理器执行时,使得一个或多个处理器执行如权利要求1至8中任一项权利要求所述横向联邦学习方法的步骤。

技术总结
本发明涉及一种横向联邦学习方法、装置、计算机设备和存储介质。该方法包括:参与者节点利用本地数据对机器学习模型进行训练,对训练后的模型参数进行同态加密,将加密数据发送给主节点;主节点对加密数据进行同态求平均处理,把经过处理的数据发送给各参与者节点;参与者节点对经过处理的数据进行解密,基于预设规则对解密后数据进行运算,并根据运算结果去除干扰数据,得到模型更新参数;参与者节点利用模型更新参数更新机器学习模型。本申请的方法,各节点之间进行数据加密传输,主节点不对接收到的加密数据进行解密,在整个横向联邦学习过程中数据传输都是在加密状态下进行的,提高了数据传输过程中的安全性,避免了数据泄露问题。问题。问题。


技术研发人员:徐奇
受保护的技术使用者:平安国际智慧城市科技股份有限公司
技术研发日:2021.05.28
技术公布日:2021/10/19
再多了解一些

本文用于企业家、创业者技术爱好者查询,结果仅供参考。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献

  • 日榜
  • 周榜
  • 月榜