面向回归任务的数值型标签噪声过滤算法

姜高霞1 王文剑1,2

1(山西大学计算机与信息技术学院 太原 030006)2 (计算智能与中文信息处理教育部重点实验室(山西大学) 太原 030006)

摘 要 回归任务中的数值型标签噪声可能误导模型训练,进而弱化模型泛化能力.作为一种常用的标签噪声处理技术,噪声过滤通过去除误标记样本来降低噪声水平,但无法保证过滤后模型能够获得更好的泛化表现.一些过滤算法过于关注噪声水平,以至于大量无噪样本也被去除.尽管已有样本过滤框架能够平衡样本去除量和噪声水平,但其形式过于复杂不利于直观理解和实际应用.根据无噪回归任务中的学习理论提出了面向数值型标签噪声数据的泛化误差界,从而明确了影响模型泛化能力的关键数据因素(数据量和噪声水平).在此基础上提出一种可解释的噪声过滤框架,其目标是以较小的样本去除代价最大程度地降低噪声水平.针对噪声估计问题,从理论上分析了噪声与覆盖区间关键指标(中心和半径)之间的变化趋势,进而构建了相对噪声估计方法.此方法与所提框架结合形成了相对噪声过滤(relative noise filtering, RNF)算法.在标准数据集和年龄估计数据上均验证了算法的有效性.实验结果表明:该算法能够适应各类噪声数据,显著提升模型泛化能力.在年龄估计数据上RNF算法检测出一些标签噪声数据,有效提升了数据质量和模型预测性能.

关键词 数值型标签噪声;回归;噪声过滤;泛化误差界;相对噪声

数据质量是决定机器学习模型可靠性的关键因素之一,尤其在高风险智能应用(如癌症检测、贷款分配等)中数据质量至关重要.然而数据质量在机器学习中起到的作用正在被低估,“每个人都想做模型工作,而不是数据工作”[1].监督学习是机器学习中比较成熟的学习模式,在人脸表情识别、医学诊断等领域不断取得成功应用,但都特别依赖以较高成本获得的大规模、高质量标注数据[2-4].实际中的多数数据,包括大量公开的标准数据集(如MNIST,CIFAR,ImageNet等),可能存在大量的标签噪声,QuickDraw数据的标签错误率甚至超过10%[5].诺贝尔奖得主Kahneman认为噪声是影响人类判断的黑洞,实际存在的噪声远比人们预期的要多[6].加州大学的知名专业学者都将噪声的干扰列为人工智能和机器学习领域的重要挑战之一[7].

监督学习根据输出数据类型分为分类和回归任务.标签噪声在两类任务中分别对应了类别型标签噪声(如将性别标签“男”误标为“女”)和数值型标签噪声(如将年龄标签“3”误标为“20”).通常数值型标签噪声问题更为复杂也更难解决,主要原因是数值型噪声取值范围更广、分布类型更复杂[8-10].实际中人们不知道一个数据集中是否有标签噪声,即使有噪声也不知道哪些样本有噪声、噪声有多大、噪声有多少、噪声服从什么分布等.因此解决回归任务中的数值型标签噪声问题是一项具有挑战性的任务.

对于标签噪声问题主要有2种解决思路:1)从算法层通过重构损失函数、样本加权或集成方式建立噪声鲁棒模型[11-13],这些模型并非对噪声完全鲁棒,还会在一定程度上受到噪声干扰[14],在未知噪声和强噪声情况下这些模型表现不够满意[15];2)从数据层面实施噪声过滤或纠正,也就是要将错误标签数据去除或改正[16].相比于鲁棒建模方法,噪声过滤方法只需要给出样本过滤结果,不需要对模型进行任何改动,因此噪声过滤的普适性更强、门槛更低.这类方法虽能降低数据噪声水平,但无法保证模型泛化能力,而且难以适应不同噪声环境.

针对回归任务中的数值型标签噪声问题,本文从泛化误差界视角分析了数据因素(样本量和噪声水平)对模型泛化能力的影响,并据此构建了样本过滤框架和噪声过滤算法.本文工作的主要贡献包括4个方面:

1) 修正了标签噪声环境下回归模型的泛化误差界,明确了影响模型泛化能力的关键因素,为提升模型泛化能力的样本过滤算法提供了理论指导.

2) 以降低泛化误差界为目标,提出一种关于相对噪声水平和相对样本量的可解释样本过滤框架,它可以与一般噪声估计方法结合,形成新的过滤算法.

3) 分析了噪声与覆盖区间中心和半径的单调关系,据此提出一种相对噪声估计方法,并与前面所提样本过滤框架结合设计了相对噪声过滤算法.

4) 在标准数据集和真实人脸年龄估计数据集的实验结果表明,所提算法可以有效提升数据质量和模型的泛化能力.

1 相关工作

标签噪声过滤通过去除噪声样本提升含标记数据的质量,进而提高监督学习模型的预测性能.

由于k近邻模型对标签噪声较为敏感,k近邻模型常用于检测和过滤标签噪声.其主要思想是,如果一个样本的标签与其多数近邻的标签不一致,则认为出现标签噪声.典型算法有编辑近邻(edited nearest neighbor, ENN)过滤器、全编辑近邻(all edited nearest neighbor, ANN)过滤器[17]、近邻感知(perception of nearest neighbor, PNN)过滤器[18].在回归任务中,如果模型误差超过一定阈值,则认为出现数值型标签噪声,这是一种面向回归的编辑近邻过滤算法(edited nearest neighbor for regression, RegENN)[9].

集成过滤方法利用多个分类模型产生样本预测标签,如果多数预测标签与其实际标签不一致,则认为标签存在错误应当去除[19-20].这些基模型可以采用不同类模型(如多数投票过滤),也可能是经不同子集训练过的同类模型(如迭代划分过滤).

