[发明专利]基于注意力机制的mobilenet-v1知识蒸馏方法、存储器及终端设备在审
申请号: | 202110121769.9 | 申请日: | 2021-01-28 |
公开(公告)号: | CN112784999A | 公开(公告)日: | 2021-05-11 |
发明(设计)人: | 黄明飞;姚宏贵;梁维斌;王昊 | 申请(专利权)人: | 开放智能机器(上海)有限公司 |
主分类号: | G06N20/00 | 分类号: | G06N20/00 |
代理公司: | 上海申新律师事务所 31272 | 代理人: | 党蕾 |
地址: | 200233 上海市徐汇区*** | 国省代码: | 上海;31 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明提供了一种基于注意力机制的mobilenet‑v1知识蒸馏方法、存储器及终端设备,其中,包括:分别选择复杂模型WRN‑50‑8以及简单模型mobilenet‑v对应的特定中间层,用以进行注意图的知识转移;处理得到复杂模型和简单模型的中间层所对应的注意力图之间的损失,记为损失值一;处理获得复杂模型和简单模型的Logit层之间的KL散度;处理获得简单模型的交叉熵损失,记为损失值二;根据损失值一、KL散度及损失值二处理得到总损失;损失值一、RL散度、损失值二以及总损失用以简单模型的参数的计算。其技术方案的有益效果在于,与现有其他蒸馏方式相比,大幅提高mobilenet‑v1学生网络的识别精度和准确率,并可以将其部署在算力有限的设备。 | ||
搜索关键词: | 基于 注意力 机制 mobilenet v1 知识 蒸馏 方法 存储器 终端设备 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于开放智能机器(上海)有限公司,未经开放智能机器(上海)有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202110121769.9/,转载请声明来源钻瓜专利网。