[发明专利]一种模型压缩方法及装置有效
申请号: | 201910239985.6 | 申请日: | 2019-03-27 |
公开(公告)号: | CN110008880B | 公开(公告)日: | 2023-09-29 |
发明(设计)人: | 吴泽衡;徐倩;杨强 | 申请(专利权)人: | 深圳前海微众银行股份有限公司 |
主分类号: | G06V40/10 | 分类号: | G06V40/10;G06V10/764;G06N20/00 |
代理公司: | 北京同达信恒知识产权代理有限公司 11291 | 代理人: | 黄志华 |
地址: | 518027 广东省深圳市*** | 国省代码: | 广东;44 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明实施例公开了一种模型压缩方法及装置,其中方法包括:使用第一模型对预测数据进行预测,得到第一模型在N个网络层中的输出以及预测数据对应的预测结果,使用预测数据和预测数据对应的预测结果训练第二模型,得到第二模型在N个网络层中的输出,根据i个网络层中第二模型相对于第一模型的输出损失调整第二模型的参数,得到对第一模型进行压缩之后的目标模型。本发明实施例中,通过采用第一模型的预测结果训练第二模型,并基于预设网络层中第二模型相对于第一模型的输出损失调整第二模型的参数,可以使得压缩之后的目标模型更好地复现第一模型的内容,从而可以提高目标模型的预测效果。 | ||
搜索关键词: | 一种 模型 压缩 方法 装置 | ||
【主权项】:
1.一种模型压缩方法,其特征在于,所述方法包括:使用第一模型对预测数据进行预测,得到所述第一模型分别在所述第一模型的N个网络层中的输出以及所述预测数据对应的预测结果;使用所述预测数据和预测数据对应的预测结果训练第二模型,得到所述第二模型分别在所述第二模型的N个网络层中的输出;所述第二模型是通过对所述第一模型进行剪枝得到的,所述第二模型的N个网络层与所述第一模型的N个网络层一一对应;根据所述第二模型在第二模型的i个网络层中的输出和所述第一模型在所述第一模型的i个网络层中的输出调整所述第二模型的参数,得到对所述第一模型进行压缩之后的目标模型。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于深圳前海微众银行股份有限公司,未经深圳前海微众银行股份有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201910239985.6/,转载请声明来源钻瓜专利网。