人脸识别模型的训练方法、人脸识别的方法和装置制造方法及图纸

本申请提供了一种人脸识别模型的训练方法、人脸识别的方法和装置，涉及人工智能的机器学习领域。该模型训练方法包括：获取第一训练样本集；将该第一训练样本集分别输入至少两个第一人脸识别模型，得到每个第一人脸识别模型输出的第一图像特征；将每个第一人脸识别模型输出的第一图像特征输入每个第一人脸识别模型对应的模型置信度估计模块，得到每个第一人脸识别模型输出的第一图像特征的置信度；根据每个第一人脸识别模型输出的第一图像特征的置信度，对至少两个第一人脸识别模型输出的第一图像特征进行融合，得到融合特征；根据该融合特征对第二人脸识别模型进行知识蒸馏，得到训练后的人脸识别模型。本申请实施例能够提高人脸识别准确率。高人脸识别准确率。高人脸识别准确率。

【技术实现步骤摘要】
人脸识别模型的训练方法、人脸识别的方法和装置


[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]第六方面，本申请实施例提供一种计算机可读存储介质，包括指令，当其在计算机上运行时使得计算机执行如第一方面或第二方面中的方法。
[0029]第七方面，本申请实施例提供一种计算机程序产品，包括计算机程序指令，该计算机程序指令使得计算机执行如第一方面或第二方面中的方法。
[0030]第八方面，本申请实施例提供一种计算机程序，计算机程序使得计算机执行如第一方面或第二方面中的方法。
[0031]通过上述技术方案，能够根据多个第一人脸识别模型输出特征的置信度，对多个第一人脸识别模型的特征进行融合，然后根据融合后的特征对第二人脸识别模型进行知识蒸馏，得到训练后的人脸识别模型，使得训练后的人脸识别模型所提取的特征逐渐收敛至与融合后的特征的分布一致。由于融合后的特征可以更加准确地表达特征的真实分布，因此通过知识蒸馏得到的人脸识别模型能够提取到更加准确的人脸图像特征，进而部署得到的人脸识别系统能够具有较高的人脸识别准确率。
[0032]另外，由于融合特征是采用多个第一人脸识别模型输出的特征以及其置信度得到的，因此基于该融合特征进行知识蒸馏得到的第二人脸识别模型具有较强的鲁棒性。
附图说明
[0033]图1为本申请实施例的方案的系统架构的一个示意图；
[0034]图2为本申请实施例提供的人脸识别方案的一个示意流程图；
[0035]图3为本申请实施例提供的第一人脸识别模型的训练方法的一个示意图；
[0036]图4为本申请实施例提供的第一人脸识别模型的训练方法的另一示意图；
[0037]图5为本申请实施例提供的模型置信度估计模本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种人脸识别模型的训练方法，其特征在于，包括：获取第一训练样本集，所述第一训练样本集包括多个人脸图像样本；将所述第一训练样本集分别输入至少两个第一人脸识别模型，得到每个所述第一人脸识别模型输出的第一图像特征；其中，所述至少两个第一人脸识别模型分别利用所述人脸图像样本对至少两个初始化第一人脸图像识别模型训练得到；将每个所述第一人脸识别模型输出的第一图像特征，输入每个所述第一人脸识别模型对应的模型置信度估计模块，得到所述每个第一人脸识别模型输出的第一图像特征的置信度；其中，每个所述第一人脸识别模型对应的模型置信度估计模块是利用每个所述第一人脸识别模型输出的图像特征和每个所述人脸图像样本对应的样本类中心向量对初始化模型置信度估计模块训练得到；根据每个所述第一人脸识别模型输出的第一图像特征的置信度，对所述至少两个第一人脸识别模型输出的第一图像特征进行融合，得到融合特征；根据所述融合特征对第二人脸识别模型进行知识蒸馏，得到训练后的所述人脸识别模型，所述人脸识别模型用于对待识别的人脸图像进行人脸识别。2.根据权利要求1所述的方法，其特征在于，所述根据所述融合特征对第二人脸识别模型进行知识蒸馏，得到训练后的所述人脸识别模型，包括：将所述第一训练样本集输入第二人脸识别模型，得到第二图像特征；根据所述融合特征和所述第二图像特征的相似度，确定知识蒸馏损失；根据所述知识蒸馏损失，对所述第二人脸识别模型的参数进行调整，直到满足第一停止训练条件，并将满足所述第一停止训练条件所确定的所述第二人脸识别模型输出为所述人脸识别模型。3.根据权利要求1所述的方法，其特征在于，还包括：重复执行以下步骤，直至每个所述第一人脸识别模型满足第二停止训练条件：获取第二训练样本集，所述第二训练样本集包括多个人脸图像样本；将所述第二训练样本集中的当前人脸图像样本输入当前第一人脸识别模型，得到所述当前人脸图像样本对应的第一当前图像特征；获取所述当前人脸图像样本对应的当前类中心向量；其中，所述当前类中心向量用于指示所述当前人脸图像样本所属的类别；根据所述第一当前图像特征、所述当前类中心向量和所述当前人脸图像样本的标签信息，得到所述当前第一人脸识别模型的第一当前损失；根据所述第一当前损失，对所述当前第一人脸识别模型的当前参数进行调整；将满足所述第二停止训练条件所确定的所述当前第一人脸识别模型输出为所述第一人脸识别模型。4.根据权利要求3所述的方法，其特征在于，所述根据所述第一当前图像特征、所述当前类中心向量和所述当前人脸图像样本的标签信息，得到所述当前第一人脸识别模型的第一当前损失，包括：根据所述第一当前图像特征、所述当前类中心向量、所述当前人脸图像样本的标签信息和超参数，得到所述当前第一人脸识别模型的第一当前损失，其中，所述至少两个第一人脸识别模块的对应的超参数不同。
5.根据权利要求1所述的方法，其特征在于，所述至少两个初始化第一人脸图像识别模型对应的初始值不同。6.根据权利要求1所述的方法，其特征在于，还包括：重复执行以下步骤，直至每个所述第一人脸识别模型对应的模型置信度估计模块满足第三停止训练条件：获取第三训练样本集，所述第三训练样本集包括多个人脸图像样本；将所述第三训练样本集中的当前人脸图像样本输入每个所述第一人脸识别模型，得到所述当前人脸图像样本对应的第二当前图像特征；获取所述当前人脸图像样本对应的当前类中心向量；其中，所述当前类中心向量用于指示所述当前人脸图像样本所属的类别；将所述第二当前图像特征输入当前模型置信度估计模块，得到所述第二当前图像特征的当前置信度；其中，所述当前置信度用于评估所述当前人脸图像样本通过每个所述第一人脸识别模型进行人脸识别的置信度；根据所述第二当前图像特征、所述当前类中心向量和所述当前置信度，得到所述当前模型置信度估计模块的第二当前损失；根据所述第二当前损失，对所述当前模型置信度估计模块的当前参数进行调整；将满足所述第三停止训练条件所确定的所述当前模型置信度估计模块输出为...

【专利技术属性】
技术研发人员：许剑清，
申请(专利权)人：腾讯科技深圳有限公司，
类型：发明
国别省市：