一种基于伪孪生网络的网络剪枝方法
基本信息
申请号 | CN201910400920.5 | 申请日 | - |
公开(公告)号 | CN110097177A | 公开(公告)日 | 2019-08-06 |
申请公布号 | CN110097177A | 申请公布日 | 2019-08-06 |
分类号 | G06N3/04;G06N3/08 | 分类 | 计算;推算;计数; |
发明人 | 闵锐;蓝海 | 申请(专利权)人 | 电科瑞达(成都)科技有限公司 |
代理机构 | 成都点睛专利代理事务所(普通合伙) | 代理人 | 孙一峰 |
地址 | 610041 四川省成都市高新区高朋大道21号1幢5层 | ||
法律状态 | - |
摘要
摘要 | 本发明属于神经网络模型压缩技术领域,具体为一种基于伪孪生网络的网络剪枝方法。本发明在传统的三值神经网络的基础上,提出了一种通用的裁剪大型卷积神经网络中冗余连接的方法,具体为:构造一个与被裁减网络N一样架构的三值网络Ternary‑N,用相同的图片进行每一步的训练,训练过程中采用知识蒸馏算法,训练完成后用Ternary‑N中的权值对N中相对应位置的权重进行裁剪,最后对N进行微调,微调后的网络连接数相比于原网络大大减少。 |
