Invention Publication
- Patent Title: 一种对网络模型的剪枝方法、装置、电子设备及存储介质
-
Application No.: CN202010964152.9Application Date: 2020-09-14
-
Publication No.: CN112101547APublication Date: 2020-12-18
- Inventor: 谷宇章 , 邱守猛 , 袁泽强 , 张晓林
- Applicant: 中国科学院上海微系统与信息技术研究所
- Applicant Address: 上海市长宁区长宁路865号
- Assignee: 中国科学院上海微系统与信息技术研究所
- Current Assignee: 中国科学院上海微系统与信息技术研究所
- Current Assignee Address: 上海市长宁区长宁路865号
- Agency: 广州三环专利商标代理有限公司
- Agent 郝传鑫; 贾允
- Main IPC: G06N3/08
- IPC: G06N3/08 ; G06N3/04

Abstract:
本申请实施例所公开的一种对网络模型的剪枝方法、装置、电子设备及存储介质,包括获取训练图像集和当前网络模型,将训练图像输入当前网络模型中,根据当前网络模型的输出确定多个卷积层中每个卷积层对应的参数,基于每个卷积层对应的预设剪枝率,对每个卷积层对应的参数进行衰减处理,得到衰减参数,若衰减参数与预设阈值的差值在预设区间内,剔除卷积层中与衰减参数对应的参数,得到剪枝后的网络模型。基于本申请实施例,通过对卷积层对应参数进行衰减处理,迫使待剔除参数的卷积层对应的参数学习的知识转移,在减少参数量的同时,不增加训练负担,并且可以保证网络模型的识别准确率。
Public/Granted literature
- CN112101547B 一种对网络模型的剪枝方法、装置、电子设备及存储介质 Public/Granted day:2024-04-16
Information query