受到特征选择的启发,文献[21]提出一种基于互信息(mutual information, MI)的噪声过滤算法.如果样本去除其近邻后特征与标签之间的互信息均发生明显的变化(超过某个阈值),则认为此样本存在标签噪声.

回归任务的标签可以通过离散化的方式将其转化为分类任务,这样就可以利用分类标签噪声过滤算法来识别回归中的数值型标签噪声.离散编辑近邻(edited nearest neighbor based on discretization, DiscENN)对数值型标签做离散化处理后采用近邻过滤来识别噪声[22].

多数噪声过滤算法虽然能够降低数据的噪声水平,但无法保证提升模型预测性能.为此文献[10]从泛化误差界视角提出一种噪声过滤的最优样本过滤框架和覆盖距离过滤(covering distance filtering, CDF)算法.此框架为降低过滤后模型的泛化误差界提供了理论保障,对指导噪声过滤具有重要意义.

上述方法中,MI,RegENN,DiscENN,CDF是面向回归任务的数值型标签噪声过滤算法.部分算法的过滤效果比较依赖阈值,而阈值通常根据经验指定,缺乏对噪声数据的自适应性.阈值设置不当容易导致过度清洗,也就是去掉大量无噪样本.虽然文献[10]给出了过滤的理论依据,但过滤目标函数中包含参数较多且形式复杂,不利于直观理解和实际应用.在这些参数中,误差界的置信度对结果影响不大,但增加了目标函数复杂度;有些模型的VC维是无穷或不可计算,在实际应用中需要根据经验指定.

2 泛化界视角下的噪声过滤框架

本节通过修正无噪条件下的学习理论得到含标签噪声情况下的泛化误差界,并据此提出含噪数据的过滤框架.

2.1 基本定义

设有回归数据集表示第i个样本的输入特征或属性,yi表示第i个样本的实值输出标签.若数据中存在数值型标签噪声,则实际标签yi可能不等于未知的真实标签标签噪声:

(1)

定义1. 真实经验误差.回归模型f(x)经数据D训练后基于无噪标签的真实经验误差:

(2)

定义2. 实际经验误差.模型在第i个样本的实际误差ri=f(xi)-yi,基于含噪标签的实际经验误差:

(3)

实际中由于真实标签未知,真实经验误差Remp(f,D)无法计算.虽然可以采用式(3)来近似真实经验误差,但通常会低估真实误差.

2.2 泛化误差界

引理1[23-25]. 对于平方损失下的无噪回归任务,以下泛化误差(上)界以1-δ概率成立:

R(f,D)≤Remp(f,Dε(D),

(4)

其中,R(f,D)表示经数据D训练后模型f的泛化误差,Remp(f,D)表示模型fD上的经验误差,是关于D的样本量n的函数,h表示模型VC维,δ为概率常数.

定理1. 对于回归任务设标签yi含有零均值标签噪声,则在数据D上训练后的模型f(x)具有泛化误差(上)界:


2Cov(ei,ri)]×ε(D).

(5)

证明. 含标签噪声时的真实经验误差:

2Cov(ei,ri)+2ED(ei)ED(ri)=

(6)

其中模型误差ri=f(xi)-yiE(·)为期望函数,Cov(ei,ri)表示噪声ei与模型误差ri的协方差.

ε(D)不受标签噪声的影响.由引理1可得模型具有泛化误差(上)界:

(7)

证毕.

当噪声与模型误差相互独立时,协方差Cov(ei,ri)=0.这时其中表示数据D的平均噪声水平.可见,当数据存在标签噪声时,实际经验误差会低估真实经验误差,进而导致泛化误差界被低估.也就是说,标签噪声使得模型泛化误差(界)变大,因此传统泛化误差界理论不再适用.

2.3 噪声过滤框架

本节利用定理1的结果确定影响泛化误差界的关键因素,并据此来构造合理的噪声过滤目标函数.

噪声过滤的目的是通过去除含噪样本来提升数据质量和模型泛化能力.将原始数据D经过过滤后的数据记为D*.为使得回归模型f(x)经数据D*训练后的泛化性能最佳,根据定理1可得误差界最低的目标:


2Cov(ei,ri)]×ε(D*),

(8)

其中,模型的实际经验误差一般在模型训练阶段可以获得较低值.

由于噪声均值为零(ED*(ei)=0),噪声方差这时协方差:

其中Var(·)和ρ(·,·)分别表示方差和相关系数.将协方差代入目标函数可得:

(9)

