OpenAI 研究员:数据不足时,如何实现监督学习

7a218bed5963440f31e6ad9e6c59db22.jpeg

来源:AI科技评论

作者:Lilian Weng

编译:丛末

编辑:陈彩娴

现实应用中,数据易得,而有标签的数据少有。

一般而言,当监督学习任务面临标签数据不足问题时,可以考虑以下四种解决办法:

1.预训练+微调:首先在一个大规模无监督数据语料库上对一个强大的任务无关模型进行预训练(例如通过自监督学习在自由文本上对语言模型进行预训练,或者在无标签图像上对视觉模型进行预训练),之后再使用一小组标签样本在下游任务上对该模型进行微调。

2.半监督学习:同时从标签样本和无标签样本中学习,研究者使用该方法在视觉任务上进行了大量研究。

3.主动学习:标注成本高,即便成本预算有限,也希望尽可能收集更多的标签数据。主动学习学习选择最有价值的无标签样本,并在接下来的收集过程中收集此类数据,在预算有限的情况下,帮助模型尽可能达到预期效果。

4.预训练+数据集自动生成:给定一个功能强大的预训练模型,可以利用该模型来自动生成更多得多的标签样本。受小样本学习大获成功的驱动,该方法在语言领域的应用尤其普遍。

本文全面介绍半监督学习这一方法,作者为OpenAI研究员Lilian Weng,博客地址:https://lilianweng.github.io/

1

什么是半监督学习?

半监督学习同时使用标签数据和无标签数据来训练模型。

有趣的是,现有关于半监督学习的文献大多集中在视觉任务上。而预训练+微调方法才是语言任务中更常见的范式。

本文所提到的所有方法的损失,都由两部分构成:2618a9c44904b136fde1db9decc6956d.jpeg。其中监督损失156a15c103bd71e7e97e559a1f810bb4.jpeg

在样本全部为标签样本的情况下非常容易计算出来。我们需要重点关注如何设计无监督损失e3caf48b6618844db90079adfcf7bf41.jpeg。加权项93b62e00509d44263d12c1534b44cc25.jpeg通常选择使用斜坡函数,其中t是训练步数,随着训练次数的增加,a35dd82fdee3c99e094fe42e57acbe94.jpeg的占比提升。

声明:此文并不覆盖所有半监督方法,仅聚焦于模型架构调优方面的。关于在半监督学习中,如何使用生成模型和基于图的方法,可以参考《深度半监督学习全览》(An Overview of Deep Semi-Supervised Learning)这篇论文。

2

符号说明表

符号

含义

4e1329a804c0977d1d26c8a93c268060.png

唯一标签的数量。

65f5219a43c04713311707e91f02833d.png

标签数据集,其中7a145c34ce0ffe3c566dc7015374fd52.png是真实标签的独热表示。

ebf39af2a239245f773e76758eecc4af.png

无标签数据集。

9c11f0c0543d0e7e2a53160876117942.png

整个数据集,包括标签样本和无标签样本。

b7b0247e6690510272beca973d87ab3b.png

可以表示无标签样本, 也可以表示标签样本。

44238d7a007bae2a9672cdb25cbba7b0.png

经过增强处理的无标签样本或标签样本。

fc3572935ad5ca85821a2c51b28458ba.png

第i个样本。

543dbc356fd023e5d77385296eec7dd4.png

分别表示损失,监督损失,无监督损失

903387b2668fc43b889912f3cd38393b.png

无监督损失权重,随着训练步数增加而增加。

328190442b49a3d3b69c7801215411c7.png

给定输入情况下,标签数据集的条件概率。

6d0506454c39b338ae71188cb9159d35.png

使用加权θ生成的神经网络,即期望训练出的模型。

be81e9a94bd70421ddeedd695d177ed5.png

逻辑函数f的输出值的向量。

70cdbb8e8b3133d889a03435d0f0eb59.png

预测的标签分布。

d50ccdba426a13c64c82df6bd8d08a3b.png

两个分布间的距离函数,例如均方误差、交叉熵、KL散度等。

d5db9d2c79476da9d206c4b3a54390dd.png

Teacher 模型权重的移动平均线加权超参数。

49af3e04cebcca70bf087c5522d522e5.png

α为混合样本的系数 ,790b1a0a60bb5abf73c1fb7e9b1f1bbc.png

2eb85247873cd3c98cdcb3736fe9cb12.png

锐化预测分布的温度。

c61117295f74eb87476781eac3ddd5c8.png

选择合格的预测结果的置信度阈值。

3

假设

在已有研究文献中,讨论了以下几种假设来支撑对半监督学习方法中的某些设计进行决策。

假设1:平滑度假设(Smoothness Assumptions)

如果两个数据样本在特征空间的高密度区域接近,它们的标签应该会相同或非常相似。

假设2:聚类假设(Cluster Assumptions)

特征空间既有密集区域,也有稀疏区域。密集分组的数据点很自然地形成聚类。同一聚类中的样本应具有相同的标签。这是对假设1的一个小扩展。

假设3:低密度分离假设(Low-density Separation Assumptions)

类之间的决策边界往往位于稀疏的低密度区域,因为如果不这样的话,决策边界就会将高密度聚类分割为分别对应两个聚类的两个类,这就会导致假设1和假设2都失效。

假设4:流形假设(Manifold Assumptions)

