联邦学习模型的训练方法及装置

基本信息

申请号 CN202010564686.2 申请日 -
公开(公告)号 CN111860829A 公开(公告)日 2020-10-30
申请公布号 CN111860829A 申请公布日 2020-10-30
分类号 G06N3/08(2006.01)I;G06N20/20(2019.01)I 分类 计算;推算;计数;
发明人 夏家骏;鲁颖;张珣;沈敏均;陈楚元;张佳辰 申请(专利权)人 光之树(北京)科技有限公司
代理机构 北京清亦华知识产权代理事务所(普通合伙) 代理人 光之树(北京)科技有限公司
地址 100125北京市朝阳区光明大厦B座0307
法律状态 -

摘要

摘要 本申请公开了一种联邦学习模型的训练方法及装置,执行主体为参与联邦学习模型训练的目标训练节点,其中,目标训练节点为任意一个参与联邦学习模型训练的训练节点,该方法包括:获取所述目标训练节点上第一学习模型的梯度秘密份额;根据所述梯度秘密份额,恢复所述第一学习模型的梯度信息;根据所述梯度信息更新所述第一学习模型的模型参数,并重新对更新后的所述第一学习模型进行训练。本申请在半诚实假设的前提下,基于秘密共享算法对联邦学习模型进行训练,确保了数据安全、提高了模型训练有效性、缩短了运算耗时。