一种基于神经网络模型剪枝的个性化协作学习方法和装置
实质审查的生效
摘要

本发明公开了一种基于神经网络模型剪枝的个性化协作学习方法和装置,其中,该方法包括:利用中心服务器初始化全局模型,并将全局模型下发至各边缘设备;各边缘设备接收到全局模型后,分别对全局模型进行训练以得到本地模型,基于本地模型对模型参数的重要性进行评估,并生成参数掩码矩阵;各边缘设备通过参数掩码矩阵,对参与协作训练的模型进行剪枝,将剪枝后的模型作为学生网络对本地模型进行知识蒸馏,并将剪枝后的模型上传至中心服务器,以对未被剪去的参数进行聚合。本发明能够保持模型对本地数据的适应能力,同时能够增强模型的泛化能力,在参数聚合时能避免数据分布差异过大的模型相互干扰。

基本信息
专利标题 :
一种基于神经网络模型剪枝的个性化协作学习方法和装置
专利标题(英):
暂无
公开(公告)号 :
CN114418085A
申请号 :
CN202111453868.3
公开(公告)日 :
2022-04-29
申请日 :
2021-12-01
授权号 :
暂无
授权日 :
暂无
发明人 :
徐恪刘泱赵乙朱敏
申请人 :
清华大学
申请人地址 :
北京市海淀区清华园
代理机构 :
北京清亦华知识产权代理事务所(普通合伙)
代理人 :
单冠飞
优先权 :
CN202111453868.3
主分类号 :
G06N3/08
IPC分类号 :
G06N3/08  G06N3/04  
IPC结构图谱
G
G部——物理
G06
计算;推算或计数
G06N
基于特定计算模型的计算机系统
G06N3/00
基于生物学模型的计算机系统
G06N3/02
采用神经网络模型
G06N3/08
学习方法
法律状态
2022-05-20 :
实质审查的生效
IPC(主分类) : G06N 3/08
申请日 : 20211201
2022-04-29 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载
  • 联系电话
    电话:023-6033-8768
    QQ:1493236332
  • 联系 Q Q
    电话:023-6033-8768
    QQ:1493236332
  • 关注微信
    电话:023-6033-8768
    QQ:1493236332
  • 收藏
    电话:023-6033-8768
    QQ:1493236332