高维数据往往位于低维流形上。尽管现实世界的数据可能是在非常高的维度上被观察到的(例如,真实世界的物体/场景的图像),但它们实际上可以被更低维的流形捕获,这种低维流形上会捕获数据的某些属性,并将一些相似的数据点进行紧密组合(例如真实世界的物体/场景的图像,并不是源自于所有像素组合的均匀分布)。这就使得模型能够学习一种更有效的表征方法去发现和评估无标签数据点之间的相似性。这也是表征学习的基础。关于此假设,更详细的阐述可参考《如何理解半监督学习中的流行假设》这篇文章。

链接:https://stats.stackexchange.com/questions/66939/what-is-the-manifold-assumption-in-semi-supervised-learning

4

一致性正则化(Consistency Regularization)

一致性正则化,也叫一致性训练,假设给定相同输入,神经网络中的随机性(例如使用 Dropout算法)或数据增强转换不会更改模型预测。本节中的每个方法都有一个一致性正则化损失:6925358d3ac3d9d3e73f1656e100e8ef.jpeg

SimCLR、BYOL、SimCSE 等多个自监督学习方法都采用了这一思想。相同样本的不同增强版本,产生的表征都相同。语言建模中的交叉视图训练(Cross-view training )和自监督学习中的多视图学习(Multi-view learning)的研究动机相同。

1.Π模型

5ce610f97a7ca561e93ba8f9e1722d24.jpeg

图1:Π-模型概览。同一个输入经过不同的随机增强和 dropout掩膜的扰动产生两个版本,通过网络得到两个输出,Π-模型预测这两个输出是一致的。(图片来源:Laine 、 Aila  2017发表的论文《半监督学习的时序集成》 )

Sajjadi 等人在2016年发表的论文《深度半监督学习的随机变换和扰动正则化》(Regularization With Stochastic Transformations and Perturbations for Deep Semi-Supervised Learning)中,提出了一种无监督学习损失,它能够将同一数据点经过随机变换(例如dropout、随机最大池化)产生两个版本,并在通过网络后输出的两个结果之间的差异最小化。由于其标签没有被明确使用,所以该损失可以应用到无标签数据集。Laine 、Aila 后来在2017年发表的论文《半监督学习的时序集成》(Temporal Ensembling for Semi-Supervised Learning)中,为这种处理方式取了一个名字,叫做 Π模型。

5495d98a2f4075864b10ef5120c7d2d7.jpeg

其中,69dc81f9105ab1360544c7db55897ad8.png指同一个神经网络应用不同的随机增强或dropout掩膜的取值。该损失使用整个数据集。

2.时序集成(Temporal ensembling)

69813ecf25e9c2dcdcf3324e45be5fd5.jpeg

图2:时序集成概览。其学习目标是对每个样本的指数移动平均值( EMA)做出标签预测。(图片来源:Laine 、 Aila  2017发表的论文《半监督学习的时序集成》)

Π模型需要每个样本通过神经网络两次,这就使得计算成本增加一倍。为了减少成本,时序集成模型持续将每个训练样本c21c5357c9216e7b1e89e67c37ef878d.jpeg的实时模型预测的指数移动平均值(EMA)作为学习目标,EMA 在每轮迭代中仅需计算和更新一次。由于时序集成模型的输出5e9daa0d33f1fb72430eaa36f6a487ba.jpeg被初始化为0,因而除以e6130665e73dcc65b880943d19020765.jpeg进行归一化来纠正这一启动偏差。出于同一原因,Adam 优化器也有这样的偏差纠正项。

e425684410312c6ce3f143ad66c1a0e1.jpeg

其中cd0b89e08c0643f74cd79d88fe0c2c37.jpeg是在第t轮迭代中的集成预测,d0eeed4e2fa4f7265c87ce453680f67c.jpeg是在当前回合的模型预测。需要注意的是,由于2e522ed0109ff7f80e47f40308d12d6b.jpeg=0,进行偏差纠正后,a8dab0f443c884e660d5585f49f138cc.jpeg就完全等于在第1轮迭代中的模型预测值ae3c1002a51ea5bd1ef2132ec292b75e.jpeg

3.均值教师(Mean teachers)

e9e5c7c5514fe47e439efc9f57fe9f06.jpeg

图3:Mean Teacher 框架概览(图片来源:Tarvaninen、 Valpola在2017年发表的论文《均值教师模型是表现更好的模范模型:加权平均一致性目标优化半监督深度学习结果》)

时序集成模型将追踪每一个训练样本的标签预测的指数移动平均值作为学习目标。然而,这种标签预测仅在每一次迭代中发生变化,当训练数据集很大时,这种方法就显得冗杂。

为了克服目标更新速度慢的问题,Tarvaninen、 Valpola在2017年发表的论文《均值教师模型是表现更好的模范模型:加权平均一致性目标优化半监督深度学习结果》(Mean teachers are better role models: Weight-averaged consistency targets improve semi-supervised deep learning results)中,提出了Mean Teacher算法,该算法通过跟踪模型权值的移动平均值而不是模型输出来更新目标。权重为θ的原始模型叫做Student模型,将权重为连续多个Student模型的移动平均权重θ′的模型,称为Mean Teacher模型:643a0564708511920c99b1246a6d4639.png

一致性正则化损失是Student模型和Teacher 模型的预测之间的距离,并且该差距应该最小化。Mean Teacher 模型能够提供比Student模型更准确的预测。该结论在实证实验中得到了证实,如图 4 所示。

a76cd57860b217b2c7b66d4472b32550.jpeg

图4:Π 模型和 Mean Teacher模型在 SVHN 数据集上的分类误差。Mean Teacher模型(用橙线表示)比Student模型(用蓝线表示)的表现更好。(图片来源:Tarvaninen、 Valpola在2017年发表的论文《均值教师模型是表现更好的模范模型:加权平均一致性目标优化半监督深度学习结果》)

