基于特征分离的跨域自适应学习模型

李 鑫 李哲民 魏居辉 杨雅婷 王红霞

(国防科技大学文理学院 长沙 410073)

摘 要 跨域训练任务是目前机器学习领域的一个开放性挑战问题.目前最新的研究都在讨论利用真实特征的跨域不变性对未知域数据进行预测,从而实现跨域泛化能力.但事实上,当知道数据来自哪个域时,综合利用真实特征和虚假特征会取得更好的预测效果.针对这一问题,设计了一个同时适用于跨域泛化和跨域适应任务的学习模型CDGA(cross-domain generalization and adaptation model).该模型的核心仍是分离出真实特征,因此新提出了一种更加稳定的训练风险函数,其在跨域泛化问题中不仅具有更高的测试准确率,还克服了现有方法容易过拟合的缺点,可以很好地嵌入到CDGA模型中.另外,通过设计的算法训练后,可使CDGA模型的数据表达部分有效地分离出真实特征和虚假特征,而分类器部分自适应学习选择泛化分类器或特定环境的分类器,从而结合应用了虚假特征,在跨域任务中实现高效预测.最后在构建的彩色手写数字数据集上测试,结果显著优于已有方法.

关键词 机器学习;特征表达;跨域训练;泛化;适应

目前,绝大部分机器学习任务都需要假设数据满足独立同分布这一条件,但在语音语义、工业视觉检测、自动驾驶等诸多机器学习的应用场景中,传感器采集到的数据可能不满足独立同分布的假设,而是来自多个不同的分布(域).例如,手写数字识别中采集的数字可能来自多种不同的书写风格,或是字体大小、颜色等在不同的环境中具有不同的分布.此时,在监督学习过程中,模型会学习到书写风格、颜色等特征,由于这些特征与标签的相关性在不同环境下发生了变化,且不是识别手写数字的主要依据,因此称之为虚假特征.而手写数字的形状特征在各个域中是独立同分布的,也即在不同环境中与标签的相关性保持不变,因此称之为真实特征,其直观解释如图1所示:

Fig. 1 Intuitive interpretation of true and false features

图1 真实特征与虚假特征的直观解释

跨域训练任务的训练集数据按不同的域分类给出,且可根据是否知道测试集所属域信息分为跨域泛化和跨域适应2个问题.当面对一个未知的测试环境时,其虚假特征的分布可能发生重大偏差,因此更倾向于使用真实特征对未知域进行预测,我们称此类问题为跨域泛化问题.另一方面,当预先知道待预测数据来自某个域时,更倾向于综合利用真实特征和虚假特征对目标进行预测,特别地,当收集到的新域数据量较少时,需要结合训练域中的足量数据共同训练出更好的识别模型,我们称此类问题为跨域适应问题.研究跨域任务具有很高的实用价值,例如,根据针对的发音人,语音识别技术可以分为特定人语音识别和非特定人语音识别,前者只需识别1个或几个人的发音,而后者则可以被任何人使用.由于不同发音人的所属地区、年龄、性别等差异可将采集的训练数据分为不同的域,此时训练语音识别模型可视为一个跨域训练任务.对于非特定人的语音识别,就必须提取到语音数据的真实特征,这对应于本项目研究的跨域泛化问题.而对于特定人的语音识别,我们希望能够综合利用真实特征和虚假特征实现更准确的预测.特别地,当采集到的特定人的语音数据较少时,我们希望可以从训练域中事先提取到真实特征和虚假特征,只需通过少量的特定人数据综合利用这些特征,就可以达到更好的识别效果,这对应于本项目研究的跨域适应问题.

本文旨在为上述跨域适应问题提供一种解决思路,其核心就是要分离并综合利用真实特征和虚假特征.由于该问题是最新提出的问题,因此还鲜有文献对其进行详细讨论,通常直接利用经验风险最小化[1]方法进行训练.而分离特征的关键在于稳定地提取真实特征,这又对应于上述跨域泛化问题,因此我们仍需要详细讨论并改进已有的关于跨域泛化问题的方法,进一步嵌入到本文新设计的模型中,可以在跨域任务中取得高效预测.

传统的机器学习方法通过最大程度地减少训练误差来学习复杂的预测规则,但数据常会有选择偏见,一些混杂因素和虚假特征会损害输入数据[2-5],导致模型学到一些虚假的特征[6-8].为了只提取真实特征,目前主要有2种研究思路:一种思路是提升分类模型的鲁棒性[9-12],Engstrom等人[13]指出简单的平移或旋转变换足以蒙骗基于神经网络的视觉模型,需要设计鲁棒的分类模型.后来Jacobsen等人[14]发现神经网络可能对任务无关的输入变化过于敏感,而通过对抗训练提升模型鲁棒性的方法会限制与任务相关的输入.Arpit等人[15-16]从熵的角度出发通过剔除低相关性特征,增强了预测模型的鲁棒性.尽管文献[9-16]的研究较好地增强了模型的鲁棒性,但在部分训练环境中,虚假特征的相关性甚至高于真实特征的相关性,此时这些方法仍会失效.而另一种研究思路发展于Arjovsky等人[17]提出的不变风险最小化,该文认为不稳定特征与标签之间具有虚假相关性,且这种相关性的大小会因环境的不同而发生变化,而稳定特征Φ(x)具有不变性,也即基于稳定特征对标签Y的预测概率P(Y|Φ(x))在不同环境下都相同,且认为这种不变性特征与标签之间具有因果关系[18-20].基于这一思想,提出了用于跨域泛化任务的不变风险最小化(invariant risk minimization, IRM)模型,通过增加了跨域不变性的惩罚项,使得模型能从不同训练域中学习到稳定特征.基于这个结果,Ahuja等人[21]将IRM问题转化为博弈论问题,给出了一种求解新思路.Krueger等人[22]提出了MM-REx和V-REx这2种训练风险函数,使得跨域泛化的准确率得到进一步提升,从而更好地提取稳定特征.