在模型层面,给定数据的情况下模型经验误差和误差方差Var(ri)均固定;在数据层面,需要降低噪声项ε(D*)以使得目标函数(泛化误差界)最小.考虑到ε(D*)是关于D*的样本量n*的单调递减函数,要使目标函数尽可能小,需要降低噪声项n*较大.总之,在保留较多样本(较大的n*)的前提下尽可能降低噪声水平(较小的能够使模型获得较好的泛化性能(较低的泛化误差界).换言之,噪声过滤需要以较低的样本舍弃代价尽可能地降低平均噪声,才能显著提升模型泛化性能.

为了将和max n*均等地综合起来,这里采用相对值来构造数据过滤层面的总目标函数:

(10)

其中,n*/n表示数据集D*D的相对样本量,表示它们的相对噪声水平.

式(10)利用相对样本量和相对噪声水平来获得较低的泛化误差界,因此只需给出每个样本的噪声相对估计值即可计算出目标函数值.式(10)可以与任意的噪声相对估计方法和绝对估计方法相结合,因此它是一种适用面较广的噪声过滤框架.相比于文献[10]中的过滤框架,式(10)放弃了次要因素(误差界的置信度和模型VC维),精简了目标函数,使其可解释性更强,且其中不需要预先指定任何参数.

图1给出了噪声过滤框架的模拟结果.图中横坐标表示去噪比例γ=1-n*/n,直线表示相对样本量随γ的变化趋势,曲线表示相对噪声水平随γ的变化趋势.其中样本量设为1 000,噪声比例为25%,噪声服从正态分布N(0,0.52).由于实际中无法将所有噪声准确估计和排序,这里设定75%的噪声能够正确排序.按照噪声先大后小的顺序依次去除,重复200次后得到平均相对噪声水平曲线.

Fig. 1 Simulation of noise filtering framework
图1 噪声过滤框架模拟

总体上,随着γ增大,过滤后数据集的相对样本量变少,噪声水平也变低,但噪声水平的变化一般遵从先快后慢的规律.这是因为刚开始去除较大的噪声,噪声水平下降较快;后面噪声较小后难以准确去除明显很大的噪声,故噪声水平下降缓慢;当噪声水平相差不大或噪声估计难以区分低噪声样本时,噪声水平趋于稳定.实际上,当噪声水平下降较慢时应当停止样本去除,这个位置处于两条线的最大间隔处,也就是使得式(10)达到最优的过滤结果.图1中约为γ=0.16,即应当去除16%的具有较大噪声的样本.

3 相对噪声估计和过滤算法

本节提出一种相对噪声估计方法,结合所提过滤框架得出新的过滤算法.

3.1 相对噪声估计

由于真实标签未知,标签噪声无法被准确计算.但是可以寻找一个以较大概率包含的区间,通过计算实际标签yi与此区间的偏离程度大致估计噪声大小.

定义3. 覆盖区间.区间[u,v]以较大的概率pC包含或覆盖真实输出标签则称[u,v]为的一个覆盖区间,其覆盖概率

假设有J个独立的基模型表示第j个模型对yi的预测值.为满足预测值之间的独立性,模型预测通过子集方式生成:首先将数据集随机划分为J个子集,然后用每个子集数据对回归模型进行训练,并在全部数据上进行预测.

真实标签有一定概率落入模型预测值之间.令:

(11)

则其覆盖概率:

(12)

根据等同无知原则,在真实标签未知的情况下可设则覆盖概率:

(13)

显然J越大,覆盖概率越大.但过大的J值可能使得训练子集规模过小,模型预测误差过大.实际中取J=5以平衡训练数据规模和覆盖概率,此时覆盖概率pC=0.937 5.

考虑到可以假设的分布密度函数关于式(11)中区间[u,v]的中心对称.

定理2. 假设真实标签服从正态分布N(c,σ2),其中式(11)区间的中心c=(u+v)/2,则:

(14)

其中区间半径s=(v-u)/2,实际标签到区间中心的距离d=|yi-c|.

证明. 期望绝对噪声:

(15)

不妨设yi>c,则实际标签到区间中心的距离d=yi-c>0,

(16)

由奇偶函数积分的性质可知:

(17)

(18)

因此有:

(19)

根据函数求导法则可得:

(20)

(21)

服从正态分布可得:

(22)

此时覆盖概率

(23)

其中,Φ(·)表示标准正态分布的分布函数.

根据式(13)可知,覆盖概率为常数1-21-J,故

J为正整数可知-Φ-1(2-J)>0,因此:

(24)

yi<c时同理可证.

证毕.

根据上述理论分析可知,当ds相差不大时可见d对噪声的影响比s更明显.据此给出噪声估计.

定义4. 相对噪声的定义为

(25)

其中,cs分别为式(11)区间的中心和半径,即c=(u+v)/2,s=(v-u)/2,实际标签到区间中心的距离d=|yi-c|.

式(25)中对s做开方处理是由于其影响比d.公式中考虑了噪声与其关键因素的正反比关系,并未设定噪声与ds的比例常数,因此是一种相对噪声估计方法.直观上,标签噪声越大,则yi到区间中心的距离越远(d越大);噪声越大,则分布的方差和区间半径越大.总体上相对噪声与ds成正比.

3.2 相对噪声过滤

通常大噪声样本应当先被去掉以获得较低的噪声水平.利用式(25)来估计所有标签噪声,然后逐个去除剩余子集中的最大噪声样本,并计算式(10)中的目标函数值.根据最大目标函数值即可找到最佳过滤结果.

算法1. 相对噪声过滤(RNF)算法.

输入:回归数据集基模型f(x);

输出:过滤后数据集D*.

① 将数据集D随机划分为5个子集,然后用每个子集数据对回归模型f(x)进行训练,并在全部数据上进行预测;

② 根据式(11)计算覆盖区间[u,v],并用式(25)计算噪声估计值;

③ 将数据按照噪声从大到小顺序排列得到D′;

④ for t=0 to n-1

其中

⑥ end for

⑦ 寻找最大目标函数t*=arg max Ft,最优过滤数据集

算法1在有限集合中求解固定的优化目标函数,因此一定存在最优解.实际中采用3近邻回归作为基模型.算法1中模型训练和预测的时间复杂度为O(n log n),其余部分均为线性时间复杂度,因此算法总时间复杂度为T(RNF)=O(n log n).如果基模型采用其他模型,则算法总时间复杂度与基模型复杂度相同.

4 实验结果及分析

本节首先介绍了过滤算法在标准数据集的实验框架、理论结果验证、实验结果与分析,然后在真实人脸年龄估计数据上做了标签噪声检测和泛化性能分析.

4.1 实验框架

实验中首先将原始标准回归数据集随机划分为训练集和测试集,两者的样本比例为7∶3;然后随机在训练集的部分输出标签上添加人工噪声,并用各种噪声过滤算法来识别噪声和过滤样本;最后将回归模型在过滤后的数据集上进行训练,并在测试集上测试其泛化能力.实验中每轮数据划分、加噪声、过滤和预测环节均重复5次以获得稳定结果.

表1列出了实验中使用的15个标准回归数据集[26-27].数据的数值型特征均被归一化到区间[-1,1].

Table 1 Dataset Information
表1 数据集信息

数据集编号数据集样本量特征数1Housing506132Forest Fires517133Energy Efficiency76884Concrete Compressive Strength103095Geographical Original of Music1059686MG138567Airfoil Self-Noise150368Space_ga310769SkillCraft1 Master Table33952010Abalone4177811Parkinsons Telemonitoring58752612Cpusmall81921213Combined Cycle Power Plant9568414Condition Based Maintenance119341615Physicochemical Properties of Protein457309

实验中包括8种数值型标签噪声,分别服从均匀分布U(-1,1)、均匀分布U(-1.5,1.5)、高斯分布N(μ=0,σ=1)、高斯分布N(μ=0,σ=1.5)、拉普拉斯分布Lp(μ=0,σ=1)、拉普拉斯分布Lp(μ=0,σ=1.5)、高斯混合分布N(μ=1,σ=0.3)+N(μ=-1,σ=0.3)、高斯混合分布N(μ=1,σ=0.1)+N(μ=-1.2,σ=0.5).最后2个混合分布中单个高斯分布噪声各占50%.噪声比例NR=0.1,0.2,0.3,0.4.

对比过滤算法包括基于互信息(MI,阈值为0.05,近邻数为9)的过滤[21]、回归近邻过滤(RegENN,阈值为5,近邻数为9)[9]、离散近邻过滤(DiscENN,近邻数为9)[22]、覆盖距离过滤(CDF,子集数为5)[10].此外,所提相对噪声过滤算法(RNF)还与未过滤(NoF)数据做了对比分析.

测试模型包括k近邻回归模型(kNN)、支持向量回归模型(SVR)、高斯过程回归(GPR)和随机森林(RF).模型在测试集上的泛化能力采用均方误差(mean square error, MSE)来度量:

(26)

4.2 理论结果验证

图1噪声过滤框架模拟结果显示,当相对样本量和相对噪声水平具有最大间隔时,式(10)取得最优解,此时的去噪比例较为合适,能够使模型获得较好的泛化能力.为验证此结论,在2个数据集上人工添加2类噪声,并在不同去噪比例下测试了模型的预测误差.具体设置为:1)对表1中第7个数据训练集30%的标签添加服从均匀分布U(-1.5,1.5)的噪声;2)对表1中第9个数据训练集30%的标签添加服从高斯分布N(μ=0,σ=1)的噪声.2种情况下均设置去噪比例γ=0∶0.02∶0.5对数据进行过滤,然后使用SVR,GPR和RF模型进行训练(kNN的测试误差较大),并在无噪测试集上测试模型误差.