根据其消融研究:

  • 输入增强方法(例如,输入图像的随机翻转、高斯噪声)或对Student模型进行dropout处理对于模型实现良好的性能是必要的。Teacher模式不需要进行dropout处理。

  • 性能对指数移动平均值的衰减超参数β敏感。一个比较好的策略是在增长阶段使用较小的β=0.99,在后期Student模型改进放缓时使用较大的β=0.999。

  • 结果发现,一致性成本函数的均方误差(MSE)比KL发散等其他成本函数的表现更好。

4.将噪声样本作为学习目标

最近的几种一致性训练方法学习将原始的无标签样本与其相应的增强版本之间的预测差异最小化。这种思路与 Π 模型非常相似,但其一致性正则化损失仅适用于无标签数据。

8232861fc6527ccb70012b09a229b57a.jpeg

图5:使用噪声样本的一致性训练

在Goodfellow等人于2014年发表的论文《解释和利用对抗性样本》(Explaining and Harnessing Adversarial Examples)中,对抗性训练(Adversarial Training)将对抗性噪声应用到输入上,并训练模型使其对此类对抗性攻击具有鲁棒性。该方法在监督学习的应用公式如下:

14f27abb3dccd2921a4746720cbd0e45.jpeg

其中7a2a66eb102d3de6304a456e0d065f2f.jpeg是真实分布,近似于真值标签的独热编码,1e6da61fe94afb22ef3fd2ecd6583b99.jpeg是模型预测,aaf17b306ac7474e9378c099e61bad99.jpeg是计算两个分布之间差异的距离函数。

Miyato 等人在2018年发表的论文《虚拟对抗性训练:对监督和半监督方法都适用的正则化方法》(Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learning)中提出虚拟对抗性训练(Virtual Adversarial Training,VAT),该方法是对抗性训练思想在半监督学习领域的一个延伸。由于9284ffd0e2e677f574064f853eb919b9.jpeg是未知的,VAT 将该未知项替换为当前权重设定为4cd0f8566d74a745255c1358f9c0de76.jpeg时,当前模型对原始输入的预测。需要注意的是,f39088d74ad9cc1abb62619e60826929.jpeg是模型权重的的固定值,因而在6e9f2a4a83ffb131d478e7855357a5e2.jpeg上不会进行梯度更新。

6a67d39e5f942c1174001116301b5671.jpeg

VAT 损失既适用于标签样本,也适用于无标签样本。它计算的是当前模型在每个数据点的预测流形的负平滑度。对这种损失进行优化能够让预测流形更加平滑。

Verma等人在2019年论文《半监督学习的插值一致性训练》(Interpolation Consistency Training for Semi-Supervised Learning)中提出插值一致性训练(Interpolation Consistency Training,ICT),通过添加更多数据点的插值来强化数据集,让模型预测和对应标签的插值尽可能一致。Hongyi Zhang等人在2018年的论文《Mixup:超越经验风险最小化》(Mixup: Beyond Empirical Risk Minimization)中提出MixUp方法,即通过简单的加权和来将两个图像进行混合处理。插值一致性训练即根据这一思路,让预测模型为一个混合样本生成标签,来匹配对应输入的预测插值:

bde60a57fac953872587054b4c95fb29.jpeg

其中e2381379448750a7cc88f78a76562ce5.jpeg表示Mean Teacher 模型的θ的移动平均值。

7627daedbd7956ce9a7970a9d2a276ae.jpeg

图6:插值一致性训练概览。使用MixUp方法生成更多以插值标签作为学习目标的插值样本。(图片来源:Verma等人在2019年论文《半监督学习的插值一致性训练》)

由于两个随机选择的无标签样本属于不同类别的概率很高(例如ImageNet中就有1000个目标类别),因此在两个随机无标签样本之间应用Mixup方法,就很可能生成在决策边界附近的插值。根据低密度分离(Low-density Separation)假设,决策边界往往位于低密度区域。

815f4c73cf77e538e8d5b84668073a79.jpeg

其中f84f004fd124d694d6373bc1745cf860.jpeg表示θ的移动平均值。

与VAT类似,Xie 等人在2020年的论文《一致性训练的无监督数据增强》(Unsupervised Data Augmentation for Consistency Training)中提出的无监督数据增强(Unsupervised Data Augmentation,UDA),学习给无标签样本和增强样本预测相同的输出。UDA特别聚焦于研究噪声的“质量”如何通过一致性训练来影响半监督学习的性能。要想生成有意义和有效的噪声样本,使用先进的数据增强方法至关重要。良好的数据增强方法应该能够产生有效的(即不改变标签)和多样的噪声,并带有有针对性的归纳偏置(Inductive Biases)。

针对图像领域,UDA 采用的是RandAugment方法,该方法由Cubuk 等人在2019年的论文《RandAugment: 减少搜索空间的实用型自动数据增强方法》(RandAugment: Practical automated data augmentation with a reduced search space)中提出。它对Python图像处理库PIL中可用的增强操作进行统一采样,无需学习或优化,因此比使用AutoAugment方法,成本要便宜得多。

0226bb898c17e6c4a2381bcebecf64ac.jpeg

图7:CIFAR-10分类中各种半监督学习方法的比较。在未经过RandAugment处理的50000个样本进行训练时,Wide-ResNet-28-2和PyramidNet+ShakeDrop 在完全监督下,错误率分别为**5.4**和**2.7**。