尽管基于IRM的4种方法已经在分离真实特征方面取得了不错的效果,但仍存在一些不足:

1) 各种方法的惩罚项在形式上差异明显,如何解释和评价这些不变惩罚项,以及如何设计合适的不变惩罚项还需要进一步讨论;

2) 这些方法无法预先选择最佳的训练轮数,训练易过拟合.实验表明随着训练轮数的增加,训练准确率会继续增加(上界由训练环境中的标签与虚假特征的相关性大小决定),但测试准确率会先增加后减小,这表明随着训练轮数的增加模型绕过了不变性惩罚学习到虚假特征.

针对上述问题,本文在充分讨论跨域泛化问题的基础上,设计了一种更加稳定风险损失函数,在此基础上,构建了一个基于特征分离的跨域自适应学习模型,很好地解决了跨域适应问题.具体有3方面贡献:

1) 提取真实特征的关键在于设计合适的训练风险函数,我们发现总体预测风险中引入不变性惩罚应与虚假特征在不同环境下和标签的相关性差异有本质联系.基于这一思路,比较了各种分布差异惩罚的优劣,并根据准确率差异设计了一种新的不变性惩罚.

2) 基于上述不变性惩罚提出了一种新的风险损失函数ADR(accuracy difference risk),该风险函数不仅可以更好地提取真实特征,而且有效地避免了IRM等方法容易过拟合的缺点,因此,该风险函数便于嵌入应用到后续模型中,更具实用价值.

3) 基于ADR风险函数,我们构建一个可以综合利用稳定特征和不稳定特征的自适应学习模型CDGA(cross-domain generalization and adaptation model).该模型分为数据特征表达和分类器2部分,通过设计的训练方法训练后,该模型不仅可以提升了解决跨域泛化问题的效果,还充分利用训练域的足量数据和新域的少量数据很好地解决跨域适应问题.

1 背 景

在很多实际分类任务中,收集到的数据会来自多个域,定义为所有可能的数据环境的集合,其中εtr={e1,e2,…,em}为训练环境,为未知训练环境.已知来自训练环境中的足量采样数据De通过这些数据,学习训练出符合任务要求的预测模型fθ=ωΦ,其中θ为模型参数,Φ:XH为数据特征表达,ω:HY为分类器,ωΦ可理解为复合函数,也即先用Φ作用于原始数据,再用ω作用,最终得到样本的预测值.

1.1 跨域泛化任务

为了稳定地分离出真实特征,我们先对跨域泛化任务中的已有工作进行充分调研.在跨域泛化任务中,由于在不同域中虚假特征的分布会发生变化,而真实特征的分布保持不变,因此更倾向于通过真实特征对来自未知域的待预测样本进行预测.最新的研究成果大都是通过选择合适的风险函数对来自未知域的原始特征数据{},egεun进行标签预测.目前的方法主要有4种:

1) 经验风险最小化[1](ERM).设环境e中的误差风险为

(x,y)∈De(fθ(x)-y),

(1)

其中,表示期望,ERM方法直接使用所有的训练数据进行学习,使用所有的特征来最小化训练误差:

(2)

其中,为损失函数,该方法在独立同分布的训练数据中具有很好的训练效果.但当训练环境与测试环境明显不同时,则学到的模型可能失效.

2) 鲁棒性优化[10](ROB).在鲁棒性优化中,需要最小化风险函数:

(3)

其中,re为环境基准,当设定re=0时,也就是最小化误差最大的环境误差.选择这个基准是为了去除环境噪声对学习的影响.

3) 不变风险最小化[17](IRM).Arjovsky等人[17]最新提出了一种不变风险最小化的方法,将预测模型f分成2个部分f=ωΦ,其中Φ:XH为数据特征表达,ω:HY为分类器.并提出当分类器ω可以使得所有环境中的预测风险一致达到最小时,则认为预测模型用的是不变特征进行预测的,而这种不变的特征正是在跨域泛化问题中需要提取的真实特征.该方法的数学表示为

(4)

这是一个很有挑战的2级多目标优化问题,因此作者进一步将其简化为一种可操作的版本IRMv1,转化成一个2级单目标优化问题,即要最小化风险函数:

(5)

其中表示ω求梯度,并取ω=1.0,也即将分类器简化为恒等变换,且将不同环境下的预测风险对分类器参数的梯度作为惩罚项引入到总的风险函数中∈[0,∞)为平衡预测风险的超参数,由此得到了不变预测1.Φ,从而达到学习稳定特征的目的.

4) 最小化最大风险(MM-REx)和风险方差最小化[22](V-REx).Krueger等人[22]基于各环境中的预测风险,分别提出了最小化最大风险和风险方差最小化2种风险函数.MM-REx方法考虑将最大环境风险与其他环境风险加上权重,以增加不同环境风险的相似性:

(6)

其中,m为训练环境数量,超参数β为平衡参数.而V-REx方法可以进一步增强不同环境风险的相似性,以它们之间的方差作为惩罚项加入到风险函数中:

