[发明专利]基于自注意力的知识蒸馏方法、装置和计算机设备有效
| 申请号: | 202110059942.7 | 申请日: | 2021-01-18 |
| 公开(公告)号: | CN112365385B | 公开(公告)日: | 2021-06-01 |
| 发明(设计)人: | 徐泓洋;王广新;杨汉丹 | 申请(专利权)人: | 深圳市友杰智新科技有限公司 |
| 主分类号: | G06Q50/20 | 分类号: | G06Q50/20;G06Q10/06;G06N3/04;G06N20/00;G06F17/16 |
| 代理公司: | 深圳市明日今典知识产权代理事务所(普通合伙) 44343 | 代理人: | 王杰辉;曹勇 |
| 地址: | 518000 广东省深圳市南山区招商*** | 国省代码: | 广东;44 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | 本申请涉及人工智能领域,揭示了基于自注意力的知识蒸馏方法,包括:将输入数据输入第一模型得到第一模型的中间层输出的第一特征矩阵,将输入数据输入第二模型得到第二模型的中间层输出的第二特征矩阵,其中,第一模型为训练好的老师模型,第二模型为待训练的学生模型;根据第一特征矩阵计算老师模型对应的第一自注意力权重分布,根据第二特征矩阵计算学生模型对应的第二自注意力权重分布;计算第一自注意力权重分布和第二自注意力权重分布之间的分布差异;将分布差异,作为老师模型和学生模型之间的知识蒸馏损失函数;根据知识蒸馏损失函数,将老师模型的中间层的数据映射关系迁移至学生模型的中间层上,能满足不同任务类型模型的知识蒸馏训练。 | ||
| 搜索关键词: | 基于 注意力 知识 蒸馏 方法 装置 计算机 设备 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于深圳市友杰智新科技有限公司,未经深圳市友杰智新科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202110059942.7/,转载请声明来源钻瓜专利网。