针对语言领域,UDA结合使用回译( back-translation)和基于TF-IDF的词替换(word replacement)两种方法。回译保留了高层次意义,但是不保留某些词本身,而基于TF-IDF的词替换则去掉TF-IDF分数较低的无信息性词。在语言任务的实验中,研究者发现发现UDA与迁移学习和表征学习是互补的;例如,在域内无标签数据上对BERT模型进行微调(即图8中的45d3aee2f3f9080618a41d034c0a4e2b.png),能进一步提升性能。

d77e973d1f8bad4372d4bce7f771a18a.jpeg

图8:不同文本分类任务上,无监督数据增强方法采用不同初始化设置的比较。(图片来源:Xie 等人在2020年的论文《一致性训练的无监督数据增强》)

在计算 abcce92cbb54ad8ce6c574d548dc42f1.jpeg 时,UDA可以通过使用以下三种训练技巧来优化结果:

  • 低置信度掩膜(Low confidence masking):如果样本的预测置信度低于阈值69aa3ebc93bdd63ccf245f349218cc8d.png,则对其进行掩膜处理。

  • 锐化预测分布(Sharpening Prediction Distribution):在Softmax中使用低温0b0b295ac9667073278f375ca5efd774.png来对预测概率分布进行锐化。

  • 域内数据过滤(In-Domain Data Filtration):为了从大的域外数据集中提取更多的域内数据 ,研究人员训练一个分类器来预测域内标签,然后保留具有高置信度预测的样本作为域内候选样本。

b087c4159acc470d415cb6e91811c8a5.jpeg

其中,4558fdc03d6a166974888dda91a65b97.jpeg是模型权重的固定值,与VAT中的5d5eca4203df64c281e2580fe9b6ce7c.jpeg一样,因而没有梯度更新,d0fedc5e3f4579832777f13e209e3ab2.jpeg是经过增强的数据点,1345e8ac02f5df83620e74412905bab5.jpeg是预测置信度阈值,47ee30c82d2d0701612403538bd7a188.jpeg是分布锐化温度。

5

伪标签(Pseudo Labeling)

Lee等人在2013年的论文《伪标签:深度神经网络的简单而高效的半监督学习方法》(Pseudo-Label : The Simple and Efficient Semi-Supervised Learning Method for Deep Neural Networks)中提出伪标签,它基于当前模型预测的最大的softmax概率,将伪标签分配给无标签样本,然后在完全监督设置下,同时在标签样本和无标签样本上训练模型。

为什么伪标签能起作用?伪标签实际上相当于熵正则化,它将无标签数据的类概率的条件熵( conditional entropy )最小化,从而实现类之间的低密度分离。换句话说,预测的类概率实际上是计算类重叠,最小化熵相当于减少类重叠,从而降低密度分离。

da0bd4c4dd5ab027b746c6c2c2b5e45a.jpeg

图9:(a )表示仅使用600 个标签数据对模型训练后,在MINIST 测试集上进行测试输出的t-SNE可视化结果,(b)表示使用600 个标签数据以及60000 个无标签数据的伪标签对模型进行训练后,在MINIST 测试集上进行测试输出的t-SNE可视化结果。伪标签能够使学到的嵌入空间实现更好的分离效果。(图片来源:Lee等人在2013年的论文《伪标签:深度神经网络的简单而高效的半监督学习方法》)

使用伪标签进行训练自然是一个迭代过程。这里将生成伪标签的模型叫做Teacher 模型,将使用伪标签学习的模型称作Student 模型。

1.标签传播(Label propagation)

Iscen等人在2019年发表的论文《深度半监督学习的标签传播》(Label Propagation for Deep Semi-supervised Learning)中提出标签传播概念,是一种基于特征嵌入在样本之间构建相似图的思想。接着将伪标签从已知样本“扩散”到传播权重与图中成对相似性得分成正比的无标签样本。从概念上来看,它类似于k-NN分类器,两者都存在无法很好地扩展到大型数据集上的问题。

39a80aca32342e3bb3cfaa7a530c3c5c.jpeg

图10:标签传播工作原理示意图。(图片来源:Iscen等人在2019年发表的论文《深度半监督学习的标签传播》)

2.自训练(Self-Training)

自训练不是一个新概念,Scudder等人在1965年发表的论文《自适应模式识别机器的错误概率》(Probability of error of some adaptive pattern-recognition machines)、 Nigram & Ghani等人在CIKM 2000 上发表的论文《分析协同训练的有效性和适用性》(Analyzing the Effectiveness and Applicability of Co-trainin)都涉及到这个概念。它是一种迭代算法,轮番进行以下两步操作,直到每个无标签样本都匹配到一个标签:

  • 首先,它在标签数据上构建分类器。

  • 接着,它使用该分类器预测无标签数据的标签,并将置信度最高的标签转换为标签样本。

Xie 等人在2020年发表的论文《使用噪声学生进行自训练来优化 ImageNet 分类》(Self-training with Noisy Student improves ImageNet classification)中,将自训练应用于深度学习,并取得了巨大成果。在ImageNet分类任务中,研究者首先训练了一个EfficientNet模型作为Teacher模型,为3亿张无标签图像生成伪标签,然后训练了一个更大的EfficientNet模型作为Student 模型,以学习真实标签图像和伪标签图像。在他们的实验设置中,一个关键的元素是在Student模型训练期间加入噪声,而Teacher模型生成伪标签过程则不加入噪声来。因此,他们的方法被称为“噪声学生(Noisy Student )”,即使用随机深度、dropout和RandAugment方法给Student 模型加入噪声。Student 模型表现得比Teacher 模型更好,很大程度上就是加入噪声带来的好处。添加的噪声具有复合效应,可以促使模型在标签数据和无标签数据上产生的决策边界变得平滑。

