一种模型蒸馏方法、系统以及文本检索方法
实质审查的生效
摘要
本发明提供了一种模型蒸馏方法、系统以及文本检索方法,其模型蒸馏方法包括如下步骤:S1.构建一个学生模型以及多个教师预训练模型,采用对比学习方法对多个教师预训练模型以及学生模型进行训练,在对比学习训练中,获取平均教师模型相似度矩阵以及学生模型相似度矩阵;S2.获取预蒸馏学生模型的第一输出以及多个教师模型的平均输出;S3.获取集成蒸馏损失函数,并根据集成蒸馏损失函数训练所述预蒸馏学生模型,并得到蒸馏学生模型;本发明通过对比学习训练得到平均教师模型相似度矩阵与学生模型相似度矩阵,将其加入集成蒸馏损失函数中,通过集成蒸馏损失函数训练预蒸馏学生模型,使得学生模型能够更好的拟合教师模型的检索能力。
基本信息
专利标题 :
一种模型蒸馏方法、系统以及文本检索方法
专利标题(英):
暂无
公开(公告)号 :
CN114328834A
申请号 :
CN202111641566.9
公开(公告)日 :
2022-04-12
申请日 :
2021-12-29
授权号 :
暂无
授权日 :
暂无
发明人 :
郭湘黄鹏江岭
申请人 :
成都晓多科技有限公司
申请人地址 :
四川省成都市天府新区兴隆街道湖畔路西段123号
代理机构 :
成都睿道专利代理事务所(普通合伙)
代理人 :
薛波
优先权 :
CN202111641566.9
主分类号 :
G06F16/33
IPC分类号 :
G06F16/33 G06F17/11 G06F17/16 G06K9/62
IPC结构图谱
G
G部——物理
G06
计算;推算或计数
G06F
电数字数据处理
G06F16/33
••查询
法律状态
2022-04-29 :
实质审查的生效
IPC(主分类) : G06F 16/33
申请日 : 20211229
申请日 : 20211229
2022-04-12 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载