(7)

该方法通过控制不同环境中输出风险的差异,使得模型学习训练数据中的真实特征.

这4种方法在风险函数中引入了不同的不变性惩罚项,从而学习了原始数据的真实特征.但为了更好地嵌入到后续提出的CDGA模型中,我们需要对式(5)~(7)中的风险函数进一步优化,在跨域泛化任务中的测试准确率和训练稳定性这2个方面达到更好的效果.

1.2 跨域适应任务

在跨域适应任务中,需要对已知域ea的原始数据{}作出标签预测,因此若环境ea的数据中虚假特征的相关性高于真实特征的相关性,那么更倾向于使用虚假特征或真实特征与虚假特征相结合的方式进行预测.目前关于如何分离及重构稳定特征和不稳定特征的研究较少,一种最简单的方式就是直接将环境ea中的数据投入模型并用经验风险最小的方法进行训练.这显然是有效的,但在实际问题中,面对一个已知新域,往往没有收集到足够多的数据,因此如何利用已有训练环境数据De(eεtr)来预测含有少量样本的新域数据还有待进一步研究.

针对此问题,我们在改进分离真实特征中训练风险函数的基础上构建了一个解决跨域任务的CDGA模型,可以很好地分离出并综合利用真实特征和虚假特征,在跨域泛化和跨域适应问题中均取得了很好的效果.

2 基于ADR风险分离真实特征

本文的最终目标是构建一个能够综合利用真实特征和虚假特征的学习训练模型,在跨域泛化和跨域适应问题上都具有很好的效果.而该模型的核心在于如何更好地学习不同环境数据中的真实特征,因此本节先详细讨论了学习跨域稳定特征的关键是在总体预测风险中引入不变性惩罚,并提出该惩罚项与模型预测分布的差异有本质联系.进一步,通过比较了各种分布差异惩罚的优劣,基于准确率差异设计了一个更好的不变性惩罚,从而提出ADR风险函数,在分离真实特征中取得很好的效果,提升了跨域泛化能力.

2.1 多种分布差异作为不变性惩罚项

跨域泛化问题的核心是提取不同环境下训练数据的真实特征.如果模型提取了真实特征,对于同一标签的预测结果,预测为各类别的概率近似相同,也即P(Y|Φ(x))在不同环境中都相同,因此其在不同环境下对相同标签数据的预测分布应近似相同,大多数文献都基于此将构建的不变性惩罚项加入到总体训练风险函数之中,从而提取真实特征.Arjovsky等人[17]提出了不变风险最小化方法,预测模型分为数据表达Φ和分类器ω这2个部分,训练目标是优化ωΦ的参数使得所有环境中的预测风险达到一致最小,其风险函数由式(5)给出.Krueger等人[20]提出了最大风险最小化和风险方差最小化2种方法,训练目标是优化不同环境中预测风险的差异达到最小,风险函数分别由式(6)(7)给出.以上3种方法是目前提取真实特征效果较好的方法,从本质上而言,各种风险函数都专注于刻画不同环境下模型预测分布差异.如果学习到真实特征进行预测,那么其在不同环境下的预测结果应当具有类似的分布,与之相反,当学习到虚假特征时,模型在不同环境下预测结果的分布会产生较大差异.为此基于4.1节所构造的Colored MNIST数据集考察了不同的风险函数下模型训练输出的分布,其平均预测准确率见表1,预测分布情况如图2所示.

Table 1 Average Accuracy of the Model Under Different Penalty Items

表1 不同惩罚项下的模型的平均准确率 %

不变性惩罚项训练准确率测试准确率TRUE72.872.2ERM84.810.3MEAN82.631.4VAR82.527.1MEAN-VAR77.140.8IRMv169.166.3V-REx70.068.6KL50.350.1

首先,训练之前人为去除训练集的颜色特征(4.1节中构造的Colored MNIST数据集仅有颜色这一个虚假特征),此时模型只能学习到真实特征,因此期望的训练结果是训练和测试准确率均超过了72%,其在不同环境中的预测分布见图2(a1)(a2);随后,引入颜色特征,并考察多种方法的预测效果.

1) 不采用任何不变性惩罚时的预测分布见图2(b1)(b2)(即ERM方法),此时模型学习了虚假特征进行预测,训练准确率达到84.8%,但由于虚假特征在测试集中与标签的相关性发生较大变化,测试准确率仅有10.3%,该方法失效.

2) 采用分布均值差异作为不变性惩罚项的预测分布见图2(c1)(c2)(即MEAN方法),该方法惩罚了不同环境下预测分布均值的差异.此时,训练准确率为82.6%,测试准确率上升为31.4%,可见该惩罚一定程度上增强了对真实特征的学习,但并不严格,仍能学习到很多虚假特征.

3) 采用分布方差差异作为不变性惩罚项的预测分布见图2(d1)(d2)(即VAR方法),采用分布均值差异和分布方差差异相结合的方式见图2(e1)(e2),此时预测分布差异得到更细致的刻画,测试准确率上升到40.8%,可见进一步严格预测分布惩罚可以学习到更多稳定特征.

4) 采用IRMv1作为不变性惩罚项的预测分布见图2(f1)(f2),该方法考虑到在不同环境下的预测结果分布应当是平移不变的,更加严格地限制了分布的差异,其测试准确率提升到66.3%.

