- 专利标题: 一种计算机视觉深度学习模型剪枝方法及系统
-
申请号: CN202110931051.6申请日: 2021-08-13
-
公开(公告)号: CN113762506B公开(公告)日: 2023-11-24
- 发明人: 林广栋 , 吴尔杰 , 黄光红 , 刘小明
- 申请人: 中国电子科技集团公司第三十八研究所
- 申请人地址: 安徽省合肥市高新技术开发区香樟大道199号
- 专利权人: 中国电子科技集团公司第三十八研究所
- 当前专利权人: 中国电子科技集团公司第三十八研究所
- 当前专利权人地址: 安徽省合肥市高新技术开发区香樟大道199号
- 代理机构: 合肥昊晟德专利代理事务所
- 代理商 何梓秋
- 主分类号: G06N3/088
- IPC分类号: G06N3/088
摘要:
本发明公开了一种计算机视觉深度学习模型剪枝方法及系统,属于计算机视觉技术领域,包括以下步骤:S1:层选择;S2:节点相似度比较;S3:节点合并;S4:相似度遍历判断;S5:剩余层处理。本发明对被合并的节点与前一层相连的权重、与后一层相连的权重均进行处理,最大程度地保持权重信息,可减少剪枝技术对深度学习模型最终输出的影响;并对批量归一化操作的参数进行合并,最大限度地保留节点的信息;还对被剪枝的层之后的最大池化层(maxpool)、平均池化层(avgpool)、随机丢弃层(dropout)进行相应处理,值得被推广使用。
公开/授权文献
- CN113762506A 一种深度学习模型剪枝方法及系统 公开/授权日:2021-12-07