Student 模型自训练还有其他几个重要的技术设置,包括:

  • Student 模型加应该足够大(即比Teacher 模型大),以适用于更多数据。

  • 加入噪声的Student 模型应该结合数据平衡方法,这对于平衡每个类重的伪标签图像的数量尤其重要。

  • 软伪标签比硬标签效果更好。

加入噪声的Student 模型还提高了防御 FGSM(快速梯度符号攻击,其使用输入数据的损失梯度,并调整输入数据从而最大化损失)的对抗鲁棒性,即便该模型未针对对抗鲁棒性进行优化。

Du 等人在2020年的论文《自训练优化自然语言理解的预训练》(Self-training Improves Pre-training for Natural Language Understanding)中提出SentAugment 方法,旨在解决语言领域进行自训练时,域内无标签数据不足的问题。它依靠句向量从大型语料库中找到无标签的域内样本,并使用检索到的句子进行自训练。

3.减小确认偏误(Confirmation Bias)

确认偏误是由于Teacher模型还不够成熟导致提供错误的伪标签的问题。对错误标签进行过拟合可能并不会产生更好的Student模型。

为了减小确认偏误,Eric Arazo等人在论文《深度半监督学习中的伪标签和确认偏误》(Pseudo-Labeling and Confirmation Bias in Deep Semi-Supervised Learning)中提出了两种新方法。

一种是采用软标签的Mixup方法,给定f3e8d0f691a9a926bc2e2a1a42e5925e.jpeg两个样本和它们对应的真标签和伪标签43b361b642f57d2309151b48869eb8f2.jpeg,插值标签方程可以转化为softmax输出的交叉熵损失:

66eeb06abd61086f45a3083148da4492.jpeg

如果标签样本太少,使用Mixup方法还不够。因此论文作者通过对标签样本进行过采样,从而对每个小批量中的标签样本设置最小数量。这比对标签样本进行权重补偿效果更好,因为它会更频繁地更新,而不是频次少的更大幅度的更新——这种更新其实更不稳定。

与一致性正则化一样,数据增强和dropout方法对于发挥伪标签的作用,也很重要。

Hieu Pham等人在2021年论文《元伪标签》(Meta Pseudo Labels)中提出元伪标签,根据Student模型对在标签数据集上表现的反馈,不断调整Teacher模型。Teacher模型和Student模型同步进行训练,Teacher模型学习生成更好的伪标签,Student模型从伪标签中学习。

将Teacher模型和Student模型的权重分别设为7be1acc58f1512309180f955d47810fe.jpegc273dcae1e7e638b6ffa44a622a66201.jpeg,Student模型在标签样本上的损失定义为81951bec3e032fdfaa446875bfb58a00.jpeg的函数9d592a3177a73a82fb237fc57fc2db87.jpeg,并倾向于通过优化Teacher模型来相应地最小化这一损失。

364aed381d88ae79a680f22e4eafcef6.jpeg

然而,优化上述方程并非易事。借用MAML (模型无关的元学习, Model-Agnostic Meta-Learnin)的思路,它近似于在进行多步d2a4e78aff2a22d56accdaf6a624d2d0.jpeg5391e8d3fb9ea4608c27c9ed8ff2e255.png计算的同时,对b187d5b0762c9eef5118ad0ec28ba66b.png进行一步梯度更新。

790c42a83958d8b921f49499c549c5d0.jpeg

由于使用的是软伪标签,上述目标函数是可微函数。但是如果使用硬伪标签,则是不可微函数,因此需要用到REINFORCE等强化学习方法。

优化过程是在两个模型之间的交替进行的:

  • Student模型更新:给定一批无标签样本d9e64d0dfe3be48db4c0960199a9b845.jpeg,我们可以通过函数cdf3e165fbd5274519c3c7430786b842.jpeg生成伪标签,并使用一步随机梯度下降优化fcf00ed47d06fc0cb46ec80fbc88fbc4.png8ac5360474950b01159edbd3571a86c2.jpeg

  • Teacher模型更新:给定一批标签样本7dbda11a392083c88df9b1712f34d2e0.jpeg,我们重复使用Student模型的更新来优化16153e80c6ff4853217ee9faa64ce8a4.jpeg61062a9cf19c878817a236406e0a17cd.jpeg。此外,还将UDA对象应用于Teacher模型以兼并一致性正则化。

05a28148908d29ecf917c8d20f9693d6.jpeg

图11:元伪标签与其他半监督或自监督学习方法在图像分类任务中的性能比较。(图片来源:Hieu Pham等人在2021年论文《元伪标签》)

6

一致性正则化+伪标签

可以将一致性正则化、伪标签两种方法结合起来,应用到半监督学习。 

1.MixMatch

Berthelot等人在2019年的论文《MixMatch: 一种面向半监督学习的整体性方法》(MixMatch: A Holistic Approach to Semi-Supervised Learning)中提出的MixMatch 方法,是一种应用于半监督学习的整体性方法,它通过整合以下方法来使用无标签数据:

  • 一致性正则化(Consistency regularization):让模型对受到扰动的无标签样本输出相同的预测。

  • 熵最小化(Entropy minimization):让模型对无标签数据输出置信预测。

  • MixUp 增强:让模型在样本之间进行线性行为。

