一种基于联邦学习的模型训练方法及系统

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于联邦学习的模型训练方法及系统
申请号:CN202411544459
申请日期:2024-10-31
公开号:CN119378651B
公开日期:2025-07-11
类型:发明专利
摘要
本发明公开了一种基于联邦学习的模型训练方法及系统。该方法包括:通过中心服务器依据目标数据集的类别总量、客户端数量及基础客户端的数据类别量得到基础客户端的子训练集;通过目标客户端依据目标子训练集分部梯度更新训练基础神经网络,得到待选提取层参数、目标特征识别层、目标类平均激活向量及目标类威布尔概率分布;通过中心服务器平均计算全量待选提取层参数得到目标提取层参数,结合预设元分类器及全量目标特征识别层构建初始全局模型;通过目标客户端基于目标子训练集、目标类平均激活向量及目标类威布尔概率分布训练初始全局模型,得到目标全局模型。通过本发明的技术方案,能够实现异构数据场景的模型训练,提高了模型的准确性。
技术关键词
客户端 中心服务器 基础 训练集 参数 模型训练方法 数据 训练场景 标签 分类器 模型训练系统 总量 列表 异构 校验规则 样本