图2显示了2种设置下模型测试误差(MSE)在不同去噪比例下的变化趋势.各模型最低测试误差采用实心圆点标出.目标函数曲线对应右侧坐标轴,在目标函数最大值(即相对样本量与相对噪声水平的最大间隔)处用竖虚线标出.图2(a)中,GPR模型在目标函数最大值处具有最小测试误差,其余2个模型在目标函数最大值处的误差非常接近最小测试误差.图2(b)中,3个模型在目标函数最大值附近具有最小测试误差.由此可见,模型最小测试误差下的去噪比例通常位于目标函数最大值附近.根据目标函数最大值来确定去噪比例,能够使模型获得最低或较低的测试误差,实验证实了所提框架的有效性.

Fig. 2 Model test errors under different denoising ratios
图2 不同去噪比例下的模型测试误差

定理2表明期望噪声与ds均成正比,其中覆盖区间半径s=(v-u)/2,实际标签到覆盖区间中心的距离d=|yi-c|.为验证此结论,在表1所有数据上添加6种人工噪声,并构造覆盖区间.根据绝对噪声|ei|和区间特征s,d可以分别计算出它们的Pearson相关系数及其相关性检验的p.相关系数和p值在所有数据上的平均值在表2中列出,其中p值越小表明相关性越显著.

Table 2 Correlations Between Noise and the Characteristics of Covering Interval
表2 噪声与覆盖区间特征的相关性

噪声分布|ei|与s|ei|与d相关系数p值相关系数p值U(-1,1)0.2060.0430.7340.000U(-1.5,1.5)0.2310.0180.8230.000N(μ=0,σ=1)0.2950.0020.8820.000N(μ=0,σ=1.5)0.3230.0000.9150.000Lp(μ=0,σ=1)0.3600.0000.9060.000Lp(μ=0,σ=1.5)0.3820.0000.9290.000

注:黑体数字表示<0.01的p值.

由表2可见,所有噪声情况下2组相关性检验的p值均<0.05且相关系数为正值,表明|ei|与s,d均显著正相关;|ei|与d的相关系数明显大于|ei|与s的相关系数,表明前者的相关性更强;从噪声分布来看,拉普拉斯分布噪声所对应的相关系数最大,均匀分布噪声的相关系数较小.表2的实验结果表明定理2所表达的正反比关系在实际中是成立的.

4.3 实验结果与分析

表3列出了不同噪声比例下各数据集上不同过滤算法的测试误差结果.通常噪声比例越大,测试误差也越大.当噪声水平较低(NR=0.1,0.2)时,所提RNF过滤算法能够使得模型的测试误差最小,且有明显优势;当噪声水平较高(NR=0.3,0.4)时,RNF算法在大多数数据上取得了最小测试误差.

Table 3 Average Test Error ±Standard Deviation of Four Models with Different Noise Ratios
表3 不同噪声比例下4种模型的平均测试误差±标准差