给定一批标签数据2d25a9b25c43c6b84d7bcbaec6244119.png和无标签数据ef8b27c3f7615f928c5d494a257da011.png,通过f180324c75c3ecdf6990171a98bc750b.png运算得到其增强版本,ceee02721dbbaae67bc42686946689f8.pngfa13ab3c2a0fedd95a3d51d1c1dc9000.png分别表示增强样本和给无标签样本预测到的标签。

 86c3be9d9b02802a0d0ea773d6c04e32.jpeg

其中34f850c44761b7172fa88ccd0e734ca6.jpeg是锐化温度,用来减少猜测到的标签重叠;K 是每一个无标签样本生成的增强版本个数;9242cfed04714e3c16f3729d0bcf091c.jpeg是MixMatch 函数中的参数。

对于每一个c5f423a4b03a54b924d735c1d7e8ce2e.jpeg,MixMatch都会生成K个增强版本,c0ad5f65a5170da86261eed13685d924.jpeg等于k分别为1,....,K的83050a2a6952178b6671582928b46cdb.jpeg的增强版本,模型基于平均值b1809ac6a62601bca0bd21f9bab75b1e.jpeg来猜测伪标签。

d5eed3dbe3f56d7e8c1c027d8873459d.jpeg

图12:MixMatch中的“标签猜测”过程:K 个增强无标签样本的平均值,修正预测的边际分布,最终使分布锐化锐。(图片来源:Berthelot等人在2019年的论文《MixMatch: 一种面向半监督学习的整体性方法》)

根据该论文的消融研究,对无标签数据进行 MixUp增强尤其重要。去除伪标签分布上的温度锐化会严重影响性能。对于标签猜测,计算无标签数据多个增强版本的平均值也是必不可少的。

Berthelot等人在2020年的论文《ReMixMatch:使用分布对齐和增强锚定进行半监督学习》(ReMixMatch: Semi-Supervised Learning with Distribution Alignment and Augmentation Anchoring)中,进一步提出ReMixMatch方法,通过引入以下两种新机制来改进MixMatch方法:

215ff4b2dd63134a8fd62729edfd9b97.jpeg

图13:ReMixMatch 方法对MixMatch方法的两项改进的图示。(图片来源:

Berthelot等人在2020年的论文《ReMixMatch:使用分布对齐和增强锚定进行半监督学习》)

1.分布对齐(Distribution alignment)。该方法让边缘分布42ce594e63beff06122deb3599892721.jpeg与真值标签的边缘分布相近。设定a136f452c8f41281d13ee4fedfc7a6e8.jpeg为真值标签的类分布,ed66e002ebc540c215d05a89bac11cbb.jpeg是无标签数据上的预测类分布的移动平均。模型对无标签样本的预测dcb3d8e253c7fc23f22595433c3f8868.jpeg经过归一化处理为7da355e770c4a9ec5d7944923c175cce.jpeg以匹配真实边缘分布。

需要注意的是,如果边际分布不一致,熵最小化则不是一个有用的目标。并且,有标签和无标签数据上的类分布相互匹配的假设确实太绝对了,在现实环境中不一定是正确的。

2.增强锚定(Augmentation Anchoring)。给定一个无标签样本,首先会生成一个弱增强的“锚定”版本,接着使用 CTAugment (控制理论增强,Control Theory Augment)方法对K个强增强版本计算平均值。CTAugment仅仅对讲模型预测保持在网络容差范围内的增强版本进行采样。

ReMixMatch损失由以下几个项组成:

  • 应用了数据增强和Mixup方法的监督损失

  • 应用了数据增强和Mixup方法却使用伪标签作为目标的无监督损失

  • 不使用Mixup方法情况下,单个强增强的无标签图像的交叉熵损失

  • 自监督学习中的旋转损失( rotation loss)。

2.DivideMix

Junnan Li 等人在2020年论文《DivideMix:使用噪声标签学习来实现半监督学习》(DivideMix: Learning with Noisy Labels as Semi-supervised Learning)中提出DivideMix方法,它将半监督学习与使用噪声标签的学习(Learning with noisy labels,LNL)相结合。它通过高斯混合模型(GMM)对每个样本损失分布进行建模,将训练数据动态划分为含有干净样本的标签数据集和含有噪声样本的无标签数据集。

按照Arazo等人在2019年论文《无监督标签噪声建模和损失修正》(Unsupervised Label Noise Modeling and Loss Correction)中提出的想法,他们在每个样本交叉熵损失上拟合了一个二元高斯混合模型。干净样本期望比噪声样本更快得到更低的损失。更小均值的高斯混合模型对应干净标签的聚类,这里将其表示为c。如果高斯混合模型的后验概率9d7dc47ca190b815233a143688bef416.png(即样本属于干净样本集的概率)大于阈值,则该样本被视为干净样本,否则被视为噪声样本。

数据进行聚类的过程叫做协同划分(co-divide)。为了避免确认偏误,DividImax方法同时训练两个交叉的网络,其中每个网络都使用来自另一个网络的数据集分离出来的部分,类似于双 Q 学习 (Double Q-Learning) 的工作原理。

fd326670a33642c045588e98a14a363f.jpeg

‎图14:DivideMix独立训练两个网络,以减少确认偏误。两个网络同时运行‎协同划分,、协同细化 和协同猜测。(图片来源:Junnan Li 等人在2020年论文《DivideMix:使用噪声标签学习来实现半监督学习》)

与MixMatch相比,DivideMix有一个额外的用于处理噪音样本的‎co-divide步骤,并在训练期间做了如下改进:

