基于目标驱动的目标识别深度神经网络剪枝方法
基本信息
申请号 | CN202110159802.7 | 申请日 | - |
公开(公告)号 | CN112633418A | 公开(公告)日 | 2021-04-09 |
申请公布号 | CN112633418A | 申请公布日 | 2021-04-09 |
分类号 | G06K9/62;G06N3/04;G06N3/08 | 分类 | 计算;推算;计数; |
发明人 | 刘向华 | 申请(专利权)人 | 名匠智汇建设发展有限公司 |
代理机构 | 深圳市中科创为专利代理有限公司 | 代理人 | 彭涛;刘曰莹 |
地址 | 518000 广东省深圳市南山区粤海街道高新南九道10号深圳湾科技生态园10栋A座17-20层01-06号 | ||
法律状态 | - |
摘要
摘要 | 本发明涉及剪枝技术领域,尤其涉及一种基于目标驱动的目标识别深度神经网络剪枝方法,如下步骤:步骤S1:从最后一层隐藏层开始,计算每个神经元的重要性得分;步骤S2:选择需要被剪枝的神经元,并且将该神经元的权重更新为0;步骤S3:将每个连接神经元的权重乘以该神经元的重要性得分;步骤S4:剪枝相应的神经元以及其对应的权重和偏差;步骤S5:判断所要剪枝的层是否为输入层,若不是,则前往前一层并返回到步骤S2进行处理。通过本发明基于目标驱动的目标识别深度神经网络剪枝方法,加快了推理速度并在不降低较多网络的测试精度的情况下显著减少了体积及对内存的占用,有利于深度神经网路部署到内存有限的设备中。 |