5) 采用V-REx作为不变性惩罚项的预测分布见图2(g1)(g2),该方法惩罚了不同训练环境下预测误差的方差,适当放宽了IRMv1对分布差异的限制,训练准确率可进一步提升到68.6%.

6) 采用更严格的KL散度作为不变性惩罚项的预测分布见图2(h1)(h2),KL散度从信息的角度表征了分布之间的差异大小.此时,模型更倾向于在不同环境下作出更一致的预测结果,而随机的预测结果比任何一种预测都更能缩减其中的差异,从而导致模型无法学习到任何有用的特征.

Fig. 2 Distribution map of label prediction under different methods

图2 不同方法下的标签预测分布图

通过分析不同方法下的不变性惩罚项,发现其本质在于刻画模型在不同环境下的预测分布差异.但需要注意是,若严格限制不同环境下的预测结果的分布,会导致分类器更倾向于一个随机的预测结果,而不对任何特征进行学习,为此,需要在模型的准确性和模型的泛化能力之间作出取舍,折中设计一种更好的不变性惩罚.

2.2 ADR训练风险函数

在设计风险函数时,要尽可能地惩罚虚假特征,且保留真实特征.为实现这2个目的,我们从2个方面考虑:1)要在学习真实特征的基础上,尽可能多地惩罚虚假特征的影响.因此,我们在设计风险惩罚函数时,要能充分反映虚假特征的影响,也即,当模型学习到虚假特征进行预测时,风险函数会明显增大.2)要在惩罚虚假特征的基础上,尽可能多地保留真实特征.为此,我们需要放宽对预测分布的一致性约束.主要有2个原因:1)不同训练环境中的样本的随机性和数据量的有限性,造成了真实特征在不同环境下与标签的相关性产生差异;2)在一些实际系统中,真正决定样本分类的特征与虚假特征不具有明显的界限(如手写数字的形状特征和书写风格这一虚假特征),因此,虚假特征在不同环境下与标签的相关性差异会一定程度上影响了真实特征在不同环境下与标签的相关性.所以,我们在惩罚虚假特征的同时,也要保护形状特征不被明显“过滤掉”.基于此分析,我们基于预测准确率差异设计了一种非参数估计的惩罚项,有效权衡了“惩罚虚假特征”和“保留真实特征”两个目标,在此基础上设计了ADR训练风险函数.

ADR风险函数的设计基于思想为:当模型进行学习时,会同时学习到真实特征和虚假特征,从而在不同环境下作出不一致的预测,而在这些不一致的预测中,那些预测错误的结果是由虚假特征主导的,这是因为不同环境下虚假特征与目标的相关性会发生改变,因此我们更加关注不同环境下分类错误部分的差异并施以惩罚.因此,构建了ADR训练风险函数:

(8)

其中,labels表示分类标签集合,Nei(f)表示环境e下的i标签被分类错误的数目,μ为调节超参数.

ADR风险函数有2点重要改变:1)用分类错误数目这种离散型的方法来度量分类错误程度.采取离散型惩罚时,相较于式(7)可以更好地防止模型的过拟合训练,这是因为一旦模型要学习虚假特征,式(8)将以一个小突变的方式产生惩罚,从而在一定程度上有效地避免了先前方法随着训练轮数的增加逐渐绕过不变性惩罚学习虚假特征的情况.2)考虑特定标签下不同环境中预测分布的差异,而其他方法则考虑的为所有标签下不同环境预测分布的差异.考虑一个二分类问题,识别数字0和1,环境有2种.假如模型在环境1中对数字0分类错了100个,对数字1分类错了0个,在环境2中对数字0分类错了0个,对数字1分类错了100个,且由式(7)计算出的相等.那么此时通过式(7)计算得到的第1项中的方差为0,而式(8)的第1项却能够捕获到这种情况下模型在不同环境的识别能力的差异.因此,ADR风险函数可以更多地反映虚假特征造成的差异,从而达到更好的预测效果.

3 构建跨域任务中的CDGA模型

我们在第2节详细讨论了如何分离提取出训练数据中的真实特征,该特征可以对跨域泛化问题作出稳定的预测.事实上,在学习过程中引入了预测分布差异惩罚项后,数据表达Φ将自动地过滤掉虚假特征,而仅保留真实特征.但当已知数据来自哪个域时,应当更倾向于综合利用真实特征和虚假特征共同预测,基于此,本文建立了可以同时处理跨域泛化和跨域适应问题CDGA模型.

3.1 原理描述

对于跨域泛化任务,要对来自未知域的原始特征数据{},egεun进行标签预测.由于虚假特征在不同环境中与真实标签的相关性会发生变化,因此模型需要学习真实特征表达Φs,为此在训练过程中引入不变性惩罚:

d(D(Φs(xe1)),D(Φs(xe2)),…,D(Φs(xem))),

(9)

其中,D(·)为数据的分布,在后文的CDGA模型中采用式(8)所示的ADR风险函数.从而,通过训练可以使得Φs在保证训练准确率的基础上满足:

D(Φs(xep))≈D(Φs(xeq)),∀ep,eqεtr,

(10)

此时认为数据表达Φs主要包含真实特征,可对未知域作出稳定预测{ωΦs()}.

