[发明专利]联邦学习方法及装置在审
申请号: | 202111436383.3 | 申请日: | 2021-11-29 |
公开(公告)号: | CN114117920A | 公开(公告)日: | 2022-03-01 |
发明(设计)人: | 王美娟 | 申请(专利权)人: | 新华三技术有限公司 |
主分类号: | G06F30/27 | 分类号: | G06F30/27;G06N20/00 |
代理公司: | 北京博思佳知识产权代理有限公司 11415 | 代理人: | 王剑 |
地址: | 310052 浙*** | 国省代码: | 浙江;33 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 联邦 学习方法 装置 | ||
1.一种联邦学习方法,其特征在于,应用于联邦系统包括的服务器,所述联邦系统还包括多个客户端,所述方法包括:
接收目标客户端发送的参数上传请求,所述参数上传请求包括所述目标客户端本地模型训练后提取的第一模型参数;
本地记录所述目标客户端的第一模型参数;
对已记录的各客户端的第一模型参数进行聚合,得到第二模型参数;
向所述目标客户端下发所述第二模型参数,以使所述目标客户端基于所述第二模型参数继续训练本地模型。
2.如权利要求1所述的方法,其特征在于,所述向所述目标客户端下发所述第二模型参数,包括:
接收所述目标客户端发送的参数更新请求;
基于接收到的参数更新请求向所述目标客户端下发所述第二模型参数。
3.如权利要求2所述的方法,其特征在于,所述服务器预先配置有用于接收参数上传请求的第一端口,以及用于接收参数更新请求的第二端口,所述接收目标客户端发送的参数上传请求,包括:
通过监听所述第一端口,接收所述目标客户端发送的参数上传请求;
所述接收所述目标客户端发送的参数更新请求,包括:
通过监听所述第二端口,接收所述目标客户端发送的参数更新请求。
4.如权利要求1所述的方法,其特征在于,所述对已记录的各客户端的第一模型参数进行聚合,得到第二模型参数,包括:
对已记录的各客户端的第一模型参数进行加权平均,得到所述第二模型参数。
5.如权利要求1所述的方法,其特征在于,所述方法还包括:
记录所述第一模型参数的接收时间;
如果当前时间与所述接收时间的差值大于预设差值阈值,删除所述第一模型参数。
6.一种联邦学习方法,其特征在于,应用于联邦系统包括的客户端,所述联邦系统还包括服务器,所述方法包括:
通过本地模型训练,提取训练后模型的第一模型参数;
向所述服务器发送参数上传请求,所述参数上传请求包括所述第一模型参数,以使所述服务器本地记录所述客户端的第一模型参数,并对已记录的各客户端的第一模型参数进行聚合,得到第二模型参数;
向所述服务器发送参数更新请求,以使所述服务器基于所述参数更新请求向本客户端下发第二模型参数;
基于所述第二模型参数继续训练本地模型。
7.如权利要求6所述的方法,其特征在于,所述向所述服务器发送参数更新请求,包括:
在发送所述参数上传请求之后,开始计时;
当计时时间达到预设时间阈值时,向所述服务器发送所述参数更新请求。
8.一种联邦学习装置,其特征在于,应用于联邦系统包括的服务器,所述联邦系统还包括多个客户端,所述装置包括:
接收单元,用于接收目标客户端发送的参数上传请求,所述参数上传请求包括所述目标客户端本地模型训练后提取的第一模型参数;
记录单元,用于本地记录所述目标客户端的第一模型参数;
聚合单元,用于对已记录的各客户端的第一模型参数进行聚合,得到第二模型参数;
下发单元,用于向所述目标客户端下发所述第二模型参数,以使所述目标客户端基于所述第二模型参数继续训练本地模型。
9.如权利要求8所述的装置,其特征在于:
所述接收单元,还用于接收所述目标客户端发送的参数更新请求;
所述下发单元,具体用于基于接收到的参数更新请求向所述目标客户端下发所述第二模型参数。
10.一种联邦学习装置,其特征在于,应用于联邦系统包括的客户端,所述联邦系统还包括服务器,所述装置包括:
提取单元,用于通过本地模型训练,提取训练后模型的第一模型参数;
发送单元,用于向所述服务器发送参数上传请求,所述参数上传请求包括所述第一模型参数,以使所述服务器本地记录本客户端的第一模型参数,并对已记录的各客户端的第一模型参数进行聚合,得到第二模型参数;
所述发送单元,还用于向所述服务器发送参数更新请求,以使所述服务器基于所述参数更新请求向本客户端下发第二模型参数;
训练单元,用于基于所述第二模型参数继续训练本地模型。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于新华三技术有限公司,未经新华三技术有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202111436383.3/1.html,转载请声明来源钻瓜专利网。