- 专利标题: 基于联邦学习的模型训练方法、装置、介质以及电子设备
-
申请号: CN202310706543.4申请日: 2023-06-15
-
公开(公告)号: CN116432040B公开(公告)日: 2023-09-01
- 发明人: 马平 , 兰春嘉
- 申请人: 上海零数众合信息科技有限公司
- 申请人地址: 上海市浦东新区中国(上海)自由贸易试验区峨山路101号C3幢
- 专利权人: 上海零数众合信息科技有限公司
- 当前专利权人: 上海零数众合信息科技有限公司
- 当前专利权人地址: 上海市浦东新区中国(上海)自由贸易试验区峨山路101号C3幢
- 代理机构: 北京品源专利代理有限公司
- 代理商 王风茹
- 主分类号: G06F18/214
- IPC分类号: G06F18/214 ; G06N3/098 ; G06F21/60 ; G06F21/62
摘要:
本申请实施例公开了一种基于联邦学习的模型训练方法、装置、介质以及电子设备。方法由全局总模型的训练参与方执行,具体包括:利用本地数据集对本地子模型进行训练,以使本地子模型输出本地预测值;在全局总模型的训练参与组内分享本地预测值,并根据训练参与组内其他参与方分享的本地预测值,确定全局总模型的联合预测值;根据联合预测值,确定全局总模型的全局损失值;基于全局总模型的全局损失值,确定本地子模型的最新梯度值,并基于最新梯度值更新本地子模型以训练所述全局总模型。本申请技术方案简化了基于联邦学习的模型训练方法,降低了模型训练方法的计算复杂度和通信复杂度,提高了模型训练方法的实用性。
公开/授权文献
- CN116432040A 基于联邦学习的模型训练方法、装置、介质以及电子设备 公开/授权日:2023-07-14