对于跨域适应任务,需要对已知域ea的原始数据{}作出标签预测.此时分2种情况考虑,当ea=ejεtr时,则直接利用ej环境下的采样数据Dej进行经验风险最小化预测,便可得到很好的预测效果;当时,由于新域只有少量的观测数据若仅用少量数据来学习,则模型不能够得到充分训练.因此,我们希望能够从训练集Deeεtr中同时学到稳定特征表达Φs和不稳定特征表达ΦU,然后通过新域数据来学习分类器ωa,该分类器可以理解为对特征(Φs,ΦU)在新域中的自适应学习,然后作出更为准确的预测{ωa∘(Φs,ΦU)()},相较于仅使用稳定特征和仅学习少量新域数据等方法具有更好的预测.

基于上述跨域泛化和适应2个任务的需求,我们设计了一种可分离特征的跨域自适应学习模型.同样考虑将预测模型分为数据特征表达和分类器这2个部分,f=ωΦ,其中ω为自适应学习的分类器,包括与环境对应的分类器ωe、真实特征分类器ωs以及新域分类器ωa,对应的参数为λeλs,λa,其中另一部分为数据表达Φ,对应参数为φ,通过训练数据的学习可表达真实特征和虚假特征.超参数NφNω分别是训练参数φω的训练轮数,μυ为风险权重,在第ΓωΓΦ步之前参数μ取很小的值(为了避免模型初始时因不变惩罚过大而造成停止优化的现象).训练过程如算法1所示:

算法1. CDGA模型参数训练算法.

输入:训练数据De(eεtr)、少量新域数据Dea;

输出:预测模型ωsΦωaΦ以及ωeΦ,eεtr.

① for step=1 to Nφdo

② for k=1 to Nωdo

③ for j=1 to m do

logitj=ωejΦ(xej);

lossj=(logitj-yej);

lossj反向传播更新参数λej;

⑦ end for

logita=ωaΦ(xea);

lossa=(logita-yea) ;

lossa反向传播更新参数λea;

logits=ωsΦ(xεtr);

losss=(logits-yεtr)+μ(D(Φs(xe1)),

D(Φs(xe2)),…,D(Φs(xem)));

/*step小于阈值Γω,μ取较小值*/

losss反向传播更新参数λs;

end for

μ(D(Φs(xe1)),D(Φs(xe2)),…,

D(Φs(xem)));

/*step小于阈值ΓΦ,μ取较小值*/

lossφ反向传播更新参数φ;

end for

return Φ,ωs,ωa,ωe,eεtr.

算法1便是CDGA模型的学习过程,通过交替学习分类器和数据表达,最终学习出可分离特征的数据表达Φ,以及与环境适应的分类器ω.当只有2个训练环境时,训练过程示意图如图3所示:

Fig. 3 Schematic diagram of CDGA model training in two training environments

图3 2个训练环境的CDGA模型训练示意图

如图4所示,对于跨域泛化任务,我们可以用返回的模型ωsΦ进行预测.对于跨域适应任务,若待预测数据来自训练环境ej,则用模型ωejΦ进行预测,若待预测数据来自新环境ea,则用模型ωaΦ进行预测.因此,该模型可以同时实现跨域泛化和跨域适应2项任务.

Fig. 4 Model usage in three different scenarios

图4 3种不同场景下的模型使用情况

3.2 CDGA模型的可行性分析

针对跨域学习任务,我们建立了一个可分离特征的CDGA模型,其关键在于通过优化参数φ,使得数据表达Φ既可以表达出真实特征也可以表达出虚假特征.为了达到这一目的,优化参数φ的风险函数共包含3项:第1项为训练环境适应损失其中lossj=(ωejΦ(xej)-yej),由于该问题通常部分环境中虚假特征与标签的相关性会高于真实特征,它保证了Φ可以表达出虚假特征.第2项不变预测损失(ωsΦ(xεtr)-yεtr)与第3项不变性惩罚d(D(Φs(xe1)),D(Φs(xe2)),…,D(Φs(xem)))在第2节中做了详细的讨论,这里我们用更加稳定的ADR风险函数代替,这是因为CDGA模型要对数据特征表达Φ和分类器ω进行多次交替训练,很容易因为训练次数过多而造成过拟合现象,因此我们采用本文提出的准确率更高且更稳定的ADR风险函数.该风险函数保证了预测模型ωsΦ使用真实特征进行预测,从而确保了Φ可以表达出真实特征ν为超参数,用于调节各损失风险的比例.

对于分类器ω,每次迭代我们都会学习出适应各训练环境的最优分类器ωe,eεtr和不变预测分类器ωs,它们可以对Φ表达的真实特征和虚假特征进行加权结合,从而使得预测模型更加适应特定环境进行预测.对于未知域的待预测数据,我们可直接采用不变预测分类器ωsΦ进行预测,也即使用真实特征进行预测,可以保证预测效果.

综上,通过交替训练各环境下的分类器和数据表达,我们可以分离真实特征和虚假特征,从而在跨域任务中实现高效预测.

4 实 验

4.1 实验设定

实验选取由MNIST数据集生成的Colored MNIST数据集,该数据集由Arjovsky等人[17]提出用于检测算法提取真实特征的能力.其构建过程如图5所示,原始MNIST数据集由分辨率为28×28的0~9手写数字图像组成,每张图片x有对应的标签而Colored MNIST数据集将问题简化为二分类问题.第1步,定义新的标签:

(11)

第2步,以0.25的概率随机翻转得到最终的标签y.第3步,对于环境e中的标签y,我们以概率pe的随机翻转得到颜色标签z,然后将图像x降采样后,当颜色标签z=1时涂为红色,z=0时涂为绿色(见图5最右侧一列),得到染色后的图像大小为14×14×2.本文所有的算法均在Pytorch框架下实现,所有仿真实验都在Intel® Xeon® Silver 4110CPU@2.10 GHz, Nvidia GeForce GRX 2080Ti 的工作站上进行.