标签协同细化(Label co-refinement:):它将真值标签b16f19c40f99be1e521353b605d631e8.jpeg与网络预测ba084e63f29f8d76dd2ec877305d3c84.jpeg进行线性组合,其中7acafeec1fee8ffd29668a67e1af5916.jpeg是另一个网络产生干净数据集概率 ed6426d9c4a29157443c901870b8a5ca.jpeg条件下,4340e9461cc5ca54426db1e886ebb18a.jpeg多个增强版本的平均值。

标签协同猜测(Label co-guessing):它对两个模型对无标签数据样本的预测进行平均。

48558b2a7c1a3f90f8fb29802db604ff.jpeg

图15:DivideMix算法。(图片来源:Junnan Li 等人在2020年论文《DivideMix:使用噪声标签学习来实现半监督学习》) 

3.FixMatch

Sohn等人在2020年的论文《FixMatch: 使用一致性和置信度简化半监督学习》(FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence)中提出的FixMatch方法,通过弱增强方法在无标签样本上生成伪标签,并且只保持高置信度的预测。在这里,弱增强和高置信度过滤都有助于生成高质量的可信伪标签目标。然后,FixMatch在给定一个经过大幅增强的样本情况下,预测这些伪标签。

6a05a7089188bb69c079dde210d071b0.jpeg图16:FixMatch方法的工作原理图示。(图片来源:Sohn等人在2020年的论文《FixMatch: 使用一致性和置信度简化半监督学习》) 

5770fcea98f1b36b1018111cf6fe06ab.jpeg

其中bda30a71835f60bc815cdc477ba68978.jpeg是无标签样本的伪标签;63b4775d9e44cd01631e58422e2e0145.jpeg是决定d473414ffced10ca7c10497686d23fff.jpeg909b9af3120067713000e3a90c7fe151.jpeg的相对大小的超参数。

弱增强15abab81dc7e22b894ca277991eaae6d.jpeg:标准的平移和变换增强。

强增强4008df2f4e486a997243cb932a79b751.jpeg:AutoAugment、 Cutout、 RandAugment和 CTAugment等数据增强方法。

155100d9e276dbe55fd7dc85e75a23ab.jpeg

图17:FixMatch和其他几种半监督学习方法在图像分类任务上的性能。(图片来源:Sohn等人在2020年的论文《FixMatch: 使用一致性和置信度简化半监督学习》) 

根据FixMatch的消融研究,

  • 当使用阈值τ时,使用温度参数T对锐化预测分布不会产生显著影响。

  • Cutout和CTAugment是强增强方法,对模型达到良好的性能“功不可没”。

  • 当标签猜测使用强增强来取代弱增强时,模型在训练早期就发散了。如果舍弃弱增强,模型就会过度拟合猜测的标签。

  • 使用弱增强而不是强增强进行伪标签预测,会导致模型性能不稳定。强数据增强,对于模型性能的稳定性而言,至关重要。

7

结合强大的预训练

该方法通过自监督学习在大型无监督数据语料库上对与任务无关的模型进行预训练,然后使用小型标签数据集在下游任务上对该模型进行微调,这是一种常见的范式,尤其是在语言任务中。研究表明,如果将半监督学习与预训练相结合,模型可以获得额外的收益。

Zoph 等人在2020年的论文《重新思考预训练和自训练》(Rethinking Pre-training and Self-training),研究了自训练比预训练更有效多少。他们的实验设置是使用ImageNet 进行预训练或自训练,从而改进 COCO结果。需要注意的是,在使用 ImageNet 进行自训练时,它会丢弃标签,仅将 ImageNet 样本用作无标签的数据点。何恺明等人在2018年的论文《重新思考ImageNet 预训练》(Rethinking ImageNet Pre-training)中已经证明,如果下游任务(例如目标检测)非常不同,ImageNet 分类预训练的效果就不是很好。

5758204c7057f372114c086110469964.jpeg

图18:(a) 数据增强(从弱到强)的结果和 (b) 标签数据集大小对目标检测性能的影响。在图例中:“Rand Init”表示经过随机权重初始化的模型;`ImageNet` 使用在ImageNet数据集上Top-1准确率为84.5%的预训练模型进行初始化;`ImageNet++`使用在 ImageNet数据集上Top-1准确率为86.9%的预训练模型进行初始化。(图片来源:Zoph 等人在2020年的论文《重新思考预训练和自训练》)

该实验获得了一系列有趣的发现:

  • 可用于下游任务的标签样本越多,预训练的有效性就越低。预训练在低数据模式(20%)下是有帮助的,但在高数据情况下是中性的或起反作用的。

  • 在高数据/强增强模式下,即便预训练会起反作用,自训练也是有帮助的。

  • 即使使用相同的数据源,自训练也可以在预训练的基础上带来额外的改进。

  • 自监督预训练(例如通过 SimCLR进行预训练)会损害模型在高数据模式下的性能,跟监督预训练差不多。

  • 联合训练监督和自监督学习目标有助于解决预训练和下游任务之间的不匹配问题。预训练、联合训练和自训练都是加性的。

  • 噪声标签或非目标标签(即预训练标签未与下游任务标签对齐)比目标的伪标签更差。

  • 自训练在计算上比在预训练模型上进行微调,更昂贵。

Ting Chen等人在2020年的论文《大型自监督模型是强大的半监督学习者》(Big Self-Supervised Models are Strong Semi-Supervised Learners)中,提出了一个三步程序(three-step procedure)方法,将自监督预训练、有监督微调以及自训练的优势都结合在一起:

1.使用无监督或子监督方法对一个大模型进行训练;

2.在一些标签示例上对该模型进行有监督微调,其中使用大型(深且宽)的神经网络至关重要,因为使用更少的标签样本下,模型越大,性能更佳。

