发明公开
- 专利标题: 一种基于联邦学习模型的训练方法
-
申请号: CN202310971765.9申请日: 2023-08-03
-
公开(公告)号: CN117035058A公开(公告)日: 2023-11-10
- 发明人: 刘睿霖 , 程娇 , 杜金浩 , 张震 , 石瑾 , 高圣翔 , 刘发强
- 申请人: 国家计算机网络与信息安全管理中心
- 申请人地址: 北京市朝阳区裕民路甲3号
- 专利权人: 国家计算机网络与信息安全管理中心
- 当前专利权人: 国家计算机网络与信息安全管理中心
- 当前专利权人地址: 北京市朝阳区裕民路甲3号
- 代理机构: 北京精翰专利代理有限公司
- 代理商 卓邦荣
- 主分类号: G06N3/098
- IPC分类号: G06N3/098 ; G06N3/04 ; G06N20/00 ; G06F21/60
摘要:
本发明属于联邦学习领域,提供了一种基于联邦学习模型的训练方法,包括以下步骤:S11,定义问题:确定需要解决的机器学习问题、本地数据的来源、以及参与联邦学习的设备或节点;S12,模型选择和初始化:选择相应的模型,并在所有的本地设备或节点上初始化相应的模型,并下发至所有用户端;S13,本地训练:每个本地设备或节点使用其本地数据集对初始化的模型进行训练,得到一个本地模型;S14,模型聚合:在中央服务器上聚合本地模型;本发明通过在每个本地设备或节点都可以进行本地模型的训练和更新,进一步分散计算负载,提高训练速度和效率;通过设定停止条件来控制模型更新的频率,进而避免过度拟合等问题。