Fig. 5 Flow chart of constructing Colored MNIST data set

图5 构建Colored MNIST数据集流程图

4.2 基于ADR风险函数提取真实特征

当模型学习到样本中的真实特征时,其能够在不同域中均具有不错的表现.为了验证本文提出的ADR模型能够较好地提取样本的稳定特征,我们取pe=0.1,0.2分别生成25 000张图像作为2个训练环境,以pe=0.9生成10 000张图像作为测试环境.当采用ADR风险函数训练模型,模型的预测结果数据分布情况如图6所示.可以发现相比较于其他方法,其在错误预测上具备更稳定的分布特征,由此削弱了模型对虚假特征的学习程度,可以更好地学习真实特征进行模型预测.

然后,我们将采用ADR风险函数的实验结果与经验风险最小化模型(ERM)、不变风险最小化模型(IRMv1)、鲁棒性优化模型(Rob)和风险方差模型(V-REx)进行了对比.每种方法进行20次训练实验,每次训练2 000轮.选取模型在测试集上的准确率作为模型优劣的度量指标.实验结果如图7和图8所示,图中的各种线表示不同方法下20次实验的准确率均值,阴影区域表示20次实验结果的包络区域.图7展示了不同方法在训练环境中的准确率随训练轮数的变化情况.我们可以看出ERM方法和Rob方法的训练准确率超过90%,可见模型充分学习了颜色特征.而其他3种方法的训练准确率均在70%左右,有效地抑制了对颜色特征的学习.而图8展示了模型在pe=0.9的测试环境中的准确率随训练轮数的变化情况,我们提出的方法的平均测试准确率为69.0%±1.8%,具有最高的测试准确率.另外,随着训练轮数的增加,其他方法的测试准确率逐渐开始下降,而我们的方法保持不变,这就避免了因迭代轮数选择过大而影响训练效果,有效地克服了其他方法容易过拟合学习到颜色特征的缺点,解决了测试准确率因迭代轮数增加而明显降低的问题.因此,ADR方法可以稳定地提取真实特征,可很好地嵌入到我们的CDGA模型中.

Fig. 6 The label prediction distribution map using the ADR risk function

图6 采用ADR风险函数的标签预测分布图

Fig. 7 Comparison of training precision under different methods

图7 不同方法下训练准确率比较图

4.3 基于CDGA模型解决跨域训练问题

在实际运用中,仅使用真实特征去识别往往会导致识别准确率不高,因为一些虚假特征往往对识别准确率的提升有一定的积极作用.而这些虚假特征的分布会随着域的变化而改变,在已知某个样本来自某个域的情况下对该样本进行识别被称之为跨域适应问题.例如,在实际问题中,有时可能需要去识别一些具备不同背景颜色的数字,而这些来自新域的数字很少被标注,经常直接使用这些样本训练模型后,模型在测试集表现不佳.而直接使用MNIST作为训练集进行训练会面临跨域适应的难题.已有的工作提出了能够学习真实特征的模型,但是没有对如何结合适用少量新域数据中的虚假特征进行讨论.本文对于跨域适应问题的讨论,基于假设域的信息通过来自该域的其他少量标注样本给出,这一假设和实际背景契合,具有合理性.

Fig. 8 Comparison of test precision under different methods

图8 不同方法下测试准确率比较图

实验以pe=0.1,0.2分别生成25 000张图像作为2个训练环境,分别以pe=0.4,0.5,0.6,0.7,0.8,0.9生成10 000张图像作为测试环境.此外,以pe=0.4,0.5,0.6,0.7,0.8,0.9生成1 000张来自新域的标注图像,且这些图像的原图像与测试集不相同.实验选取了3种模型:第1种为本文提出的跨域泛化模型,该模型使用生成的50 000张训练集进行训练;第2种为经验风险最小化模型(ERM),该模型分别在pe=0.4,0.5,0.6,0.7,0.8,0.9的1 000张新域图像上进行训练;第3种为本文提出的CDGA模型,使用50 000张训练集和1 000张新域图像进行训练.

实验结果如表2所示,我们不难看出本项目提出的CDGA模型对新域的预测结果要显著优于跨域泛化和ERM方法.当虚假特征与标签的相关性较弱时(pe的值接近0.5),CDGA模型侧重利用真实特征作出预测,且少量利用虚假特征使预测结果比泛化模型更好;当虚假特征与标签的相关性较强时(pe的值接近0或1),CDGA模型侧重利用虚假特征作出预测,且少量利用真实特征使预测结果比ERM方法还要好,从而验证了CDGA模型在跨域适应问题中的优势.

Table 2 Comparison of the Effects of Different Models with a Small Amount of New Domain Data

表2 利用少量新域数据情况下不同模型效果比较

注:黑体值表示新域和待测域的pe相同时的预测结果.

