[发明专利]基于嵌入特征相似度的注意力迁移的神经网络压缩方法在审
申请号: | 202111387298.2 | 申请日: | 2021-11-22 |
公开(公告)号: | CN114387447A | 公开(公告)日: | 2022-04-22 |
发明(设计)人: | 公茂果;刘昊;李豪;乔文远;张明阳;唐泽栋;刘洁怡;蒋祥明 | 申请(专利权)人: | 西安电子科技大学 |
主分类号: | G06V10/44 | 分类号: | G06V10/44;G06V10/764;G06V10/82;G06K9/62;G06N3/04;G06N3/08;G06N20/00 |
代理公司: | 西安嘉思特知识产权代理事务所(普通合伙) 61230 | 代理人: | 王萌 |
地址: | 710000 陕*** | 国省代码: | 陕西;61 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种基于嵌入特征相似度的注意力迁移的神经网络压缩方法,包括:获取多个训练样本,每个训练样本包括样本图像以及样本图像对应的类别;确定第一待训练神经网络模型后,利用训练样本训练第一待训练神经网络模型,得到教师网络;利用教师网络生成第一注意力图向量;确定第二待训练神经网络模型后,利用训练样本及预设损失函数训练第二待训练神经网络,并根据第二待训练神经网络模型生成的第二注意力图向量及输出结果确定损失值;当预设损失函数的损失值满足预设条件时,第二待训练神经网络训练完成,获得压缩后的学生网络。本发明在达到任务分类要求的同时,有效提高了训练的速度和精度,得到了效果更好的小规模网络。 | ||
搜索关键词: | 基于 嵌入 特征 相似 注意力 迁移 神经网络 压缩 方法 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于西安电子科技大学,未经西安电子科技大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202111387298.2/,转载请声明来源钻瓜专利网。