联邦学习模型的训练方法及装置
基本信息
申请号 | CN202010651671.X | 申请日 | - |
公开(公告)号 | CN111950740A | 公开(公告)日 | 2020-11-17 |
申请公布号 | CN111950740A | 申请公布日 | 2020-11-17 |
分类号 | G06N20/20 | 分类 | 计算;推算;计数; |
发明人 | 夏家骏;鲁颖;张珣;沈敏均;陈楚元;张佳辰 | 申请(专利权)人 | 光之树(北京)科技有限公司 |
代理机构 | 北京清亦华知识产权代理事务所(普通合伙) | 代理人 | 光之树(北京)科技有限公司 |
地址 | 100125 北京市朝阳区光明大厦B座0307 | ||
法律状态 | - |
摘要
摘要 | 本申请公开了一种联邦学习模型的训练方法及装置,执行主体为目标节点,该方法包括:在联邦学习模型的训练的过程中,获取处于掉线状态的第一训练节点,从本地缓存中获取第一训练节点的替代中间结果秘密份额;根据替代中间结果秘密份额,获取第一训练节点在目标节点上的第一损失函数秘密份额;接收处于在线状态的剩余参与节点发送的第二中间结果秘密份额,并根据第二中间结果秘密份额,获取剩余参与节点在目标节点上的第二损失函数秘密份额;根据第一梯度秘密份额、第二梯度秘密份额和目标节点自身的损失函数,获取联邦学习模型的损失函数。由此,若训练节点掉线,目标节点从本地缓存中获取替代中间结果秘密份额继续训练,减小节点掉线导致的影响。 |