模型pe=0.4pe=0.5pe=0.6pe=0.7pe=0.8pe=0.9泛化模型0.682±0.0160.687±0.0170.684±0.0170.680±0.0170.681±0.0190.678±0.024ERM(pe=0.4)0.638±0.0080.619±0.0060.597±0.0080.577±0.0070.560±0.0130.540±0.014ERM(pe=0.5)0.625±0.0110.619±0.0080.621±0.0080.618±0.0130.617±0.0130.615±0.012ERM(pe=0.6)0.595±0.0100.611±0.0060.631±0.0060.650±0.0090.669±0.0090.687±0.014ERM(pe=0.7)0.555±0.0140.597±0.0130.636±0.0080.677±0.0080.718±0.0120.756±0.017ERM(pe=0.8)0.507±0.0100.568±0.0090.626±0.0080.686±0.0080.745±0.0090.804±0.010ERM(pe=0.9)0.443±0.0100.526±0.0060.609±0.0040.689±0.0050.771±0.0050.856±0.009适应模型(pe=0.4)0.698±0.0120.676±0.0190.650±0.0290.625±0.0370.602±0.0450.578±0.053适应模型(pe=0.5)0.692±0.0080.694±0.0100.694±0.0130.693±0.0180.693±0.0240.691±0.032适应模型(pe=0.6)0.643±0.0230.669±0.0160.694±0.0110.720±0.0080.742±0.0080.769±0.014适应模型(pe=0.7)0.550±0.0290.608±0.0220.662±0.0110.718±0.0040.773±0.0070.828±0.014适应模型(pe=0.8)0.467±0.0240.547±0.0170.627±0.0110.707±0.0050.789±0.0130.869±0.010适应模型(pe=0.9)0.408±0.0190.504±0.0140.605±0.0110.700±0.0040.798±0.0040.897±0.009

综上,我们的模型在跨域适应问题中可以综合利用真实特征和虚假特征作出很好的预测.而且在训练过程中也学习了跨域泛化预测模型ωsΦ,因此我们初步验证了CDGA模型可以同时在跨域泛化和跨域适应任务中作出高效预测.

5 总结与展望

本文基于机器学习领域中的跨域任务的最新研究,构建了一个能够实现特征分离的自适应学习模型——CDGA模型,该模型在跨域泛化和跨域适应问题上都具有不错的表现.

一方面,对于跨域泛化问题,需要使用真实特征对未知新域数据进行预测.由于真实特征在不同环境下的分布保持不变,因此考虑在训练风险中引入不变性惩罚来剔除虚假特征.通过分析发现,该惩罚项的本质是表征模型在不同环境下输出分布的差异.基于这一思路,比较了现有方法的优劣,并提出了一种新的训练风险函数ADR,可以更好地学习到数据中的真实特征.该方法不仅具有更高的测试准确率,还克服现有方法容易过拟合的缺点,解决了测试准确率因迭代轮数增加而明显降低的问题.

另一方面,基于稳定的跨域泛化学习方法,本文构建的CDGA模型可以在跨域适应任务取得很好的效果.该模型训练出的数据表达Φ同时可以表达真实特征和虚假特征.若已知待测数据来自某个训练域,则只需选择相应域的分类器即可;若已知待测数据来自某个新域ea,则只需收集该域的少量数据Dea训练相应的分类器ωa即可;若待测数据来自未知域,则选择稳定预测分类器ωsΦ进行预测.因此,不论是跨域泛化任务还是跨域适应任务,该模型均可选择相应的分类器ω作出高效预测.

本文提出的模型可以应用于跨域适应和跨域泛化2项任务,相较于原有方法取得一定的进展.但对于跨域学习任务,仍然有不少问题值得进一步研究.例如,如何寻找新的不变性惩罚项以更好地提取真实特征并过滤虚假特征;如何拓展该模型在多种虚假特征混杂的训练数据中的应用;如何将不变性惩罚方法与鲁棒性增强的方法相结合以解决跨域训练任务等.

作者贡献声明:李鑫负责提出研究选题、提出模型、论文撰写;李哲民负责编写代码和实施实验过程;魏居辉负责论文作图和撰写论文;杨雅婷负责调研整理文献并设计论文框架;王红霞负责修订和完善论文.

参考文献

[1]Vapnik V. Principles of risk minimization for learning theory[C] //Proc of Advances in Neural Information Processing Systems. San Francisco: Margan Kaufmann, 1991: 831-838

[2]Sturm B L. A simple method to determine if a music information retrieval system is a “Horse”[J]. IEEE Transactions on Multimedia, 2014, 16(6): 1636-1644

[3]Torralba A, Efros A. Unbiased look at dataset bias[C] //Proc of the IEEE Computer Society Conf on Computer Vision and Pattern Recognition. Los Alamitos, CA: IEEE Computer Society, 2011: 1521-1528

[4]Jabri A, Joulin A, Der Maaten L V, et al. Revisiting visual question answering baselines[C] //Proc of European Conf on Computer Vision. Berlin: Springer, 2016: 727-739

[5]Liu Huan, Zheng Qinghua, Luo Minnan, et al. Cross-domain adversarial learning for zero-shot classification[J]. Journal of Computer Research and Development, 2019, 56(12): 2521-2535 (in Chinese)(刘欢, 郑庆华, 罗敏楠, 等. 基于跨域对抗学习的零样本分类[J]. 计算机研究与发展, 2019, 56(12): 2521-2535)

[6]Jo J, Bengio Y. Measuring the tendency of CNNs to learn surface statistical regularities[J]. arXiv preprint, arXiv:1711.11561, 2017

[7]Geirhos R, Rubisch P, Michaelis C, et al. ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustness[C/OL] //Proc of Int Conf on Learning Representations, 2019 [2020-07-13]. https://arxiv.org/abs/1811.12231

[8]Ilyas A, Santurkar S, Tsipras D, et al. Adversarial examples are not bugs, they are features[C] //Proc of Conf on Neural Information Processing Systems. Cambridge, MA: MIT Press, 2019: 125-136

