基于联合学习的模型训练方法及系统

授权

摘要

本申请实施例公开了一种基于联合学习的模型训练方法及系统，属于机器学习和数据聚合技术领域。所述方法包括：每个客户端对局部模型进行训练，将得到的局部模型参数发送给相连的边缘设备；边缘设备将局部模型参数发送给服务器，确定训练速度不满足预设条件的第一客户端和训练速度满足预设条件的第二客户端，并获取各个第一客户端中的模型信息和训练数据；服务器对所有局部模型参数进行聚合更新，若更新后的局部模型参数由第一客户端训练得到，则由边缘设备代替第一客户端根据训练更新后的局部模型；若更新后的局部模型参数由第二客户端训练得到，则由第二客户端训练更新后的局部模型。本申请实施例可以提高全局模型训练的成功率。

基本信息

专利标题：

基于联合学习的模型训练方法及系统

专利标题（英）：

暂无

公开（公告）号：

CN112261137A

申请号：

CN202011136661.9

公开（公告）日：

2021-01-22

申请日：

2020-10-22

授权号：

CN112261137B

授权日：

2022-06-14

发明人：

王堃

申请人：

江苏禹空间科技有限公司

申请人地址：

江苏省南京市建邺区嘉陵江东街50号康缘智慧港C座1101室

代理机构：

北京德崇智捷知识产权代理有限公司

代理人：

黄雪

优先权：

CN202011136661.9

主分类号：

H04L29/08

IPC分类号：

H04L29/08 H04L9/08 H04L29/06 G06N20/00