噪声比例NR数据集编号NoFMIRegENNDiscENNCDFRNF10.812±0.1100.774±0.0860.768±0.0910.750±0.0800.601±0.0400.554±0.04221.386±0.0501.272±0.0431.298±0.0551.296±0.0521.004±0.0810.911±0.08130.493±0.1070.446±0.0940.440±0.0870.445±0.0900.246±0.0320.193±0.02640.543±0.1060.503±0.0920.520±0.1010.491±0.0900.313±0.0300.286±0.02950.756±0.1160.741±0.1130.721±0.1060.678±0.0910.538±0.0450.432±0.03761.685±0.1041.641±0.1211.568±0.0931.616±0.0911.338±0.0241.322±0.02271.804±0.1421.702±0.1131.720±0.1071.733±0.1231.396±0.0521.320±0.0510.181.054±0.1140.991±0.1110.987±0.0980.973±0.0930.797±0.0160.784±0.01590.625±0.0870.613±0.0860.624±0.0940.603±0.0810.498±0.0300.407±0.024101.383±0.1681.324±0.1471.343±0.1431.280±0.1191.077±0.0421.044±0.040110.493±0.1120.472±0.1020.443±0.0860.430±0.0820.268±0.0100.261±0.014121.884±0.1541.849±0.1261.878±0.1231.853±0.1251.656±0.0721.548±0.065131.750±0.1411.633±0.1221.624±0.1061.620±0.1041.391±0.0521.311±0.048140.392±0.1000.351±0.0850.341±0.0810.349±0.0800.157±0.0220.123±0.017150.499±0.1080.477±0.1010.461±0.0970.449±0.0920.291±0.0440.220±0.033

续表3

噪声比例NR数据集编号NoFMIRegENNDiscENNCDFRNF10.965±0.1390.904±0.1320.915±0.1310.854±0.1060.588±0.0400.538±0.04621.635±0.1141.449±0.0741.473±0.0711.520±0.0831.108±0.0590.998±0.06730.714±0.1770.619±0.1430.619±0.1410.575±0.1210.315±0.0350.286±0.03340.749±0.1690.626±0.1290.633±0.1320.592±0.1100.340±0.0350.324±0.03450.908±0.1760.867±0.1610.871±0.1580.836±0.1440.608±0.0530.509±0.04561.629±0.1881.630±0.1591.630±0.1391.542±0.1411.299±0.0381.264±0.03671.798±0.2071.761±0.1771.711±0.1511.691±0.1451.462±0.0491.408±0.0480.281.189±0.1571.151±0.1251.109±0.1251.111±0.1130.828±0.0290.818±0.02990.849±0.1440.841±0.1410.823±0.1340.791±0.1120.609±0.0480.503±0.041101.532±0.2271.471±0.1911.462±0.1851.418±0.1531.157±0.0621.133±0.061110.631±0.1670.532±0.1320.524±0.1290.480±0.0980.283±0.0170.278±0.017122.108±0.1712.112±0.1742.012±0.1582.075±0.1721.727±0.0651.635±0.061131.914±0.1791.787±0.1311.754±0.1521.789±0.1531.503±0.0591.422±0.056140.526±0.1570.414±0.1110.430±0.1230.391±0.0980.180±0.0270.141±0.021150.614±0.1710.530±0.1330.541±0.1370.503±0.1210.343±0.0520.255±0.03911.184±0.2011.073±0.1681.052±0.1411.002±0.1260.683±0.0470.651±0.04721.858±0.2211.635±0.1221.631±0.1371.574±0.1221.141±0.0341.033±0.03530.941±0.2010.842±0.1680.811±0.1560.769±0.1380.421±0.0470.397±0.04541.049±0.2290.871±0.1560.809±0.1390.838±0.1480.450±0.0470.456±0.04851.181±0.2121.133±0.1891.053±0.1571.063±0.1610.765±0.0720.686±0.06661.887±0.2361.782±0.1981.811±0.2051.735±0.1791.413±0.0261.399±0.02571.841±0.2161.823±0.2071.745±0.1661.759±0.1951.456±0.0641.454±0.0670.381.387±0.2081.249±0.1601.191±0.1331.196±0.1360.863±0.0310.860±0.03291.107±0.1891.087±0.1861.101±0.1691.011±0.1370.829±0.0650.695±0.052101.649±0.2321.603±0.2241.555±0.1961.492±0.1641.255±0.0671.236±0.066110.749±0.2140.651±0.1720.553±0.1240.528±0.1130.311±0.0240.310±0.024122.160±0.2652.066±0.2332.071±0.1621.992±0.1481.814±0.0611.732±0.059131.938±0.2141.873±0.1871.863±0.1741.847±0.1621.517±0.0581.453±0.056140.664±0.2050.554±0.1660.484±0.1280.464±0.1210.219±0.0320.199±0.029150.705±0.2120.646±0.1870.616±0.1730.569±0.1440.375±0.0570.270±0.04211.414±0.2291.299±0.1791.161±0.1371.176±0.1390.777±0.0630.813±0.06522.019±0.2941.790±0.2151.779±0.1931.799±0.2211.241±0.0521.111±0.02131.269±0.2261.231±0.2161.156±0.1871.113±0.1720.842±0.1190.886±0.12641.357±0.2691.221±0.2031.135±0.1871.162±0.1900.841±0.1140.892±0.12151.653±0.2761.577±0.2341.526±0.2241.472±0.2041.321±0.1861.238±0.17561.981±0.2811.919±0.2421.855±0.2211.865±0.1931.649±0.1451.644±0.14672.125±0.2881.993±0.2371.921±0.2071.914±0.2081.761±0.1541.759±0.1550.481.471±0.2541.355±0.1911.337±0.1811.245±0.1340.951±0.0410.977±0.04391.490±0.2061.484±0.2041.374±0.1381.367±0.1561.175±0.0681.037±0.065101.732±0.3121.630±0.2441.575±0.2141.580±0.2141.201±0.0641.199±0.065110.865±0.2560.681±0.1670.599±0.1360.577±0.1220.341±0.0330.358±0.035122.333±0.2752.343±0.2372.162±0.1822.176±0.1651.854±0.0611.787±0.061132.135±0.2532.091±0.2351.985±0.2021.935±0.1511.706±0.0671.661±0.066140.816±0.2490.622±0.1690.566±0.1420.536±0.1350.283±0.0440.269±0.043150.894±0.2720.755±0.2040.648±0.1490.652±0.1540.460±0.0660.372±0.056

注:黑体数字表示最好的实验结果.