3.在自训练中采用伪标签,对无标签示例进行蒸馏。

a.可以将一个大模型的知识蒸馏到一个小模型中,因为这种特定任务的用法并不需要学到的表示的额外容量。

b.蒸馏损失公式如下,其中Teacher网络是固定的,权重为47ed7afcff61802485b18a0643fe06dc.jpeg

 0adaba53642f70f44e6ce2006211e843.jpeg

3c956dbf1106f035d214d4451700c424.jpeg

图19:半监督学习框架通过与任务无关的无监督预训练(左)和特定任务的自训练和蒸馏(右)来使用无标签的数据语料库。(图片来源:Ting Chen等人在2020年的论文《大型自监督模型是强大的半监督学习者》) 

论文作者在 ImageNet 分类任务上进行了实验。自监督预训练使用 SimCLRv2,SimCLR 的直接改进版本。他们在实证研究中的观察结果,证实了Zoph 等人在2020年提出的一些成果:

  • 更大模型的标签学习更加高效;

  • SimCLR 中更大/更深的project heads可以改善表征学习;

  • 使用无标签数据进行蒸馏,能优化半监督学习。

1028725b946eecd2678f30f100e82b0d.jpeg

图20:SimCLRv2 + 半监督蒸馏在 ImageNet 分类上的性能对比。(图片来源:Ting Chen等人在2020年的论文《大型自监督模型是强大的半监督学习者》)

通过对最近半监督学习方法的总结,我们可以发现不少方法都旨在减少确认偏误:

  • 通过先进的数据增强方法将有效且多样化的噪声应用于样本。

  • 在处理图像时,MixUp 是一种有效的数据增强方法。该方法同样可用于语言任务,实现较小的增量优化((Guo et al. 2019)。

  • 设置阈值,并去掉置信度低的伪标签。

  • 设置每个小批量中的标签样本的最少数量。

  • 锐化伪标签分布来减少类重叠。 

如需引用,请注明:

 
 

注明:

@article{weng2021semi,  title   = "Learning with not Enough Data Part 1: Semi-Supervised Learning",
  author  = "Weng, Lilian",
  journal = "lilianweng.github.io",
  year    = "2021",
  url     = "https://lilianweng.github.io/posts/2021-12-05-semi-supervised/"
}

参考文献:

[1] Ouali, Hudelot & Tami. “An Overview of Deep Semi-Supervised Learning” arXiv preprint arXiv:2006.05278 (2020).

[2] Sajjadi, Javanmardi & Tasdizen “Regularization With Stochastic Transformations and Perturbations for Deep Semi-Supervised Learning.” arXiv preprint arXiv:1606.04586 (2016).

[3] Pham et al. “Meta Pseudo Labels.” CVPR 2021.

[4] Laine & Aila. “Temporal Ensembling for Semi-Supervised Learning” ICLR 2017.

[5] Tarvaninen & Valpola. “Mean teachers are better role models: Weight-averaged consistency targets improve semi-supervised deep learning results.” NeuriPS 2017

[6] Xie et al. “Unsupervised Data Augmentation for Consistency Training.” NeuriPS 2020.

[7] Miyato et al. “Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learning.” IEEE transactions on pattern analysis and machine intelligence 41.8 (2018).

[8]Ian J. Goodfellow et al.”Explaining and Harnessing Adversarial Examples“,2014

[9] Verma et al. “Interpolation consistency training for semi-supervised learning.” IJCAI 2019

[10] Lee. “Pseudo-label: The simple and efficient semi-supervised learning method for deep neural networks.” ICML 2013 Workshop: Challenges in Representation Learning.

[11] Iscen et al. “Label propagation for deep semi-supervised learning.” CVPR 2019.

[12] Xie et al. “Self-training with Noisy Student improves ImageNet classification” CVPR 2020.

[13] Jingfei Du et al. “Self-training Improves Pre-training for Natural Language Understanding.” 2020

[14 Iscen et al. “Label propagation for deep semi-supervised learning.” CVPR 2019

[15] Arazo et al. “Pseudo-labeling and confirmation bias in deep semi-supervised learning.” IJCNN 2020.

[16] Berthelot et al. “MixMatch: A holistic approach to semi-supervised learning.” NeuriPS 2019

[17] Berthelot et al. “ReMixMatch: Semi-supervised learning with distribution alignment and augmentation anchoring.” ICLR 2020

[18] Eric Arazo et al. Unsupervised Label Noise Modeling and Loss Correction,2019

[19] Sohn et al. “FixMatch: Simplifying semi-supervised learning with consistency and confidence.” CVPR 2020

[20] Junnan Li et al. “DivideMix: Learning with Noisy Labels as Semi-supervised Learning.” 2020 [code]

[21] Zoph et al. “Rethinking pre-training and self-training.” 2020.

[22] Chen et al. “Big Self-Supervised Models are Strong Semi-Supervised Learners” 2020

未来智能实验室的主要工作包括:建立AI智能系统智商评测体系,开展世界人工智能智商评测;开展互联网(城市)大脑研究计划,构建互联网(城市)大脑技术和企业图谱,为提升企业,行业与城市的智能水平服务。每日推荐范围未来科技发展趋势的学习型文章。目前线上平台已收藏上千篇精华前沿科技文章和报告。

  如果您对实验室的研究感兴趣,欢迎加入未来智能实验室线上平台。扫描以下二维码或点击本文左下角“阅读原文”

2fe0a7de23178bfe79f9d375fa9402bc.jpeg

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值