[9]Bagnell J. Robust supervised learning[C] //Proc of the 20th AAAI Conf on Artificial Intelligence. Menlo Park, CA: AAAI, 2005: 714-719

[10]Ben-Tal A, Nemirovski A. Robust Optimization[M]. Princeton: Princeton University Press, 2009

[11]Duchi J C, Glynn P W, Namkoong H, et al. Statistics of robust optimization: A generalized empirical likelihood approach[J]. arXiv preprint, arXiv: 1610.03425, 2016

[12]Sinha A, Namkoong H, Duchi J. Certifying some distributional robustness with principled adversarial training[C/OL] //Proc of Int Conf on Learning Representations. 2018 [2020-07-23]. https://arxiv.org/abs/1710.10571

[13]Engstrom L, Tran B, Tsipras D, et al. Exploring the landscape of spatial robustness[J]. arXiv preprint, arXiv: 1712.02779, 2017

[14]Jacobsen J, Behrmann J, Zemel R S, et al. Excessive invariance causes adversarial vulnerability[C/OL] //Proc of Int Conf on Learning Representations. 2019 [2020-06-18]. https://arxiv.org/abs/1811.00401

[15]Arpit D, Xiong Caiming, Socher R, et al. Predicting with high correlation reatures[J]. arXiv preprint, arXiv: 1910.00164, 2019

[16]Arpit D, Xiong Caiming, Socher R, et al. Entropy penalty: Towards generalization beyond the IID assumption[C/OL] //Proc of Int Conf on Learning Representations. 2020 [2020-07-28]. https://openreview.net/pdf?id=Byes0TNFDS

[17]Arjovsky M, Bottou L, Gulrajani I, et al. Invariant risk minimization[J]. arXiv preprint, arXiv: 1907.02893, 2020

[18]Lopezpaz D. From dependence to causation[J]. arXiv preprint, arXiv: 1607.03300, 2016

[19]Lake B M, Ullman T, Tenenbaum J B, et al. Building machines that learn and think like people[J]. Behavioral and Brain Sciences, 2017, 40: No.e253

[20]Marcus G F. Deep learning: A critical appraisal[J]. arXiv preprint, arXiv: 1801.00631, 2018

[21]Ahuja K, Shanmugam K, Varshney K R, et al. Invariant risk minimization games[J]. arXiv preprint, arXiv: 2002.04692, 2020

[22]Krueger D, Caballero E, Jacobsen J, et al. Out-of-distribution generalization via risk extrapolation(REx)[J]. arXiv preprint, arXiv: 2003.00688, 2020

Cross-Domain Adaptive Learning Model Based on Feature Separation

Li Xin, Li Zhemin, Wei Juhui, Yang Yating, and Wang Hongxia

(College of Liberal Arts and Sciences, National University of Defense Technology, Changsha 410073)

Abstract Cross-domain training tasks are currently an open challenge in the field of machine learning. At present, the latest researches are discussing the use of the cross-domain invariance of real features to predict unknown domain data, so as to achieve cross-domain generalization capabilities. But in fact, when it is known that the data comes from a certain domain, the comprehensive use of real features and false features will achieve better prediction results. This paper focuses on this issue and designs a learning model that is suitable for both cross-domain generalization and adaptation tasks(CDGA). The core of the model is still to separate the real features, so this paper proposes a new more stable training risk function, which not only has a higher test accuracy in the cross-domain generalization problem, but also overcomes the shortcomings of traditional methods that are easy to overfit, so it can be well embedded in the CDGA model. In addition, through the designed training method, the data expression part of the CDGA model can effectively separate the real features and false features, and the classifier part adaptively learns to select the generalized classifier or the classifier of the specific environment, thereby combining the application of false features to achieve efficient prediction in cross-domain tasks. Finally, it is tested on the constructed Colored MNIST data set, and the results are significantly better than the existing methods.

Key words machine learning; feature expression; cross-domain training; generalization; adaptation

(li_xin@nudt.edu.cn)

中图法分类号 TP391

收稿日期20200923;修回日期:20210304

基金项目国家自然科学基金项目(61977065);科技部重点研发计划项目(2020YFA0713504)

This work was supported by the National Natural Science Foundation of China (61977065) and the Key Research and Development Program of the Ministry of Science and Technology (2020YFA0713504).

通信作者王红霞(wanghongxia@nudt.edu.cn)

Li Xin, born in 1995. PhD. His main research interests include complex network, causal inference, machine learning.

李 鑫,1995年生.博士.主要研究方向为复杂网络、因果推断和机器学习.

Li Zhemin, born in 1997. PhD. His main research interests include data complementation and machine learning. (lizhemin15@163.com)

李哲民,1997年生.博士.主要研究方向为数据补全和机器学习.

Wei Juhui, born in 1997. Master. His main research interests include machine learning, data processing, and information fusion. (2655798881@qq.com)

魏居辉,1997年生.硕士.主要研究方向为机器学习、数据处理和信息融合.

Yang Yating, born in 1997. Master. Her main research interests include network science and data analysis. (ytbwc@foxmail.com)

杨雅婷,1997年生.硕士.主要研究方向为网络科学和数据分析.

Wang Hongxia, born in 1977. PhD, professor. Her main research interests include the mathematical theory of computational imaging and the optimization algorithm in neural networks.

王红霞,1977年生.博士,教授.主要研究方向为计算成像的数学理论、神经网络中的优化算法.