模型训练方法、装置、计算机设备和存储介质与流程

专利检索2023-03-18  18



1.本技术涉及计算机技术领域,特别是涉及一种模型训练方法、装置、计算机设备、存储介质和计算机程序产品。


背景技术:

2.知识蒸馏技术是在模型训练过程中,使用一个规模较大的模型作为老师模型进行训练,提取出图像样本中的特征信息,然后将特征信息传递给规模较小的学生模型,使得规模较小的学生模型不仅速度较快,还能借助特征信息提升模型性能。
3.然而,传统的知识蒸馏技术是直接将蒸馏损失函数应用在所有图像样本上,并没有考虑图像样本之间的差异性,差异性包括图像样本的类别数量和图形样本包含的信息量,使得在模型训练过程中模型会更倾向于信息量较少的多数类样本,而忽视信息量较大的少数类样本,造成学生模型在信息量较大的少数类样本上的预测准确性较低。


技术实现要素:

4.基于此,有必要针对上述技术问题,提供一种能够提升学生模型预测准确率的模型训练方法、装置、计算机设备、计算机可读存储介质和计算机程序产品。
5.第一方面,本技术提供了一种模型训练方法。所述方法包括:获取已训练的教师模型对样本图像中各像素的类别预测结果;根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量;在各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数;根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型;所述训练完成的学生模型用于对输入的图像进行语义分割。
6.在其中一个实施例中,各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数包括:在各预设图像类别下,根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,确定所述样本图像中各像素的信息量的权重;根据各预设图像类别下所述样本图像中各像素的信息量和所述样本图像中各像素的信息量的权重,得到所述待训练的学生模型的目标损失函数。
7.在其中一个实施例中,在各预设图像类别下,根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别
预测结果,确定所述样本图像中各像素的信息量的权重,包括:根据所述样本图像中各像素的类别预测结果,从所述各预设图像类别中确定出所述样本图像中各像素所属的图像类别;根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息散度;所述信息散度表示所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果之间的距离;在所述各预设图像类别下,根据所述样本图像中各像素的信息散度,依次确定所述样本图像中所属的图像类别与所述预设图像类别相同的像素的信息量的权重,得到所述样本图像中各像素的信息量的权重。
8.在其中一个实施例中,根据所述样本图像中各像素的信息量和所述样本图像中各像素的信息量的权重,得到所述待训练的学生模型的目标损失函数,包括:在所述各预设图像类别下,分别根据样本图像中所属的图像类别与所述预设图像类别相同的像素的信息量和所述与所述预设图像类别相同的像素的信息量的权重,确定所述样本图像在所述各预设图像类别下的总信息量;根据所述样本图像在所述各预设图像类别下的总信息量之和的平均值,得到所述待训练的学生模型的目标损失函数。
9.在其中一个实施例中,根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量,包括:根据所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素在所述各预设图像类别下的类别信息量;根据所述各像素在所述各预设图像类别下的类别信息量,得到所述各像素的信息量。
10.在其中一个实施例中,根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量,还包括:分别从所述各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为所述各像素的目标预测结果;分别根据所述各像素的目标预测结果,得到所述样本图像中各像素的信息量。
11.在其中一个实施例中,分别从所述各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为所述各像素的目标预测结果,包括:分别从所述各像素的类别预测结果中,筛选出类别预测概率最大的类别预测结果,作为所述各像素的目标预测结果。
12.在其中一个实施例中,分别从所述各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为所述各像素的目标预测结果,包括:分别从所述各像素的类别预测结果中,筛选出类别预测概率最大和类别预测概率第二大的类别预测结果,作为所述各像素的目标预测结果。
13.在其中一个实施例中,根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型,包括:获取所述待训练的学生模型的初始损失函数;
根据所述初始损失函数和所述目标损失函数,得到总损失函数;根据所述总损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型。
14.在其中一个实施例中,根据所述初始损失函数和所述目标损失函数,得到总损失函数,包括:将所述初始损失函数与所述目标损失函数进行相加,得到所述总损失函数。
15.第二方面,本技术还提供了一种模型训练装置。所述装置包括:像素预测模块,用于获取已训练的教师模型对样本图像中各像素的类别预测结果;信息提取模块,用于根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量;函数获取模块,用于在各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数;模型获取模块,用于根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型;所述训练完成的学生模型用于对输入的图像进行语义分割。
16.第三方面,本技术还提供了一种计算机设备。所述计算机设备包括存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时实现以下步骤:获取已训练的教师模型对样本图像中各像素的类别预测结果;根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量;在各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数;根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型;所述训练完成的学生模型用于对输入的图像进行语义分割。
17.第四方面,本技术还提供了一种计算机可读存储介质。所述计算机可读存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时实现以下步骤:获取已训练的教师模型对样本图像中各像素的类别预测结果;根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量;在各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数;根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型;所述训练完成的学生模型用于对输入的图像进行语义分割。
18.第五方面,本技术还提供了一种计算机程序产品。所述计算机程序产品,包括计算机程序,该计算机程序被处理器执行时实现以下步骤:
获取已训练的教师模型对样本图像中各像素的类别预测结果;根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量;在各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数;根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型;所述训练完成的学生模型用于对输入的图像进行语义分割。
19.上述模型训练方法、装置、计算机设备、存储介质和计算机程序产品,通过根据已训练的教师模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息量;在各预设图像类别下,基于样本图像中各像素的信息量、已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到待训练的学生模型的目标损失函数;进而根据目标损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型;训练完成的学生模型用于对输入的图像进行语义分割。采用本方法,通过样本图像中各像素的信息量来得到待训练的学生模型的目标损失函数,目标损失函数能够使学生模型在训练过程中更关注样本图像中占有较高信息量的少数类,提高了学生模型在信息量较大的像素上的预测准确性,从而提升了学生模型整体的预测准确性。
附图说明
20.图1为一个实施例中模型训练方法的流程示意图;图2为一个实施例中获取待训练的学生模型的目标损失函数步骤的流程示意图;图3为另一个实施例中模型训练方法的流程示意图;图4为一个实施例中模型训练装置的结构框图;图5为一个实施例中计算机设备的内部结构图。
具体实施方式
21.为了使本技术的目的、技术方案及优点更加清楚明白,以下结合附图及实施例,对本技术进行进一步详细说明。应当理解,此处描述的具体实施例仅仅用以解释本技术,并不用于限定本技术。
22.在一个实施例中,如图1所示,提供了一种模型训练方法,本实施例以该方法应用于服务器进行举例说明,可以理解的是,该方法也可以应用于服务器,还可以应用于包括终端和服务器的系统,并通过终端和服务器的交互实现。其中,样本图像可以是预先存储在服务器中的,也可以是终端设备发送到服务器中的。本实施例中,该方法包括以下步骤:步骤s101,获取已训练的教师模型对样本图像中各像素的类别预测结果。
23.其中,教师模型和学生模型是知识蒸馏场景下对不同分支模型的描述,教师模型的参数的数量多于学生模型的参数的数量。
24.实际应用中,知识蒸馏作为一种重要的模型压缩手段,可以将教师模型在知识机器学习过程挖掘得到的知识,迁移到学生模型中,以使得学生模型用更少的空间复杂度和
训练时间得到与教师模型类似的训练效果,并且学生模型的拟合能力能够逼近甚至超过教师模型。
25.其中,像素的类别预测结果包括像素在各预设图像类别下的预测结果。
26.举例说明,假设预设图像类别包括类别1、类别2和类别3,则像素的类别预测结果会包括预测像素分别在类别1、类别2和类别3下的概率。
27.具体地,服务器获取已经预先训练完成的教师模型和样本图像,将样本图像输入到已训练教师模型中,通过已训练的教师模型对样本图像中各像素的类别进行预测,得到已训练的教师模型输出的样本图像中各像素的类别预测结果。
28.步骤s102,根据已训练的教师模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息量。
29.其中,信息量是指样本图像中的像素包含的信息。信息论中通过事件的概率来描述不确定性,本实施例中像素包含的信息是通过类别预测概率来描述不确定性。
30.其中,样本图像中易分割、预测简单的像素,其包含的信息量较小,样本图像中难分割、预测困难的像素,其包含的信息量较大。
31.需要说明的是,由于待训练的学生模型还缺乏训练,使得待训练的学生模型的预测结果并不可靠,因此,使用已训练的教师模型来获取样本图像中各像素的类别预测结果。
32.具体地,服务器根据已训练的教师模型对样本图像中各像素的类别预测结果,通过信息论中的信息量评价指标来计算样本图像中每个像素的不确定性,即像素的信息量;由此,服务器在得到信息量之后,将信息量作为处理依据执行后续的目标损失函数获取步骤。
33.步骤s103,在各预设图像类别下,基于样本图像中各像素的信息量、已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到待训练的学生模型的目标损失函数。
34.其中,目标损失函数用于调整模型训练过程中样本图像的类别权重。
35.具体地,服务器根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果之间的距离,可以构建初步的损失函数,再根据样本图像中各像素的信息量,确定样本图像中各像素在各预设图像类别下的权重,进而根据样本图像中各像素在各预设类别下的权重和各预设类别下的初步的损失函数,得到待训练的学生模型的目标损失函数。
36.实际应用中,初步的损失函数的构建可以通过信息散度(kullback-leiblerdivergence,kld)、余弦相似度、交叉熵等技术实现。
37.步骤s104,根据目标损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型;训练完成的学生模型用于对输入的图像进行语义分割。
38.其中,教师模型和学生模型可以是用于语义分割的网络模型。
39.其中,语义分割是指对样本图像进行像素级别的预测或分类,为达到较优的预测性能,往往需要较大的参数量来构建出准确率较高的模型,即教师模型,同时,这也会使得教师模型的预测速度较慢,无法适用于移动场景中的速度和时间需求;而训练完成的学生模型,由于自身参数数量较小的网络结构,使得学生模型在保持较快的处理速度的同时,还通过教师模型迁移过来的知识,提升了自身的预测性能,从而比训练完成的学生模型教师
模型拥有更强的处理速度和处理准确率。
40.具体地,服务器将目标损失函数作为待训练的学生模型的训练过程的损失函数,对待训练的学生模型进行迭代训练,当学生模型满足预设的训练条件时,得到训练完成的学生模型。由此,服务器在得到训练完成的学生模型之后,将训练完成的学生模型作为处理依据执行后续的语义分割步骤。通过训练完成的学生模型对输入的图像进行语义分割,得到语义分割结果,将语义分割结果存储在服务器中或者发送给终端设备。
41.上述模型训练方法中,通过根据已训练的教师模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息量;在各预设图像类别下,基于样本图像中各像素的信息量、已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到待训练的学生模型的目标损失函数;进而根据目标损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型;训练完成的学生模型用于对输入的图像进行语义分割。采用本方法,通过样本图像中各像素的信息量来得到待训练的学生模型的目标损失函数,能够使学生模型在训练过程中更关注样本图像中占有较高信息量的少数类,提高了学生模型在信息量较大的像素上的预测准确性,从而提升了学生模型整体的预测准确性。
42.在一个实施例中,如图2所示,在各预设图像类别下,基于样本图像中各像素的信息量、已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到待训练的学生模型的目标损失函数,具体包括如下内容:步骤s201,在各预设图像类别下,根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,确定样本图像中各像素的信息量的权重;步骤s202,根据各预设图像类别下样本图像中各像素的信息量和样本图像中各像素的信息量的权重,得到待训练的学生模型的目标损失函数。
43.其中,像素的信息量的权重用于在学生模型的训练过程中,赋予样本图像中信息量更高的像素更多的关注。
44.举例说明,语义分割场景中会面临像素的信息量不均衡的问题,例如,样本图像中包含有100个像素点,该样本图像的预设图像类别包括a、b、c三类,其中,c类中90%以上的像素点属于较容易分割区域,10%像素点的区域较难分割区域,由于较容易分割区域在c类中占比较大,使得较容易分割区域主导了较难分割区域的梯度,导致语义分割模型无法关注到各预设图像类别中信息量较大的较难分割区域;通过调整像素的信息量的权重,能够赋予信息量更高的像素更多的关注,即赋予样本图像中较难分割区域更多的关注。
45.进一步地,语义分割场景中还会面临像素的类别不均的问题,例如,样本图像中包含有100个像素点,该样本图像的预设图像类别包括a、b、c三类,其中,属于a类的像素点有68个,属于b类的像素点有15个,属于c类的像素点有17个;在该情况下,会导致模型在训练过程更多的关注在a类像素点的训练上,而忽略了b类和c类像素点;通过调整像素的权重,能够赋予少数类更高的关注,即赋予b类和c类像素点更高的关注。
46.具体地,服务器分别在各预设图像类别下,根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,确定样
本图像中各像素的信息量在对应图像类别下的权重,进而根据各预设图像类别下样本图像中各像素的信息量,以及样本图像中各像素的信息量在对应图像类别下的权重,得到待训练的学生模型的目标损失函数。
47.在实际的语义分割场景中,语义分割是一个密集分类任务,物体的不同部位的分类难易程度亦有所不同,样本图像中大部分都是相对较简单、易分割的部分,其包含的信息量亦较少,所以蒸馏损失函数基本上是由简单的部分占主导作用,使得梯度较小,而样本图像中真正富含信息的占小部分的像素被压制。由于占比更多的像素很有可能是背景图像,占比更少的像素是真正需要精确识别的人物、物品图像,因此本公开不仅考虑了教师模型和学生模型在预测结果上的差异性,还通过信息量来调整各预设图像类别下的像素的权重,使得学生模型在信息量更高的像素上具有更高的预测准确率。
48.本实施例中,在各预设图像类别下,根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,确定样本图像中各像素的信息量的权重,以使样本图像中各像素的信息量具有不同的权重;通过根据各预设图像类别下样本图像中各像素的信息量和样本图像中各像素的信息量的权重,得到待训练的学生模型的目标损失函数,进而实现了在目标损失函数中考虑样本图像中每个像素之间的差异性,从而提高了学生模型的预测准确性。
49.在一个实施例中,上述步骤s201,在各预设图像类别下,根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,确定样本图像中各像素的信息量的权重,具体包括如下内容:根据样本图像中各像素的类别预测结果,从各预设图像类别中确定出样本图像中各像素所属的图像类别;根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息散度;信息散度表示已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果之间的距离;在各预设图像类别下,根据样本图像中各像素的信息散度,依次确定样本图像中所属的图像类别与预设图像类别相同的像素的信息量的权重,得到样本图像中各像素的信息量的权重。
50.具体地,服务器检测样本图像中各像素的类别预测结果,得到各像素的类别预测结果中数值最大的预测结果,从各预设图像类别中,确定出数值最大的预测结果所属的图像类别,作为样本图像中像素所属的图像类别;例如,样本图像中一像素的类别预测结果为(95%,2%,3%),即该像素属于图像类别1的概率为95%,该像素属于图像类别2的概率为2%,该像素属于图像类别3的概率为3%,因为95%》3%》2%,所以该像素属于图像类别1。根据信息散度可以通过已训练的教师模型对样本图像中各像素的类别预测结果的信息熵和待训练的学生模型对样本图像中各像素的类别预测结果的信息熵之间的差值,得到样本图像中各像素的信息散度;分别在各预设图像类别下,依次判断样本图像中各像素所属的图像类别与预设图像类别是否相同,若样本图像中像素所属的图像类别与预设图像类别相同,则根据该像素的信息散度,确定该像素的信息量的权重;若样本图像中像素所属的图像类别与预设图像类别不相同,则继续判断样本图像中下一个像素所属的图像类别与预设图像类别是否相同,直到得到样本图像中各像素的信息量的权重。
51.举例说明,预设图像类别包括图像类别1、图像类别2和图像类别3,则先确定样本
图像中属于图像类别1的像素的信息量的权重,再确定样本图像中属于图像类别2的像素的信息量的权重,最后确定样本图像中属于图像类别3的像素的信息量的权重,最终得到样本图像中各像素的信息量的权重。
52.本实施例中,服务器根据样本图像中各像素的类别预测结果,从各预设图像类别中确定出样本图像中各像素所属的图像类别;根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息散度;进而在各预设图像类别下,根据样本图像中各像素的信息散度,依次确定样本图像中所属的图像类别与预设图像类别相同的像素的信息量的权重,得到样本图像中各像素的信息量的权重,从而实现了获取样本图像中各像素的信息量的权重,考虑了样本图像中各像素的信息量。
53.在一个实施例中,上述步骤s202,根据样本图像中各像素的信息量和样本图像中各像素的信息量的权重,得到待训练的学生模型的目标损失函数,具体包括如下内容:在各预设图像类别下,分别根据样本图像中所属的图像类别与预设图像类别相同的像素的信息量和与预设图像类别相同的像素的信息量的权重,确定样本图像在各预设图像类别下的总信息量;根据样本图像在各预设图像类别下的总信息量之和的平均值,得到待训练的学生模型的目标损失函数。
54.其中,总信息量是指样本图像中所有像素的信息量之和。
55.具体地,服务器分别在各预设图像类别下,依次判断样本图像中各像素所属的图像类别与预设图像类别是否相同,若样本图像中像素所属的图像类别与预设图像类别不相同,则将该像素的信息量和像素的信息量的权重设置为0;若样本图像中像素所属的图像类别与预设图像类别相同,则根据样本图像中像素的信息量和像素的信息量的权重,确定样本图像在各预设图像类别下的总信息量。将样本图像分别在各预设图像类别下的总信息量相加,得到样本图像在所有预设图像类别下的总信息量之和,根据预设图像类别的数量,获取样本图像在各预设图像类别下的总信息量之和的平均值,根据样本图像在各预设图像类别下的总信息量之和的平均值,得到待训练的学生模型的目标损失函数。
56.实际应用中,可以通过如下公式来确定待训练的学生模型的目标损失函数:其中,k表示为预设图像类别的数量,h表示为样本图像的高,w表示为样本图像的宽,表示为样本图像中像素点x的信息量,表示为像素x的信息散度,表示为待训练的学生模型对样本图像中像素x的类别预测结果,表示为已训练的教师模型对样本图像中像素x的类别预测结果,1(y
x
=k)表示像素x所属的图像类别是否等于预设图像类别k,当像素x所属的图像类别等于预设图像类别k时,则1(y
x
=k)为1,否则1(y
x
=k)为0。
57.本实施例中,服务器在各预设图像类别下,分别根据样本图像中所属的图像类别与预设图像类别相同的像素的信息量和与预设图像类别相同的像素的信息量的权重,确定样本图像在各预设图像类别下的总信息量,根据总信息量确定待训练的学生模型的目标损
失函数。采用本方法,分别根据各预设图像类别、各像素的信息量和信息量的权重来确定目标损失函数,不仅实现了通过像素的信息量来调整像素的权重,还实现了通过每个预设图像类别的总信息量来调整各预设图像类别之间的权重,解决了知识蒸馏在语义分割场景中的样本图像的像素信息量不均和像素类别不均的问题,从而提升了学生模型整体的预测准确性。
58.在一个实施例中,根据已训练的教师模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息量,具体包括如下内容:根据样本图像中各像素的类别预测结果,得到样本图像中各像素在各预设图像类别下的类别信息量;根据各像素在各预设图像类别下的类别信息量,得到各像素的信息量。
59.其中,信息量与各预设图像类别下的类别预测结果分布相关,预设图像类别下的类别预测结果的分布越极端,即类别预测结果中最大预测概率的数值越大,则像素的信息量越小;反之,类别预测结果的分布越均衡,即类别预测结果中最大预测概率的数值越小,则像素的信息量越大。
60.具体地,服务器在各预设图像类别下,对样本图像中各像素的类别预测结果的对数进行求负,得到样本图像中各像素在各预设图像类别下的类别信息量,将各像素在各预设图像类别下的类别信息量进行相加,得到各像素的信息量。
61.实际应用中,基于香农熵获取样本图像中各像素的信息量,则样本图像中各像素的信息量可以通过如下公式得到:其中,x=[1,2,3,

,hw],h表示为样本图像的高,w表示为样本图像的宽,k表示为预设图像类别的数量,表示为已训练的教师模型对样本图像中像素x的类别预测结果。
[0062]
也就是说,先分别确定像素在各预设图像类别下的类别信息量,再将所有图像类别的类别信息量之和,作为像素的信息量。
[0063]
本实施例中,服务器根据样本图像中各像素的类别预测结果,得到样本图像中各像素在各预设图像类别下的类别信息量,进而根据各像素在各预设图像类别下的类别信息量,得到各像素的信息量,从而实现基于样本图像中各像素的类别预测结果准确的确定各像素的信息量。
[0064]
在一个实施例中,根据已训练的教师模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息量,还包括:分别从各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为各像素的目标预测结果;分别根据各像素的目标预测结果,得到样本图像中各像素的信息量。
[0065]
具体地,服务器根据预设的信息量度量方式,确定第一预设条件,分别从各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为各像素的目标预测结果;然后分别根据各像素的目标预测结果和预设的信息量度量方式,确定样本图像中各像素的信息量。
[0066]
本实施例中,服务器分别从各像素的类别预测结果中,筛选出满足第一预设条件
的类别预测结果,作为各像素的目标预测结果,根据目标预测结果,确定像素的信息量,从而实现基于样本图像中各像素的类别预测结果准确的确定各像素的信息量。
[0067]
在一个实施例中,分别从各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为各像素的目标预测结果,包括:分别从各像素的类别预测结果中,筛选出类别预测概率最大的类别预测结果,作为各像素的目标预测结果。
[0068]
其中,类别预测概率是指像素属于各预设图像类别的概率。
[0069]
举例说明,样本图像中某一像素的类别预测结果为(95%,2%,3%),其中,类别预测概率是指95%、2%、3%,而95%就是类别预测概率最大的类别预测结果。
[0070]
具体地,服务器还可以根据最低置信度(least confident)来确定像素的信息量,将设置第一预设条件设置为类别预测概率最大,从各像素的类别预测结果中,筛选出类别预测概率最大的类别预测结果,作为各像素的目标预测结果,即第一预设条件为类别预测概率最大的类别预测结果;此时,目标预测结果的概率越大,像素的信息量越小,反之,目标预测结果的概率越小,像素的信息量越大。
[0071]
在本实施例中,服务器将各像素的类别预测结果中类别预测概率最大的,作为各像素的目标预测结果,进而根据目标预测结果,得到各像素的信息量,从而实现基于样本图像中各像素的类别预测结果准确的确定各像素的信息量。
[0072]
在一个实施例中,分别从各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为各像素的目标预测结果,包括:分别从各像素的类别预测结果中,筛选出类别预测概率最大和类别预测概率第二大的类别预测结果,作为各像素的目标预测结果。
[0073]
具体地,服务器还可以根据边界采样(margin sampling)来确定像素的信息量,从各像素的类别预测结果中,筛选出类别预测概率最大的类别预测结果和类别预测概率第二大的类别预测结果,作为各像素的目标预测结果,即第一预设条件为类别预测概率最大的类别预测结果和类别预测概率第二大的类别预测结果;根据概率数值最大的类别预测结果和概率数值第二大的类别预测结果之间的差值,确定像素的信息量,差值越大,像素的信息量越小,反之,差值越小,像素的信息量越大。
[0074]
举例说明,样本图像中某一像素的类别预测结果为(95%,2%,3%),其中,类别预测概率是指95%、2%、3%,而95%和3%分别是指类别预测概率最大的类别预测结果和类别预测概率第二大的类别预测结果。
[0075]
在本实施例中,服务器根据像素的类别预测结果中类别预测概率最大的类别预测结果和类别预测概率第二大的类别预测结果之间的差值,确定像素的信息量,从而实现基于样本图像中各像素的类别预测结果准确的确定各像素的信息量。
[0076]
在一个实施例中,根据目标损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型,包括:获取待训练的学生模型的初始损失函数;根据初始损失函数和目标损失函数,得到总损失函数;根据总损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型。
[0077]
其中,初始损失函数可以是待训练的学生模型的交叉熵、信息散度、余弦相似度等损失函数。
[0078]
具体地,获取待训练的学生模型的初始损失函数和损失权重,将损失权重作为目
标损失函数的目标损失权重,根据初始损失函数、目标损失函数和目标损失函数的目标损失权重,得到总损失函数;根据总损失函数,对待训练的学生模型进行迭代训练。
[0079]
进一步地,当检测到待训练的学生模型的预测结果满足第二预设条件时,将预测结果满足第二预设条件的学生模型作为训练完成的学生模型;或者,当待训练的学生模型经过迭代训练的次数达到预设训练次数时,将达到预设训练次数时的学生模型作为训练完成的学生模型。其中,当目标损失函数为1时,训练完成的学生模型通常具有较好的预测效果,当然目标损失函数也可以根据实际应用场景取其他数值。
[0080]
本实施例中,服务器通过获取待训练的学生模型的初始损失函数,根据初始损失函数和目标损失函数,得到总损失函数,进而根据总损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型,从而基于训练完成的学生模型可以对输入的样本图像进行语义分割处理。
[0081]
在一个实施例中,根据初始损失函数和目标损失函数,得到总损失函数,具体包括如下内容:将初始损失函数与目标损失函数进行相加,得到总损失函数。
[0082]
具体地,将目标损失函数和目标损失函数的目标损失权重进行相乘,得到相乘后损失函数,将相乘后损失函数与初始损失函数进行相加,得到总损失函数。
[0083]
举例说明,总损失函数l可以通过如下公式得到:其中,l
stu
表示为待训练的学生模型的初始损失函数;表示为待训练的学生模型的损失权重,而为1时,通过总损失函数训练得到的学生模型通常具有较好的预测效果;表示为待训练的学生模型的目标损失函数。
[0084]
在实际应用中,教师模型可以是101层的深度残差网络(deep residual network,resnet),即resnet-101,学生模型可以是18层的深度残差网络,即resnet-18。基于resnet-101教师模型和resnet-18学生模型对本公开提出的模型训练方法进行测试,选择语义分割场景中常使用的城市景观(cityscapes)数据集和pascal context数据集;在教师模型、学生模型、教师模型和学生模型的预设超参数都相同的情况下,分别采用本公开中提出的模型训练方法与语义分割场景中的传统技术kd(distilling the knowledge in a neural network)、sdk(structured knowledge distillation for dense prediction)、ifvd(intra-class feature variation distillation for semantic segmentation)、cscace(knowledge distillation for semantic segmentation using channel and spatial correlations and adaptive crossentropy)和ka(knowledge adaptation for efficient semantic segmentation),对resnet-18学生模型进行训练,得到本公开提出的模型训练方法得到的训练完成的学生模型的预测准确率与其他方法得到的训练完成的学生模型的预测准确率,预测准确率的对比结果如表1所示。
[0085]
表1模型的预测准确率(%)
由表1可知,通过本公开中提出的模型训练方法得到的训练完成的学生模型,在城市景观(cityscapes)数据集和pascal context数据集上,比传统技术kd、sdk、ifvd、cscace和ka得到的训练完成的学生模型具有更高的准确率。因此,本公开中提出的模型训练方法,在不影响学生模型的预测速度、不增加学生模型的参数、不改变学生模型的预设超参数的同时,还实现了学生模型的预测准确率的有效提升。
[0086]
本实施例中,服务器将初始损失函数和目标损失函数进行相加,得到总损失函数,进而根据总损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型,从而基于训练完成的学生模型可以对输入的样本图像进行语义分割处理。
[0087]
在一个实施例中,如图3所示,提供了另一种模型训练方法,以该方法应用于服务器为例进行说明,包括以下步骤:步骤s301,获取已训练的教师模型对样本图像中各像素的类别预测结果。
[0088]
步骤s302,根据样本图像中各像素的类别预测结果,得到样本图像中各像素在各预设图像类别下的类别信息量;根据各像素在各预设图像类别下的类别信息量,得到各像素的信息量。
[0089]
步骤s303,分别从各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为各像素的目标预测结果;分别根据各像素的目标预测结果,得到样本图像中各像素的信息量;其中,步骤s303还包括步骤s303-1,分别从各像素的类别预测结果中,筛选出类别预测概率最大的类别预测结果,作为各像素的目标预测结果;步骤s303还包括步骤s303-2,分别从各像素的类别预测结果中,筛选出类别预测概率最大和类别预测概率第二大的类别预测结果,作为各像素的目标预测结果。
[0090]
步骤s304,根据样本图像中各像素的类别预测结果,从各预设图像类别中确定出样本图像中各像素所属的图像类别。
[0091]
步骤s305,根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练
的学生模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息散度。
[0092]
步骤s306,在各预设图像类别下,根据样本图像中各像素的信息散度,依次确定样本图像中所属的图像类别与预设图像类别相同的像素的信息量的权重,得到样本图像中各像素的信息量的权重。
[0093]
步骤s307,在各预设图像类别下,分别根据样本图像中所属的图像类别与预设图像类别相同的像素的信息量和与预设图像类别相同的像素的信息量的权重,确定样本图像在各预设图像类别下的总信息量。
[0094]
步骤s308,根据样本图像在各预设图像类别下的总信息量之和的平均值,得到待训练的学生模型的目标损失函数。
[0095]
步骤s309,获取待训练的学生模型的初始损失函数;根据初始损失函数和目标损失函数,得到总损失函数;其中,步骤s309还包括步骤s309-1,将初始损失函数与目标损失函数进行相加,得到总损失函数。
[0096]
步骤s310,根据总损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型。
[0097]
上述模型训练方法,能够实现以下有益效果:分别根据各预设图像类别、各像素的信息量和信息量的权重来确定目标损失函数,不仅实现了通过像素的信息量来调整像素的权重,还实现了通过每个预设图像类别的总信息量来调整各预设图像类别之间的权重,解决了知识蒸馏在语义分割场景中的样本图像的像素信息量不均和像素类别不均的问题,从而提升了学生模型整体的预测准确性。
[0098]
应该理解的是,虽然如上所述的各实施例所涉及的流程图中的各个步骤按照箭头的指示依次显示,但是这些步骤并不是必然按照箭头指示的顺序依次执行。除非本文中有明确的说明,这些步骤的执行并没有严格的顺序限制,这些步骤可以以其它的顺序执行。而且,如上所述的各实施例所涉及的流程图中的至少一部分步骤可以包括多个步骤或者多个阶段,这些步骤或者阶段并不必然是在同一时刻执行完成,而是可以在不同的时刻执行,这些步骤或者阶段的执行顺序也不必然是依次进行,而是可以与其它步骤或者其它步骤中的步骤或者阶段的至少一部分轮流或者交替地执行。
[0099]
基于同样的发明构思,本技术实施例还提供了一种用于实现上述所涉及的模型训练方法的模型训练装置。该装置所提供的解决问题的实现方案与上述方法中所记载的实现方案相似,故下面所提供的一个或多个模型训练装置实施例中的具体限定可以参见上文中对于模型训练方法的限定,在此不再赘述。
[0100]
在一个实施例中,如图4所示,提供了一种模型训练装置400,包括:像素预测模块401、信息提取模块402、函数获取模块403和模型训练模块404,其中:像素预测模块401,用于获取已训练的教师模型对样本图像中各像素的类别预测结果。
[0101]
信息提取模块402,用于根据已训练的教师模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息量。
[0102]
函数获取模块403,用于在各预设图像类别下,基于样本图像中各像素的信息量、已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像
中各像素的类别预测结果,得到待训练的学生模型的目标损失函数。
[0103]
模型获取模块404,用于根据目标损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型;训练完成的学生模型用于对输入的图像进行语义分割。
[0104]
在一个实施例中,函数获取模块403,还用于在各预设图像类别下,根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,确定样本图像中各像素的信息量的权重;根据各预设图像类别下样本图像中各像素的信息量和样本图像中各像素的信息量的权重,得到待训练的学生模型的目标损失函数。
[0105]
在一个实施例中,模型训练装置400还包括权重确定模块,用于根据样本图像中各像素的类别预测结果,从各预设图像类别中确定出样本图像中各像素所属的图像类别;根据已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息散度;信息散度表示已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果之间的距离;在各预设图像类别下,根据样本图像中各像素的信息散度,依次确定样本图像中所属的图像类别与预设图像类别相同的像素的信息量的权重,得到样本图像中各像素的信息量的权重。
[0106]
在一个实施例中,模型训练装置400还包括函数确定模块,用于在各预设图像类别下,分别根据样本图像中所属的图像类别与预设图像类别相同的像素的信息量和与预设图像类别相同的像素的信息量的权重,确定样本图像在各预设图像类别下的总信息量;根据样本图像在各预设图像类别下的总信息量之和的平均值,得到待训练的学生模型的目标损失函数。
[0107]
在一个实施例中,信息提取模块402,还用于根据样本图像中各像素的类别预测结果,得到样本图像中各像素在各预设图像类别下的类别信息量;根据各像素在各预设图像类别下的类别信息量,得到各像素的信息量。
[0108]
在一个实施例中,信息提取模块402,还用于分别从各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为各像素的目标预测结果;分别根据各像素的目标预测结果,得到样本图像中各像素的信息量。
[0109]
在一个实施例中,模型训练装置400还包括第一条件确定模块,用于分别从各像素的类别预测结果中,筛选出类别预测概率最大的类别预测结果,作为各像素的目标预测结果。
[0110]
在一个实施例中,模型训练装置400还包括第二条件确定模块,用于分别从各像素的类别预测结果中,筛选出类别预测概率最大和类别预测概率第二大的类别预测结果,作为各像素的目标预测结果。
[0111]
在一个实施例中,模型获取模块404,还用于获取待训练的学生模型的初始损失函数;根据初始损失函数和目标损失函数,得到总损失函数;根据总损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型。
[0112]
在一个实施例中,模型训练装置400还包括函数相加模块,用于将初始损失函数与目标损失函数进行相加,得到总损失函数。
[0113]
上述模型训练装置中的各个模块可全部或部分通过软件、硬件及其组合来实现。
上述各模块可以硬件形式内嵌于或独立于计算机设备中的处理器中,也可以以软件形式存储于计算机设备中的存储器中,以便于处理器调用执行以上各个模块对应的操作。
[0114]
在一个实施例中,提供了一种计算机设备,该计算机设备可以是服务器,其内部结构图可以如图5所示。该计算机设备包括通过系统总线连接的处理器、存储器和网络接口。其中,该计算机设备的处理器用于提供计算和控制能力。该计算机设备的存储器包括非易失性存储介质和内存储器。该非易失性存储介质存储有操作系统、计算机程序和数据库。该内存储器为非易失性存储介质中的操作系统和计算机程序的运行提供环境。该计算机设备的数据库用于存储样本图像、样本图像中各像素的预测结果和各像素的信息量等数据。该计算机设备的网络接口用于与外部的终端通过网络连接通信。该计算机程序被处理器执行时以实现一种模型训练方法。
[0115]
本领域技术人员可以理解,图5中示出的结构,仅仅是与本技术方案相关的部分结构的框图,并不构成对本技术方案所应用于其上的计算机设备的限定,具体的计算机设备可以包括比图中所示更多或更少的部件,或者组合某些部件,或者具有不同的部件布置。
[0116]
在一个实施例中,还提供了一种计算机设备,包括存储器和处理器,存储器中存储有计算机程序,该处理器执行计算机程序时实现上述各方法实施例中的步骤。
[0117]
在一个实施例中,提供了一种计算机可读存储介质,其上存储有计算机程序,该计算机程序被处理器执行时实现上述各方法实施例中的步骤。
[0118]
在一个实施例中,提供了一种计算机程序产品,包括计算机程序,该计算机程序被处理器执行时实现上述各方法实施例中的步骤。
[0119]
需要说明的是,本技术所涉及的用户信息(包括但不限于用户设备信息、用户个人信息等)和数据(包括但不限于用于分析的数据、存储的数据、展示的数据等),均为经用户授权或者经过各方充分授权的信息和数据。
[0120]
本领域普通技术人员可以理解实现上述实施例方法中的全部或部分流程,是可以通过计算机程序来指令相关的硬件来完成,所述的计算机程序可存储于一非易失性计算机可读取存储介质中,该计算机程序在执行时,可包括如上述各方法的实施例的流程。其中,本技术所提供的各实施例中所使用的对存储器、数据库或其它介质的任何引用,均可包括非易失性和易失性存储器中的至少一种。非易失性存储器可包括只读存储器(read-only memory,rom)、磁带、软盘、闪存、光存储器、高密度嵌入式非易失性存储器、阻变存储器(reram)、磁变存储器(magnetoresistive random access memory,mram)、铁电存储器(ferroelectric random access memory,fram)、相变存储器(phase change memory,pcm)、石墨烯存储器等。易失性存储器可包括随机存取存储器(random access memory,ram)或外部高速缓冲存储器等。作为说明而非局限,ram可以是多种形式,比如静态随机存取存储器(static random access memory,sram)或动态随机存取存储器(dynamic random access memory,dram)等。本技术所提供的各实施例中所涉及的数据库可包括关系型数据库和非关系型数据库中至少一种。非关系型数据库可包括基于区块链的分布式数据库等,不限于此。本技术所提供的各实施例中所涉及的处理器可为通用处理器、中央处理器、图形处理器、数字信号处理器、可编程逻辑器、基于量子计算的数据处理逻辑器等,不限于此。
[0121]
以上实施例的各技术特征可以进行任意的组合,为使描述简洁,未对上述实施例中的各个技术特征所有可能的组合都进行描述,然而,只要这些技术特征的组合不存在矛
盾,都应当认为是本说明书记载的范围。
[0122]
以上所述实施例仅表达了本技术的几种实施方式,其描述较为具体和详细,但并不能因此而理解为对本技术专利范围的限制。应当指出的是,对于本领域的普通技术人员来说,在不脱离本技术构思的前提下,还可以做出若干变形和改进,这些都属于本技术的保护范围。因此,本技术的保护范围应以所附权利要求为准。