图3给出各模型测试误差的临界差异图(critical difference, CD).CD图不仅能给出不同算法的排名,还能显示算法之间的差异是否显著.图3中算法排名越小表示算法对应的测试误差越小;算法平均排名之间的距离不超过CD值时使用横线连接,表示算法之间差异不显著.图3中算法的平均排名是基于15个数据集和8种噪声水平的模型测试误差.

Fig. 3 CD diagram of each regression model error
图3 各回归模型误差CD图

由图3可见,所提RNF算法在各个模型上都取得了最小测试误差.在kNN模型中,RNF与CDF算法无显著性差异;在其他模型中,RNF比其他算法均有显著性优势.已有过滤算法中,CDF表现最佳,RegENN,DiscENN和MI无明显差异.所有过滤算法均比未过滤数据(NoF)效果更好,即过滤能够提升模型泛化能力,所提RNF算法的提升效果最明显.

表4列出了不同噪声比例下各模型平均测试误差.总体上噪声比例越大,测试误差也越大.所提RNF算法的测试误差最小.同时注意到,当噪声比例较低时,RNF的测试误差明显小于其他算法的误差;当噪声比例较大(NR=0.4)时,RNF的测试误差略微优于CDF算法.总之,在低噪声比例情况下RNF的优势更加明显.

Table 4 Average Test Error of Each Model
表4 各模型平均测试误差

NR模型NoFMIRegDiscCDFRNF0.10.20.30.4kNN1.671.551.521.480.940.86SVR0.780.770.780.760.690.65GPR0.780.770.780.780.730.69RF0.910.850.850.870.730.66kNN2.111.891.861.751.040.96SVR0.800.790.780.790.720.67GPR0.830.840.830.820.770.74RF1.000.930.930.950.770.70kNN2.512.252.071.981.151.09SVR0.880.850.860.850.780.73GPR0.900.910.920.900.830.81RF1.131.021.041.020.840.79kNN2.952.592.332.251.551.52SVR1.041.011.011.020.900.86GPR1.001.031.011.020.950.95RF1.301.231.201.200.960.94

注:黑体数字表示最好的实验结果.

图4显示了各个过滤算法在5个不同数据规模上的过滤运行时间.其中MI算法运行时间最长;RegENN和DiscENN运行时间略短;CDF和RNF的运行时间最短,而且比其他算法快至少一个数量级.在小规模数据上,RNF的效率比CDF略高,两者一般相差不大.

Fig. 4 Runtime of filtering algorithms
图4 过滤算法的运行时间

4.4 年龄标签噪声过滤

人脸年龄估计是一个具有挑战性的监督学习问题,ICCV和CVPR等计算机视觉顶会在竞赛任务中曾公开过人脸图像和年龄标注数据[28-29].每个图像的年龄标签是多个标记者所给年龄估计的均值,这些数据中存在部分标记与人脸图像不匹配的情况.通过RNF噪声过滤可以找到这些标签噪声数据,进而提升模型预测性能.

原始数据来自ICCV 2015和CVPR 2016[28-29],共有18 424张图像和对应年龄标签,其中2个数据子集中存在一些重复图像,每个图像通过左右翻转做了增强.图像特征采用经典的VGG16深度网络提取特征.重复执行5次RNF得到平均年龄标签噪声估计结果,表5列出了平均相对噪声排名前60的部分年龄标签噪声,图像按照相对噪声从大到小排列.表5中部分图像相同(如编号为4,7的图像),但它们所属子集不同,年龄标签也不同.

Table 5 Age Label Noises Recognized by Relative Noise
表5 根据相对噪声识别的年龄标签噪声

图像编号图像图像名称年龄标签平均相对噪声偏差数据来源1005152t.jpg11.310.63偏高CVPR2016valid2004712.jpg27.53.24偏低CVPR2016valid3002827t.jpg50.52.31偏高CVPR2016train4image_432.jpg512.17偏低ICCV2015train5001749t.jpg49.82.12偏高CVPR2016train6image_1194.jpg442.06偏低ICCV2015valid7002376t.jpg55.12.02偏低CVPR2016train8image_2084.jpg71.73偏高ICCV2015valid

续表5

图像编号图像图像名称年龄标签平均相对噪声偏差数据来源9image_519.jpg501.70偏低ICCV2015train10002914.jpg7.11.69偏高CVPR2016train11image_2876.jpg411.67偏低ICCV2015train12004287t.jpg531.66偏低CVPR2016valid13image_2926.jpg411.65偏高ICCV2015train14000500.jpg46.81.59偏低CVPR2016train15000206.jpg38.81.55偏低CVPR2016train16image_1111.jpg441.44偏低ICCV2015train17003928t.jpg19.71.32偏高CVPR2016train18003737t.jpg37.51.31偏低CVPR2016train

表5中既有年龄标签偏高的情况(如编号为1,5,8的图像),也有年龄标签偏低的情况(如编号为4,6,18的图像).在因特尔8核3.6 GHz处理器8 GB内存的单机上进行实验,每轮RNF过滤的时间不超过10 s.可见RNF过滤算法能够快速准确地找到标签噪声.

经RNF过滤后的数据集大约包括86%的原始样本,回归模型在过滤后的数据集上训练后,在另一个wiki年龄数据[30]上进行测试.表6中列出了误差较小的k近邻和随机森林模型的测试结果.年龄测试误差采用平均绝对误差(mean absolute error, MAE)度量.

Table 6 Comparison of Test Errors with Various Filters
表6 各种过滤算法的测试误差比较

对比样本集模型样本量测试误差NoFCDFRNF误差减小比例∕%t检验的p值RNF vs NoFRNF vs CDF全部MAE(NoF)>5MAE(NoF)>10kNN297465.37±4.425.31±4.395.29±4.341.510.0000.182RF297465.57±4.365.55±4.275.54±4.470.510.1310.543kNN136308.86±4.288.81±4.348.80±4.330.680.0000.359RF138139.15±4.379.09±4.359.01±4.251.520.0000.000kNN375313.46±4.4613.31±4.4613.16±4.382.280.0000.000RF413913.84±4.2413.71±4.3213.46±4.412.710.0000.000

