用于对神经网络中卷积层进行剪枝的方法和装置
基本信息
| 申请号 | CN202010171150.4 | 申请日 | - |
| 公开（公告）号 | CN113392953A | 公开（公告）日 | 2021-09-14 |
| 申请公布号 | CN113392953A | 申请公布日 | 2021-09-14 |
| 分类号 | G06N3/04（2006.01）I;G06N3/08（2006.01）I | 分类 | 计算；推算；计数; |
| 发明人 | 聂远飞;董祯;冯欢 | 申请（专利权）人 | 澜起科技股份有限公司 |
| 代理机构 | 北京市君合律师事务所 | 代理人 | 毛健;杜小锋 |
| 地址 | 200233上海市徐汇区宜山路900号1幢A6 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本申请公开了一种用于对神经网络中卷积层进行剪枝的方法和装置。其中用于对神经网络中卷积层进行剪枝的方法包括：获取目标神经网络，目标神经网络包括待剪枝卷积层，待剪枝卷积层包括C个滤波器，每个滤波器中包括K个卷积核，每个卷积核包括M行N列个权重值，其中C、K、M和N为大于等于1的正整数；基于卷积核中权重值的数量M×N和目标压缩率确定每个卷积核中待剪枝的权重值的数量P，其中P为小于M×N的正整数；以及将待剪枝卷积层的每个卷积核中绝对值最小的P个权重值置零，形成剪枝后卷积层。 |
