对象识别模型生成方法、装置、电子设备和存储介质制造方法及图纸

技术编号:35267904 阅读:23 留言:0更新日期:2022-10-19 10:33
本申请公开了一种对象识别模型生成方法、装置、电子设备和存储介质,该方法包括:确定目标任务样本与多个关联任务样本之间的样本关联信息。基于多个关联任务样本各自对应的导师模型,对目标任务样本中的样本对象进行对象识别,得到与多个导师模型各自对应的识别标签。基于样本关联信息和识别标签,确定目标任务样本对应的训练标签。基于目标任务样本以及训练标签,对待训练模型进行模型训练,得到对象识别模型。该方法可以在模型训练中融合不同任务场景的导师模型的预测能力,提高了对象识别模型的通用性,且模型训练时不需要目标任务样本中包括标注信息,从而可以保证在缺少标注信息的情况下模型训练的有效性。的情况下模型训练的有效性。的情况下模型训练的有效性。

【技术实现步骤摘要】
对象识别模型生成方法、装置、电子设备和存储介质


[0001]本申请涉及计算机
,尤其涉及对象识别模型生成方法、装置、电子设备和存储介质。

技术介绍

[0002]在应用神经网络模型执行多种任务场景对应的任务时,可以训练通用任务场景下的神经网络模型,使得该神经网络模型可以应用于任何一种任务场景中。
[0003]现有技术中,在对通用任务场景的神经网络模型进行模型训练时,由于存在多种任务场景,从而导致某些任务场景中缺少某些标注数据,使得训练得到的模型存在准确性低,导致模型的通用性低的问题。

技术实现思路