注:测试误差的黑体数字为各模型的最佳结果,t检验的黑体数字表示<0.05的p值.

表6中对比了2个模型经未过滤原始数据(NoF)、CDF和RNF过滤数据训练后的测试误差.对比的测试样本集包括wiki全部有效年龄在0~80岁样本集、不过滤情况下测试误差大于5和10的样本集.结果表明,与不过滤NoF原始数据相比,RNF在所有情况下都能够降低模型测试误差,在多数情况下能显著降低测试误差(t检验的p<0.05).与CDF过滤算法相比,RNF在wiki全部样本上与其无显著差异,在大噪声样本MAE>10上有明显优势.

总体上,RNF过滤算法在人脸年龄数据上检测出许多标签噪声数据,能够有效提升数据质量和模型预测性能.

5 结 论

数值型标签噪声问题给回归任务带来严峻挑战.噪声过滤可以有效识别噪声数据,但缺乏模型泛化能力提升的理论保障,实际中还存在过度清洗、自适应差、依赖参数设置等问题.本文根据无噪回归任务中的学习理论给出了面向数值型标签噪声数据的泛化误差界,从而明确了影响模型泛化能力的关键数据因素(数据量和噪声水平).在此基础上提出一种可解释的噪声过滤框架,其目标是以较小的样本去除代价最大程度地降低噪声水平.此框架不仅适用于普通噪声估计方法,也适用于相对噪声估计,只需知道噪声之间的比值关系即可.

针对噪声估计问题,从理论上分析了噪声与覆盖区间关键指标之间的变化趋势,进而构建了相对噪声估计方法.此方法与所提框架结合形成了RNF过滤算法.在标准数据集和真实人脸年龄估计数据上均验证了算法的有效性.所提框架和相对噪声估计方法均有理论支撑,可以确保算法取得良好的过滤效果.实验结果证实了RNF算法在不同噪声分布、不同噪声比例、不同数据集、不同回归模型等复杂情况下具有较好的适应性.

所提过滤框架可用于解决其他过滤算法的超参数优化和适应性问题;所提噪声估计和过滤算法为有序回归或分类任务中的标签噪声问题提供了新思路.

参考文献

[1]Sambasivan N, Kapania S, Highfill H, et al. Everyone wants to do the model work, not the data work: Data cascades in high-stakes AI[C] //Proc of Conf on Human Factors in Computing Systems. New York: ACM, 2021: 1-15

[2]Esteva A, Kuprel B, Novoa R A, et al. Dermatologist-level classification of skin cancer with deep neural networks[J]. Nature, 2017, 542(7639): 115-118

[3]Liao Haibin, Xu Bin. Robust face expression recognition based on gender and age factor analysis[J]. Journal of Computer Research and Development, 2021, 58(3): 528-538 (in Chinese)(廖海斌, 徐斌. 基于性别和年龄因子分析的鲁棒性人脸表情识别[J]. 计算机研究与发展, 2021, 58(3): 528-538)

[4]Kermany D S, Goldbaum M, Cai W, et al. Identifying medical diagnoses and treatable diseases by image-based deep learning[J]. Cell, 2018, 172: 1122-1131

[5]Northcutt C, Jiang L, Chuang I. Confident learning: Estimating uncertainty in dataset labels[J]. Journal of Artificial Intelligence Research, 2021, 70: 1373-1411

[6]Kahneman D, Sibony O, Sunstein C R. Noise: A Flaw in Human Judgment[M]. New York: Little, Brown Spark, 2021

[7]Stoica I, Song D, Popa R A, et al. A Berkeley view of systems challenges for AI[J]. arXiv preprint, arXiv: 1712.05855, 2017

[8]Kordos M, Blachnik M. Instance selection with neural networks for regression problems[C] //Proc of Int Conf on Artificial Neural Networks. Berlin: Springer, 2012: 263-270

[9]Kordos M, Bialka S, Blachnik M. Instance selection in logical rule extraction for regression problems[C] //Proc of Int Conf on Artificial Intelligence and Soft Computing. Berlin: Springer, 2013: 167-175

[10]Jiang Gaoxia, Wang Wenjian, Qian Yuhua, et al. A unified sample selection framework for output noise filtering: An error-bound perspective[J]. Journal of Machine Learning Research, 2021, 22(18): 1-66

[11]Patrini G, Rozza A, Menon A K, et al. Making deep neural networks robust to label noise: A loss correction approach[C] //Proc of IEEE Conf on Computer Vision and Pattern Recognition. Piscataway, NJ: IEEE, 2017: 1944-1952

[12]Sabzevari M, Martinez-Munoz G, Suarez A. Vote-boosting ensembles[J]. Pattern Recognition, 2018, 83: 119-133

[13]Shu Jun, Xie Qi, Yi Lixuan, et al. Meta-Weight-Net: Learning an explicit mapping for sample weighting[C] //Advances in Neural Information Processing Systems. Cambridge, MA: MIT Press, 2019: 1917-1928

[14]Yao Jiangchao, Wang Jiajie, Tsang I W, et al. Deep learning from noisy image labels with quality embedding[J]. IEEE Transactions on Image Processing, 2018, 28(4): 1909-1922

[15]Han Bo, Yao Quanming, Yu Xingrui, et al. Co-teaching: Robust training of deep neural networks with extremely noisy labels[C] //Advances in Neural Information Processing Systems. Cambridge, MA: MIT Press, 2018: 8536-8546

[16]Zhang Zenghui, Jiang Gaoxia, Wang Wenjian. Label noise filtering method based on dynamic probability sampling[J]. Journal of Computer Applications, 2021, 41(12): 3485-3491 (in Chinese)(张增辉, 姜高霞, 王文剑. 基于动态概率抽样的标签噪声过滤方法[J]. 计算机应用, 2021, 41(12): 3485-3491)

