← 返回列表

机器学习方法、装置、设备、联邦学习系统及存储介质

申请号: CN202410230008.0
申请人: 浪潮电子信息产业股份有限公司
申请日期: 2024/2/29

摘要文本

本发明公开了一种机器学习方法、装置、设备、联邦学习系统及存储介质,属于边缘计算领域,用于在模型聚合过程中识别并剔除恶意用户的模型数据,解决了恶意用户攻击导致模型精度下降的问题。一方面边缘计算设备可以在簇内选择与自身的相关性达标的边缘计算设备的模型参数对自身的模型参数进行更新,另一方面簇头在进行簇内聚合时可以排除异常的边缘计算设备的模型参数,通过两层的风险识别,可以精准的识别并忽略异常的模型参数,排除了风险,有利于提升最终得到的全局网络模型的性能,使中心服务器在工业设备故障预测、网络安全问题识别与图片分类时发挥最优的效果,也避免了全局网络模型学习到恶意用户希望的特定行为。 来自:

专利详细信息

项目 内容
专利名称 机器学习方法、装置、设备、联邦学习系统及存储介质
专利类型 发明申请
申请号 CN202410230008.0
申请日 2024/2/29
公告号 CN117808126A
公开日 2024/4/2
IPC主分类号 G06N20/00
权利人 浪潮电子信息产业股份有限公司
发明人 李仁刚; 郭振华; 范宝余; 张润泽; 赵坤; 赵雅倩; 鲁璐; 曹芳; 王立; 贺蒙
地址 山东省济南市高新区草山岭南路801号9层东侧

专利主权项内容

1.一种安全分布式的机器学习方法,其特征在于,应用于联邦学习系统中的边缘计算设备,包括:获取联邦学习系统中各个边缘计算设备的分簇信息;其中,所述分簇信息包括各个簇中各边缘计算设备的设备号,各簇为基于同簇的边缘计算设备的数据类别相同的原则划分得到的;遍历同簇中除自身外的其他各个边缘计算设备,确定出与自身的模型参数相关性达标的目标边缘计算设备,并根据所述目标边缘计算设备的本地网络模型的模型参数,对自身的本地网络模型的模型参数进行更新;若自身为簇头,将簇内各个边缘计算设备的本地网络模型的模型参数中异常的模型参数所属的边缘计算设备,作为风险设备;将由簇内除所述风险设备外的其他边缘计算设备的模型参数聚合得到的簇网络模型发送至联邦学习系统的中心服务器,以便进行全局网络模型的聚合,通过所述全局网络模型执行预设任务;其中,所述预设任务包括预测工业设备的故障、识别网络安全问题以及对图片进行分类中的任一种。