OMSCS-DL课程笔记18-Unsupervised and Semi-Supervised Learning

这个系列是Gatech OMSCS 深度学习课程(CS 7643: Deep Learning)的同步课程笔记。课程内容涉及深度学习的基本理论方法以及它在计算机视觉、自然语言处理以及决策理论等领域中的应用,本节主要介绍无监督学习和半监督学习。

Introduction

目前我们课程中主要介绍了深度神经网络在监督学习中的应用。在监督学习中我们需要为每一个数据赋予标签,然后学习到这个映射函数来实现不同的功能。与监督学习对应的学习任务称为无监督学习(unsupervised learning),此时我们没有任何标签的信息当仍然希望可以从数据中进行学习。在监督学习和无监督学习之间还存在着半监督学习(semi-supervised learning)自监督学习(self-supervised learning)等类型,在这样的任务中我们则需要同时处理包含标签和无标签的数据。

无监督学习的目标是学习到数据自身分布的特征,一般来说可以分为建模(modeling)聚类(clustering)表征(representation)几种不同类型的任务。在深度学习时代我们也有使用深度学习来解决这些问题的方法,比如说深度生成模型、度量学习以及表征学习等等。

无监督学习的难点在于如何处理没有标签的数据。在监督学习中我们可以使用各种损失函数来训练网络,但在无监督学习中我们可能无法使用监督学习中常用的那些损失函数进行训练。同时如何设计训练过程、网络架构同样也是无监督学习的难点。

为了解决这些问题人们设计了不同的训练策略。常用的方法包括使用伪标签来标记数据、利用数据增强来提升模型性能、通过元学习的方法来进行模型初始化以及使用代理任务来辅助训练。

Semi-Supervised Learning

在现实的机器学习任务中往往有着大量的无标签数据,而半监督学习的目标是把这些无标签的数据和有标签的数据结合起来并提升模型的性能。

近期的一些研究指出,我们甚至可以通过半监督学习的方法来获得比监督学习更好的性能。

An Old Idea: Predictions of Multiple Views

半监督学习的一种经典思想是使用模型在未标记的数据上进行预测,然后把预测的结果作为新的带标签数据进行训练。这种方法并不是一个很新的方法,实际上90年代人们就提出过类似的思想来训练数据。

Pseudo-Labeling

更加现代的做法是把数据集划分为有标签数据和无标签数据两类,对于有标签的数据按照通常的监督学习进行训练,而对于无标签数据则为它们赋予伪标签进行训练。在训练伪标签数据时我们需要结合数据增强的方法,通过最小化无标签数据在不同数据增强方法下模型输出的交叉熵来实现无标签数据的训练。

在实际训练时我们可以把标签数据和无标签数据放在一起同时进行训练。这样整个训练流程可以分成标签数据和无标签数据两个分支并且有着各自的损失函数,而把两个分支合并到一起就可以实现端到端的训练了。

当然这种训练方法还有很多技巧,比如说合理地设计标签数据和无标签数据在一个batch中的比例、如何选取生成伪标签的阈值、如何设计学习率策略、利用滑动平均来更新模型参数等等。

试验结果表明,使用半监督学习的方法可以提升分类模型的性能

Other Methods

除了上面介绍过的伪标签方法,在无监督学习中还可以使用对抗样本、知识蒸馏、label propagation等不同方法进行训练。

Few-Shot Learning

少样本学习(few-shot learning)的目的在于解决数据集上样本分布极不均衡的问题。很多常见的标签都有大量的数据,但是一些罕见的标签则通常只有很少的数据作为支持。因此我们希望可以通过在有着大量数据的标签上进行训练从而提升在这些罕见标签上模型的性能。

Finetuning

少样本学习最直接的思路是使用fine-tuning的方式先在常见标签的数据上进行训练,然后在罕见标签数据上进行微调。

一些研究表明使用余弦相似度的分类模型要比神经网络更适合这样的训练模式。

实践中发现fine-tuning的方式并不完全适合少样本学习的问题,这主要是由于训练过程和微调过程是完全割裂的,我们无法直接利用常见标签数据的信息来指导少样本学习过程。

Meta-Training

meta-training的思路是利用常见样本的数据来模拟少样本学习的过程。我们可以把训练数据假装成少样本数据,每次训练时从选择常见标签的几个样本分别作为训练数据和查询数据,这样训练过程和最终的测试过程是完全一致的。

通过meta-training的学习过程,我们可以获得MatchingNet、ProtoNet以及RelationNet等不同形式的模型。

从一个更高的层次来看,meta-training的本质是利用学习的方式来设计学习算法,也就是说我们可以通过学习来改进模型的初始化和训练方式从而更适应少样本学习的任务。

本节课中我们主要介绍改进梯度下降以及模型初始化方式这两种meta-training方法。

Meta-Learner LSTM

回忆梯度下降法的参数更新方法,不难发现它和LSTM中cell的更新方法有一些类似之处。因此meta-learner LSTM的思路是把LSTM的训练方法引入到梯度下降中,这样就可以自适应地修正学习率和遗忘率。

使用LSTM训练方法的另一个好处是整个训练过程是可微的,我们可以把meta-learner LSTM展开成类似于LSTM的计算图这样就可以使用常用的网络模型进行训练。

Model-Agnostic Meta-Learning (MAML)

MAML的思路是直接学习模型的初始化,整个模型的训练过程仍是基于SGD。

和meta-learner LSTM相比,MAML往往有着更好的理论保证。

Unsupervised and Self-Supervised Learning

在无监督学习中我们没有任何关于数据标签的信息,因此我们不能简单地通过最小化损失函数来进行学习。

Autoencoders

无监督学习的经典方法是使用自编码器来学习数据的表征。输入数据通过一个编码器来获得一个低维嵌入,然后通过解码器来还原原始数据。整个模型的损失是输入和输出之间的差距,因此无需标签信息仍然可以进行训练。

自编码器训练得到的编码器还可以服务于不同类型的下游任务。我们可以使用fine tuning的方式把编码器看做是一个特征提取器,然后在下游训练一个分类模型来完成分类任务。

Deep Clustering

对于聚类任务我们希望可以通过一个神经网络把原始空间中的数据映射到一个更加紧凑的空间,从而便于各种聚类算法的应用。

为此,我们可以使用伪标签的方法训练神经网络,然后把网络的特征层作为聚类的输入。

Surrogate Tasks

在长期的工程实践中人们总结了大量不同类型的视觉任务,我们可以通过处理这些代理任务来实现无监督数据特征的学习。

这些常见的代理任务包括colorization、jigsaw、rotation prediction等。它们不需要数据的标签信息,因此可以直接在无标签的数据上进行训练

在评估性能时则可以使用自编码器的方式选择编码器的最后一层作为数据的特征。

目前instance discrimination是一种非常流行的代理任务。在训练时我们把输入图像经过数据增强后的结果作为正样本,然后把数据集中的其它图像作为负样本来训练神经网络,网络的损失函数是其中一个正样本和另一个正样本之间的相似度以及它和其它负样本之间的差异。

当然,如何选取负样本有很多的技巧。我们可以把每个batch中的样本分别作为正样本和负样本,然后计算每一对正负样本之间的损失;而更合理的方法是把负样本储存在一个memory bank中,在采样时直接从memory bank中提取上一个batch计算得到的图像特征,这样可以更有效地利用计算资源。

另一种选取负样本的方法是使用滑动平均更新后的编码器来计算负样本的特征。

试验结果表明,instance discrimination可以很好地训练出图像特征从而服务于各种不同的视觉任务。