技术特征:
1.一种模型训练方法,其特征在于,所述方法包括:获取已训练的教师模型对样本图像中各像素的类别预测结果;根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量;在各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数;根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型;所述训练完成的学生模型用于对输入的图像进行语义分割。2.根据权利要求1所述的方法,其特征在于,所述在各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数,包括:在各预设图像类别下,根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,确定所述样本图像中各像素的信息量的权重;根据各预设图像类别下所述样本图像中各像素的信息量和所述样本图像中各像素的信息量的权重,得到所述待训练的学生模型的目标损失函数。3.根据权利要求2所述的方法,其特征在于,所述在各预设图像类别下,根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,确定所述样本图像中各像素的信息量的权重,包括:根据所述样本图像中各像素的类别预测结果,从所述各预设图像类别中确定出所述样本图像中各像素所属的图像类别;根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息散度;所述信息散度表示所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果之间的距离;在所述各预设图像类别下,根据所述样本图像中各像素的信息散度,依次确定所述样本图像中所属的图像类别与所述预设图像类别相同的像素的信息量的权重,得到所述样本图像中各像素的信息量的权重。4.根据权利要求3所述的方法,其特征在于,所述根据所述样本图像中各像素的信息量和所述样本图像中各像素的信息量的权重,得到所述待训练的学生模型的目标损失函数,包括:在所述各预设图像类别下,分别根据样本图像中所属的图像类别与所述预设图像类别相同的像素的信息量和所述与所述预设图像类别相同的像素的信息量的权重,确定所述样本图像在所述各预设图像类别下的总信息量;根据所述样本图像在所述各预设图像类别下的总信息量之和的平均值,得到所述待训练的学生模型的目标损失函数。5.根据权利要求1所述的方法,其特征在于,所述根据所述已训练的教师模型对所述样
本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量,包括:根据所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素在所述各预设图像类别下的类别信息量;根据所述各像素在所述各预设图像类别下的类别信息量,得到所述各像素的信息量。6.根据权利要求1所述的方法,其特征在于,所述根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量,还包括:分别从所述各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为所述各像素的目标预测结果;分别根据所述各像素的目标预测结果,得到所述样本图像中各像素的信息量。7.根据权利要求6所述的方法,其特征在于,所述分别从所述各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为所述各像素的目标预测结果,包括:分别从所述各像素的类别预测结果中,筛选出类别预测概率最大的类别预测结果,作为所述各像素的目标预测结果。8.根据权利要求6所述的方法,其特征在于,所述分别从所述各像素的类别预测结果中,筛选出满足第一预设条件的类别预测结果,作为所述各像素的目标预测结果,包括:分别从所述各像素的类别预测结果中,筛选出类别预测概率最大和类别预测概率第二大的类别预测结果,作为所述各像素的目标预测结果。9.根据权利要求1所述的方法,其特征在于,所述根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型,包括:获取所述待训练的学生模型的初始损失函数;根据所述初始损失函数和所述目标损失函数,得到总损失函数;根据所述总损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型。10.根据权利要求9所述的方法,其特征在于,所述根据所述初始损失函数和所述目标损失函数,得到总损失函数,包括:将所述初始损失函数与所述目标损失函数进行相加,得到所述总损失函数。11.一种模型训练装置,其特征在于,所述装置包括:像素预测模块,用于获取已训练的教师模型对样本图像中各像素的类别预测结果;信息提取模块,用于根据所述已训练的教师模型对所述样本图像中各像素的类别预测结果,得到所述样本图像中各像素的信息量;函数获取模块,用于在各预设图像类别下,基于所述样本图像中各像素的信息量、所述已训练的教师模型对所述样本图像中各像素的类别预测结果和待训练的学生模型对所述样本图像中各像素的类别预测结果,得到所述待训练的学生模型的目标损失函数;模型获取模块,用于根据所述目标损失函数,对所述待训练的学生模型进行迭代训练,得到训练完成的学生模型;所述训练完成的学生模型用于对输入的图像进行语义分割。12.一种计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,其特征在于,所述处理器执行所述计算机程序时实现权利要求1至10中任一项所述的方法的步骤。13.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1至10中任一项所述的方法的步骤。

技术总结
本申请涉及一种模型训练方法、装置、计算机设备、存储介质和计算机程序产品。所述方法包括:获取已训练的教师模型对样本图像中各像素的类别预测结果;根据已训练的教师模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息量;在各预设图像类别下,基于样本图像中各像素的信息量、已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到待训练的学生模型的目标损失函数;根据目标损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型;训练完成的学生模型用于对输入的图像进行语义分割。采用本方法能够提升学生模型整体的预测准确性。方法能够提升学生模型整体的预测准确性。方法能够提升学生模型整体的预测准确性。


技术研发人员:田倬韬 易振彧 刘枢 吕江波 沈小勇
受保护的技术使用者:苏州思谋智能科技有限公司
技术研发日:2022.03.21
技术公布日:2022/4/15
转载请注明原文地址:https://win.8miu.com/read-1136105.html

最新回复(0)