logo好方法网

基于抗噪声元学习的人脸识别方法及系统


技术摘要:
本发明公开了基于抗噪声元学习的人脸识别方法及系统,首先准备人脸识别数据集,并且数据集中存在大量噪声标签,然后对数据进行预处理,生成噪声标签,并按照特定方法初始化权重参数;随后构建人脸识别模型及其元学习模型学习策略并开始训练以及进行权重参数的更新。在  全部
背景技术:
人脸识别的应用范围十分广阔,如视频监控、签到系统、人机交互等,由于其非强 制性和非接触性,并且直观、简便,使得人脸识别具有良好的发展前景。传统的人脸识别的 做法是采集人脸的图像,从图像中抽取人脸特征进行识别。随后的发展中,基于支持向量 机、基于小波变换、基于神经网络的人脸识别都不断出现。随着深度学习的发展,使用大样 本人脸图像数据集训练深层神经网络模型进行人脸识别逐渐成为主流方法。这些人脸识别 需要庞大的数据集进行深度神经网络的训练,而网络中廉价的大数据样本中会存在相当一 部分的错误标签,这会使得训练的深度学习模型不稳定,效果不佳。尽管可以过滤和除去错 误的标签,但这样做代价很大。
技术实现要素:
本发明为解决由于大型人脸识别数据集存在较多错误标签,使得训练后的人脸识 别模型存在不稳定,识别效果不佳的问题,提供了基于抗噪声元学习的人脸识别方法及系 统。 为实现以上发明目的,而采用的技术手段是: 基于抗噪声元学习的人脸识别方法,包括以下步骤: S1.获取人脸数据集及其相应标签,所述标签中包括噪声标签及干净标签; S2.对所述人脸数据集进行预处理,对于其中各人脸数据,生成合成噪声标签,且 各人脸数据初始化为相同的权重参数; S3.构建人脸识别模型及其元学习模型学习策略,所述人脸识别模型的输入为待 识别的人脸数据,输出所述待识别的人脸数据的识别结果;所述元学习模型学习策略采用 梯度下降算法更新人脸识别模型的权重参数; S4.训练元学习模型,并对其权重参数进行更新; S5.采用元学习模型学习策略对所述人脸识别模型的参数进行优化,将待识别人 脸数据输入优化后的人脸识别模型,得到人脸识别结果。 优选的,所述步骤S2具体包括步骤:从所述人脸数据集中采样取得小批量人脸数 据(X,Y),其中X={x1 ,… ,xk}是k个人脸数据,Y={y1 ,… ,yk}是各人脸数据相应的噪声标 签;对于每个噪声标签,生成多个噪声标签{Y′1 ,…,Y′M},其分布与Y相似,从而一组生成的 噪声标签Y′m={y′m1,…,y′mk};对于各人脸数据均初始化为相同的权重参数。 优选的,所述步骤S4具体包括步骤: S41 .从含k个人脸数据的小批量人脸数据中随机选择若干个样本,对于每个选定 的样本Xi,根据其所在小批量人脸数据对其相邻的样本进行排序; 5 CN 111597907 A 说 明 书 2/6 页 S42.从中随机选择一个相邻样本xi,以及该xi最接近的前s个邻居,并使用邻居的 标签Yi替换Xi的标签,即得到生成的噪声标签 S43.重复执行M次步骤S41和S42,生成M个小批量的噪声标签;θ表示当前模型的参 数,对于每个合成的小批量人脸识别数据集样本,在其所在小批量人脸数据上使用梯度下 降算法将θ更新为θ′m,公式如下: X表示人脸数据, 表示生成的噪声标签,步长α是固定的超参数。 优选的,所述步骤S5具体包括步骤: 将θ参数化的模型定义为学生模型,构造由θ′参数化的教师模型,其参数为学生参 数的指数移动平均值;每次训练时采用以下公式更新教师模型的参数: θ′=γθ′ (1-γ)θ 其中γ是平滑系数超参数; 由于教师模型不受合成噪声标签的影响,因此强制计算一致性损失,该一致性损 失确保每个更新的模型参数θ′m与同一输入样本上的教师模型具有一致的预测;元损失定 义为所有一致性损失的平均值,对元学习的损失函数进行梯度下降,在元学习权重参数更 新之后,对原始的小批量人脸数据进行随机梯度下降优化分类损失函数; 多次迭代以上步骤,直至元训练模型与教师模型预测稳定相同且达到预设准确 率,得到优化后的基于抗噪声元学习的人脸识别模型,将待识别人脸数据输入该模型后得 到人脸识别结果。 本发明还提供了基于抗噪声元学习的人脸识别系统,包括: 数据获取模块,用于获取人脸数据集及其相应标签,所述标签中包括噪声标签及 干净标签; 数据初始化模块,用于所述人脸数据集进行预处理,对于其中各人脸数据,生成合 成噪声标签,且对各人脸数据初始化相同的权重参数; 人脸识别模型及其元学习模型学习策略构建模块,用于构建人脸识别模型及其元 学习模型学习策略,所述人脸识别模型的输入为待识别的人脸数据,输出所述待识别的人 脸数据的识别结果;所述元学习模型学习策略采用梯度下降算法更新人脸识别模型的权重 参数; 元学习模型训练模块,用于训练元学习模型,并对其权重参数进行更新; 人脸识别模型优化及识别模块,用于采用元学习模型学习策略对所述人脸识别模 型的参数进行优化,将待识别人脸数据输入优化后的人脸识别模型,得到人脸识别结果。 优选的,所述数据初始化模块进一步用于从所述人脸数据集中采样取得小批量人 脸数据(X,Y),其中X={x1,…,xk}是k个人脸数据,Y={y1,…,yk}是各人脸数据相应的噪声 标签;对于每个噪声标签,生成多个噪声标签{Y′1 ,…,Y′M},其分布与Y相似,从而一组生成 的噪声标签Y′m={y′m1,…,y′mk};对于各人脸数据均初始化为相同的权重参数。 优选的,所述元学习模型训练模块进一步用于: 从含k个人脸数据的小批量人脸数据中随机选择若干个样本,对于每个选定的样 本Xi,根据其所在小批量人脸数据对其相邻的样本进行排序;从中随机选择一个相邻样本 6 CN 111597907 A 说 明 书 3/6 页 xi,以及该xi最接近的前s个邻居,并使用邻居的标签Yi替换Xi的标签,即得到生成的噪声标 签 重复执行M次步骤S41和S42,生成M个小批量的噪声标签;θ表示当前模型的参 数,对于每个合成的小批量人脸识别数据集样本,在其所在小批量人脸数据上使用梯度下 降算法将θ更新为θ′m,公式如下: X表示人脸数据, 表示生成的噪声标签,步长α是固定的超参数。 优选的,所述人脸识别模型优化及识别模块进一步用于: 将θ参数化的模型定义为学生模型,构造由θ′参数化的教师模型,其参数为学生参 数的指数移动平均值;每次训练时采用以下公式更新教师模型的参数: θ′=γθ′ (1-γ)θ 其中γ是平滑系数超参数。 由于教师模型不受合成噪声标签的影响,因此强制计算一致性损失,该一致性损 失确保每个更新的模型参数θ′m与同一输入样本上的教师模型具有一致的预测;元损失定 义为所有一致性损失的平均值,对元学习的损失函数进行梯度下降,在元学习权重参数更 新之后,对原始的小批量人脸数据进行随机梯度下降优化分类损失函数; 多次迭代以上步骤,直至元训练模型与教师模型预测稳定相同且达到预设准确 率,得到优化后的基于抗噪声元学习的人脸识别模型,将待识别人脸数据输入该模型后得 到人脸识别结果。 与现有技术相比,本发明技术方案的有益效果是: 本发明的基于抗噪声元学习的人脸识别方法及系统,通过元学习模型学习策略学 习人脸识别模型的参数,从而准备噪声标签模型;本发明的元学习训练方法,其训练过程省 时、高效,可有效防止模型对特定噪声过拟合。本发明可以在具有大量噪声标签的数据集中 稳定训练人脸识别模型,无需对其中的错误标签进行人为过滤和去除,并使得训练得到的 人脸识别模型具有较强的鲁棒性,可以高效、准确对人脸进行识别。本发明的基于抗噪声元 学习的人脸识别方法及系统可广泛应用于基于各个大型人脸识别数据集,尤其是网络上便 于采集的、廉价的、具有不规则标签噪声的数据集的人脸识别。此外,本发明也可以在干净 的数据集上实现良好的人脸识别性能。 附图说明 图1为实施例1的方法流程图。 图2为实施例2的系统模块图。
下载此资料需消耗2积分,
分享到:
收藏