发明公开
- 专利标题： 一种基于深度学习的模型剪枝方法
-
申请号： CN202211646322.4申请日： 2022-12-21
-
公开(公告)号： CN116227574A公开(公告)日： 2023-06-06
- 发明人: 张晓华 , 陈勇 , 吕杰 , 黄怡超 , 马鹏飞
- 申请人： 深圳市视美泰技术股份有限公司
- 申请人地址： 广东省深圳市南山区粤海街道高新区社区沙河西路1801号国实大厦17A
- 专利权人： 深圳市视美泰技术股份有限公司
- 当前专利权人： 深圳市视美泰技术股份有限公司
- 当前专利权人地址： 广东省深圳市南山区粤海街道高新区社区沙河西路1801号国实大厦17A
- 代理机构： 深圳昊生知识产权代理有限公司
- 代理商 董慧婷
- 主分类号： G06N3/082
- IPC分类号： G06N3/082 ; G06N3/0464 ; G06V40/16 ; G06V10/764 ; G06V10/82
摘要：
本发明提供的一种基于深度学习的模型剪枝方法，包括以下步骤：步骤一，获取人脸数据，将人脸数据进行正则化稀疏训练，形成分类模型；步骤二，对分类模型进行通道枝剪，获得新的网络结构文件和权重文件；步骤三，用人脸数据进行精度恢复训练，得到最终模型；使用结构化思想的通道剪枝形成针对人脸识别模型的训练算法，在保证模型精度几乎不变的情况下，实现对模型推理速度的提速。
