[发明专利]模型训练方法、文本分类方法、电子设备及存储介质在审
申请号: | 202010861649.8 | 申请日: | 2020-08-24 |
公开(公告)号: | CN111898707A | 公开(公告)日: | 2020-11-06 |
发明(设计)人: | 刘小康;李健铨;晋耀红 | 申请(专利权)人: | 鼎富智能科技有限公司 |
主分类号: | G06K9/62 | 分类号: | G06K9/62;G06F16/35;G06N3/04;G06N3/08 |
代理公司: | 北京超凡宏宇专利代理事务所(特殊普通合伙) 11463 | 代理人: | 蒋姗 |
地址: | 230000 安徽省合肥市*** | 国省代码: | 安徽;34 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本申请提供一种模型训练方法、文本分类方法、电子设备及存储介质。方法包括根据学生模型第二transformer层中的各层到老师模型第一transformer层中各层之间的距离以及搬运量,对第二transformer层中每一层对应的权重进行更新获得更新后的权重;利用更新后的权重计算第二transformer层与第一transformer层之间的EMD,获得第一蒸馏损失;计算第二embedding层的第二蒸馏损失以及第二prediction层的第三蒸馏损失;根据第一蒸馏损失、第二蒸馏损失和第三蒸馏损失对学生模型中的参数进行训练,获得训练后的学生模型。本申请能够使学生模型学习到老师模型中更多的知识。 | ||
搜索关键词: | 模型 训练 方法 文本 分类 电子设备 存储 介质 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于鼎富智能科技有限公司,未经鼎富智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202010861649.8/,转载请声明来源钻瓜专利网。