[17]Cao Jingjing, Kwong S, Wang Ran. A noise-detection based AdaBoost algorithm for mislabeled data[J]. Pattern Recognition, 2012, 45(12): 4451-4465

[18]Jiang Gaoxia, Fan Ruixuan, Wang Wenjian. Label noise filtering via perception of nearest neighbors[J]. Pattern Recognition

[19]Frenay B, Verleysen M. Classification in the presence of label noise: A survey[J]. IEEE Transactions on Neural Networks and Learning Systems, 2014, 25(5): 845-869

[20]Yuan Weiwei, Guan Donghai, Ma Tinghuai, et al. Classification with class noises through probabilistic sampling[J]. Information Fusion, 2018, 41: 57-67

[21]Guillen A, Herrera L J, Rubio G, et al. New method for instance or prototype selection using mutual information in time series prediction[J]. Neurocomputing, 2010, 73(10): 2030-2038

[22]Arnaiz-Gonzalez A, Diez-Pastor J F, Rodriguez J J, et al. Instance selection for regression by discretization[J]. Expert Systems with Applications, 2016, 54: 340-350

[23]Cherkassky V, Shao Xuhui, Mulier F M, et al. Model complexity control for regression using VC generalization bounds[J]. IEEE Transactions on Neural Networks, 1999, 10(5): 1075-1089

[24]Cherkassky V, Shao Xuhui. Signal estimation and denoising using VC-theory[J]. Neural Networks, 2001, 14(1): 37-52

[25]Cherkassky V, Ma Yunqian. Comparison of model selection for regression[J]. Neural Computation, 2003, 15(7): 1691-1714

[26]Dua D, Gra C. UCI machine learning repository[DB/OL]. [2020-03-28]. http://archive.ics.uci.edu/ml

[27]Chang C C, Lin C J. LIBSVM data: Classification, regression, and multi-label[DB/OL]. [2020-03-28]. http://www.csie.ntu.edu.tw/~cjlin/libsvmtools/datasets/

[28]Escalera S, Fabian J, Pardo P, et al. ChaLearn looking at people 2015: Apparent age and cultural event recognition datasets and results[C] //Proc of IEEE Int Conf on Computer Vision Workshop. Piscataway, NJ: IEEE, 2015: 243-251

[29]Huo Zengwei, Yang Xu, Xing Chao, et al. Deep age distribution learning for apparent age estimation[C] //Proc of IEEE Conf on Computer Vision and Pattern Recognition Workshops. Piscataway, NJ: IEEE, 2016: 722-729

[30]Rothe R, Timofte R, Gool L V. Deep expectation of real and apparent age from a single image without facial landmarks[J]. International Journal of Computer Vision, 2018, 126(2): 144-157

A Numerical Label Noise Filtering Algorithm for Regression Task

Jiang Gaoxia1 and Wang Wenjian1,2

1(School of Computer and Information Technology, Shanxi University, Taiyuan 030006)2(Key Laboratory of Computational Intelligence and Chinese Information Processing (Shanxi University), Ministry of Education, Taiyuan 030006)

Abstract Numerical label noise in regression may misguide the model training and weaken the generalization ability. As a popular technique, noise filtering could reduce the noise level by removing mislabeled samples, but it could rarely ensure a better generalization performance. Some filters care about the noise level so much that many noise-free samples are also removed. Although the existing sample selection framework could balance the number of removals and the noise level, it is too complicated to be understood intuitively and applied in reality. A generalization error bound is proposed for data with numerical label noise according to the learning theory in the noise-free regression task. It clarifies the key data factors, including data size and noise level, that affect the generalization ability. On this basis, an interpretable noise filtering framework is proposed, the goal of which is to minimize the noise level with a low cost of sample removal. Meanwhile, the relationship between noise and key indicators (center and radius) of the covering interval is theoretically analyzed for noise estimation. Then a relative noise estimator is proposed. The relative noise filtering (RNF) algorithm is designed by integrating the proposed framework with the estimator. The effectiveness of RNF is verified on the benchmark datasets and age estimation dataset. Experimental results show that RNF can be adapted to various types of noises and significantly improve the generalization ability of the regression model. On the age estimation dataset, RNF detects some samples with label noises. It effectively improves the data quality and model prediction performance.

Key words numerical label noise; regression; noise filtering; generalization error bound; relative noise

(jianggaoxia@sxu.edu.cn)

中图法分类号 TP181

收稿日期2022-01-09;修回日期:2022-03-28

基金项目国家自然科学基金项目(U21A20513, 62076154, 61906113, U1805263);山西省国际合作重点研发计划项目(201903D421050);山西省高等学校科技创新项目(2020L0007)

This work was supported by the National Natural Science Foundation of China (U21A20513, 62076154, 61906113, U1805263), the Key Research and Development Program of Shanxi Province International Cooperation (201903D421050), and the Scientific and Technological Innovation Programs of Higher Education Institutions in Shanxi (2020L0007).

通信作者王文剑(wjwang@sxu.edu.cn)

作者贡献声明:姜高霞负责算法的理论分析、实验设计和实现,并完成论文初稿的写作与修改;王文剑提出方法的总体思路,指导实验设计与分析,并进行论文的审阅与修改.

Jiang Gaoxia, born in 1987. PhD, associate professor, member of CCF. His main research interests include machine learning and data mining.

姜高霞,1987年生.博士,副教授,CCF会员.主要研究方向为机器学习和数据挖掘.

Wang Wenjian, born in 1968. PhD, professor, PhD supervisor, Distinguished member of CCF. Her main research interests include machine learning and data mining.

王文剑,1968年生.博士,教授,博士生导师,CCF杰出会员.主要研究方向为机器学习和数据挖掘.