[发明专利]一种对预训练的神经网络模型进行剪枝的方法在审
| 申请号： | 202110263342.2 | 申请日： | 2021-03-11 |
| 公开（公告）号： | CN115081580A | 公开（公告）日： | 2022-09-20 |
| 发明（设计）人： | 王国栋;叶剑;杜恒欣 | 申请（专利权）人： | 中国科学院计算技术研究所临沂分所(临沂中科人工智能创新研究院);临沂中科人工智能创新研究院有限公司 |
| 主分类号： | G06N3/04 | 分类号： | G06N3/04;G06N3/08 |
| 代理公司： | 北京泛华伟业知识产权代理有限公司 11280 | 代理人： | 王勇 |
| 地址： | 276002 山东省临沂市济*** | 国省代码： | 山东;37 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本发明实施例提供了一种对预训练的神经网络模型进行剪枝的方法，其中将所述预训练的神经网络模型作为基线模型，包括：S1、选择所述基线模型中需要剪枝的网络层作为目标网络层；S2、获取目标网络层中多个卷积核的各权重参数对应的梯度值；S3、针对所述目标网络层，基于每个卷积核的各权重参数对应的梯度值确定该卷积核中重要的权重参数的数量；S4、针对所述目标网络层，根据其所有的卷积核中重要的权重参数的数量对所述目标网络层进行剪枝；S5、对剪枝后的模型进行重训练，得到压缩后的模型。本发明提高了剪枝的精度、效率。 | ||
| 搜索关键词： | 一种 训练 神经网络 模型 进行 剪枝 方法 | ||
【主权项】：
暂无信息
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于中国科学院计算技术研究所临沂分所(临沂中科人工智能创新研究院);临沂中科人工智能创新研究院有限公司，未经中国科学院计算技术研究所临沂分所(临沂中科人工智能创新研究院);临沂中科人工智能创新研究院有限公司许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/202110263342.2/，转载请声明来源钻瓜专利网。