[0004]本申请提供了对象识别模型生成方法、装置、电子设备和存储介质,可以提高对象识别模型的通用性。
[0005]一方面,本申请提供了一种对象识别模型生成方法,方法包括:
[0006]确定目标任务样本与多个关联任务样本之间的样本关联信息;
[0007]基于所述多个关联任务样本各自对应的导师模型,对所述目标任务样本中的样本对象进行对象识别,得到与多个导师模型各自对应的识别标签;
[0008]基于所述样本关联信息确定多个导师模型各自对应的权重信息;
[0009]基于所述权重信息和所述识别标签,确定所述目标任务样本对应的训练标签;
[0010]基于所述目标任务样本以及所述训练标签,对待训练模型进行模型训练,得到对象识别模型。
[0011]另一方面提供了一种对象识别方法,所述方法包括:
[0012]获取待识别对象的对象信息;
[0013]将所述对象信息输入到对象识别模型中进行对象识别,得到所述待识别对象对应的识别结果,所述对象识别模型为基于目标任务样本以及训练标签进行模型训练得到的,所述训练标签为基于多个导师模型各自对应的识别标签和多个导师模型各自对应的权重信息确定的,所述识别标签为基于所述多个导师模型,分别对所述目标任务样本中的样本对象进行对象识别得到的,所述权重信息为基于所述目标任务样本,以及所述多个导师模型各自对应的关联任务样本间的样本关联信息确定的。
[0014]另一方面提供了一种对象识别模型生成装置,所述装置包括:
[0015]样本关联信息确定模块,用于确定目标任务样本与多个关联任务样本之间的样本关联信息;
[0016]识别标签获取模块,用于基于所述多个关联任务样本各自对应的导师模型,对所述目标任务样本中的样本对象进行对象识别,得到与多个导师模型各自对应的识别标签;
[0017]权重信息确定模块,用于基于所述样本关联信息确定所述多个导师模型各自对应
的权重信息;
[0018]训练标签确定模块,用于基于所述权重信息和所述识别标签,确定所述目标任务样本对应的训练标签;
[0019]模型训练模块,用于基于所述目标任务样本以及所述训练标签,对待训练模型进行模型训练,得到对象识别模型。
[0020]另一方面提供了一种对象识别装置,所述装置包括:
[0021]对象信息获取模块,用于获取待识别对象的对象信息;
[0022]对象识别模块,用于将所述对象信息输入到对象识别模型中进行对象识别,得到所述待识别对象对应的识别结果,所述对象识别模型为基于目标任务样本以及训练标签进行模型训练得到的,所述训练标签为基于多个导师模型各自对应的识别标签和多个导师模型各自对应的权重信息确定的,所述识别标签为基于所述多个导师模型,分别对所述目标任务样本中的样本对象进行对象识别得到的,所述权重信息为基于所述目标任务样本,以及所述多个导师模型各自对应的关联任务样本间的样本关联信息确定的。
[0023]另一方面提供了一种电子设备,电子设备包括处理器和存储器,存储器中存储有至少一条指令或至少一段程序,至少一条指令或至少一段程序由处理器加载并执行以实现如上述的对象识别模型生成方法或对象识别方法。
[0024]另一方面提供了一种计算机可读存储介质,存储介质包括处理器和存储器,存储器中存储有至少一条指令或至少一段程序,至少一条指令或至少一段程序由处理器加载并执行以实现如上述的对象识别模型生成方法或对象识别方法。
[0025]另一方面提供了一种计算机程序产品,包括计算机程序,计算机程序被处理器执行时实现如上述的对象识别模型生成方法或对象识别方法。
[0026]本申请提供的一种对象识别模型生成方法、装置、电子设备和存储介质,该方法可以确定目标任务样本与多个关联任务样本之间的样本关联信息。基于多个关联任务样本各自对应的导师模型,对目标任务样本中的样本对象进行对象识别,得到与多个导师模型各自对应的识别标签。基于样本关联信息和识别标签,确定目标任务样本对应的训练标签。基于目标任务样本以及训练标签,对待训练模型进行模型训练,得到对象识别模型。该方法可以在模型训练中融合不同任务场景的导师模型的预测能力,从而得到可以应用于任一一种任务场景的对象识别模型,提高了对象识别模型的通用性,且模型训练时不需要目标任务样本中包括标注信息,从而可以保证在缺少标注信息的情况下模型训练的有效性。
附图说明
[0027]为了更清楚地说明本申请实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本申请的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
[0028]图1为本申请实施例提供的一种对象识别模型生成方法的应用场景示意图;
[0029]图2为本申请实施例提供的一种对象识别模型生成方法的流程图;
[0030]图3为本申请实施例提供的一种对象识别模型生成方法中对信息关联模型进行训练的流程图;
[0031]图4为本申请实施例提供的一种对象识别模型生成方法中对识别标签进行数据映射处理的流程图;
[0032]图5为本申请实施例提供的一种对象识别模型生成方法中得到目标识别标签的流程图;
[0033]图6为本申请实施例提供的一种对象识别模型生成方法中计算权重信息的方法的流程图;
[0034]图7为本申请实施例提供的一种对象识别方法的流程图;
[0035]图8为本申请实施例提供的一种对象识别模型生成方法中生成对象识别模型的流程图;
[0036]图9为本申请实施例提供的一种对象识别模型生成方法中基于多个导师模型进行模型蒸馏生成对象识别模型的示意图;
[0037]图10为本申请实施例提供的一种对象识别模型生成装置的结构示意图;
[0038]图11为本申请实施例提供的一种对象识别装置的结构示意图;
[0039]图12为本申请实施例提供的一种用于实现本申请实施例所提供的方法的设备的硬件结构示意图。
具体实施方式
[0040]为使本申请的目的、技术方案和优点更加清楚,下面将结合附图对本申请作进一步地详细描述。显然,所描述的实施例仅仅是本申请的一部分实施例,而不是全部的实施例。基于本申请中的实施例,本领本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种对象识别模型生成方法,其特征在于,方法包括:确定目标任务样本与多个关联任务样本之间的样本关联信息;基于所述多个关联任务样本各自对应的导师模型,对所述目标任务样本中的样本对象进行对象识别,得到与多个导师模型各自对应的识别标签;基于所述样本关联信息确定多个导师模型各自对应的权重信息;基于所述权重信息和所述识别标签,确定所述目标任务样本对应的训练标签;基于所述目标任务样本以及所述训练标签,对待训练模型进行模型训练,得到对象识别模型。2.根据权利要求1的模型生成方法,其特征在于,所述基于所述多个关联任务样本各自对应的导师模型,对所述目标任务样本中的样本对象进行对象识别,得到与多个导师模型各自对应的识别标签之后,所述方法还包括:从所述多个关联任务样本各自对应的导师模型中确定参考模型;基于参考标签,对待对齐标签进行数据映射处理,得到目标识别标签,所述目标识别标签的数据分布与所述参考标签的数据分布一致,所述参考标签为所述参考模型对应的识别标签,所述待对齐标签为所述识别标签中除所述参考标签外的其他识别标签;基于所述权重信息和所述识别标签,确定所述目标任务样本对应的训练标签包括:基于所述权重信息、所述目标识别标签和所述参考标签,确定所述训练标签。3.根据权利要求2的对象识别模型生成方法,其特征在于,所述基于参考标签,对待对齐标签进行数据映射处理,得到目标识别标签包括:获取所述参考标签对应的预设数量个第一拟合点和所述待对齐标签中的预设数量个第二拟合点;所述第一拟合点基于所述参考标签的数据分布确定,所述第二拟合点基于所述待对齐标签的数据分布确定;对所述第一拟合点和所述第二拟合点进行拟合处理,得到所述待对齐标签对应的对齐参考信息;基于所述对齐参考信息,对所述待对齐标签进行映射处理,得到所述目标识别标签。4.根据权利要求1的对象识别模型生成方法,其特征在于,所述确定目标任务样本与多个关联任务样本之间的样本关联信息包括:将得到目标任务样本和得到多个关联任务样本输入到信息关联模型中进行信息关联预测,得到得到样本关联信息。5.根据权利要求4的对象识别模型生成方法,其特征在于,所述方法还包括:获取关联模型训练样本;所述关联模型训练样本包括多个训练样本项,每个训练样本项包括两项样本数据,以及所述两项样本数据的关联数据标签;将所述多个训练样本项输入到待训练关联模型中进行信息关联预测,得到样本关联训练信息;基于所述样本关联训练信息和所述关联数据标签,对所述待训练关联模型进行参数更新,得到所述信息关联模型。6.根据权利要求1的对象识别模型生成方法,其特征在于,所述基于所述样本关联信息确定所述多个导师模型各自对应的权重信息包括:基于所述样本关联信息,确定所述目标任务样本与所述多个关联任务样本间的关联参
数;对所述关联参数进行归一化处理,得到所述多个导师模型各自对应的权重信息。7.根据权利要求1的对象识别模型生成方法,其特征在于,所述基于所述权重信息和所述识别标签...

【专利技术属性】
技术研发人员:汤胜龙蓝利君李岳宸
申请(专利权)人:腾讯科技深圳有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1