Few-shot Learning
前言:CSDN就是毒瘤,已转向博客园,博客园主页https://www.cnblogs.com/ClarkGable/
一.
1.
与传统的监督学习不同,few-shot leaning的目标是让机器学会学习;使用一个大型的数据集训练模型,训练完成后,给出两张图片,让模型分辨这两张图片是否属于同一种事物。比如训练数据集中有老虎、大象、汽车、鹦鹉等图片样本,训练完毕后给模型输入两张兔子的图片让模型判断是否是同一种事物,或者给模型兔子和狗的图片去判断
![](https://i-blog.csdnimg.cn/blog_migrate/c5116ee416f74624bd23c311ec836e04.png)
2.
训练的目的是靠着Support Set提供的一点信息,让模型判断出Query中的图片是otter这个类别,尽管训练数据集中没有otter这个类别。
k-way n-shot Support Set
k-way: the support set has k classes;
n-shot: every class has n samples.
k way表示支撑集中的类别,n shot表示支撑集中每个类别包含的样本数量
![](https://i-blog.csdnimg.cn/blog_migrate/0d647d4d4b47d4016cf824947526b899.png)
3.
随着Support Set中类别增加,分类准确率会降低
![](https://i-blog.csdnimg.cn/blog_migrate/8ab0070835f1eb2b570a902b90d03347.png)
![](https://i-blog.csdnimg.cn/blog_migrate/74d4d22d8d218584b912ae5671312c35.png)
因为3选1比6选1更容易,准确率更高
同样地,Support Set中shot数量增加,分类准确率会提高
![](https://i-blog.csdnimg.cn/blog_migrate/d81c946b19429bb1ac582fed5ce9ac96.png)
![](https://i-blog.csdnimg.cn/blog_migrate/d57040f05430a4e68f6f724f8b184932.png)
4.
idea:学习一个相似度函数
sim函数来计算两张图片x和x’的相似度,
例如两张狗的图片x1和x2,一张猫的图片x3,sim(x1,x2)=1, sim(x1,x3)=0,sim(x2,x3)=0
基本思想:
(1)首先,从一个大样本数据集中学习一个相似度函数
(2)然后,用相似度函数来做预测
①用query和support set的每一个样本逐一作比较;
②找出相似度得分最高的样本
![](https://i-blog.csdnimg.cn/blog_migrate/a5f1cba86fae5ce85e252c582b3bb1d7.png)
5.常用的数据集
(1)Omniglot
https://github.com/brendenlake/omniglot or https://www.tensorflow.org/datasets/catalog/omniglot
![](https://i-blog.csdnimg.cn/blog_migrate/14ee1556247bab15a6d18da4e4fc9db6.png)
(2)Mini-ImageNet
![](https://i-blog.csdnimg.cn/blog_migrate/fa30e0b8bb448b37d3e109b6c5e544b2.png)
二.连体网络Siamese Network
两种训练Siamese Network的方法
1.每次取两个样本,比较他们的相似度
需要用到一个大的带标签的数据集来训练神经网络,利用训练集来构造正样本Positive Samples和负样本Negative Samples
Positive Samples:每次从一个类别中随机抽取两张图片,把标签设置为1,即相似度满分,用这样的方法,也从其他类别中抽取图片,标签都设置为1;
Negative Samples:随机抽取一个类中的一张图片,排除掉这个类,再从其他类中随机抽取一张图片,把标签设置为0,即相似度为0,这样构造负样本。
![](https://i-blog.csdnimg.cn/blog_migrate/a93d71fc44dc3bbd8e2ea68fca8b38a4.png)
搭建一个卷积神经网络来提取特征,输入图片记为x,输出特征向量记作f(x)
![](https://i-blog.csdnimg.cn/blog_migrate/43bd0be8c20cd48278d8974036ae0894.png)
训练神经网络,将准备好的图片输入神经网络f,提取的两个特征向量记作h1,h2,z = |h1-h2|,再通过一个全连接层输出一个标量,最后使用sigmoid函数得到一个0~1之间的输出,这个输出就可以衡量两个图片之间的相似度,sim(x1,x2)。两张图片属于同一个类别,那么输出应该接近1,如果两张图片属于不同类别,那么输出应该接近0。损失函数是标签Target=1与sim(x1,x2)之间的差别,用来更新全连接层和神经网络f的参数(注意这里的图片输入的是同一个神经网络)之所以叫做连体网络,是这个网络的结构头部连在一起,如下图所示
这样就完成了一轮训练
负样本训练过程与之类似,只是输入时两张不同类别的照片,标签Target=0
训练完成后就可以做one-shot prediction,Support Set中的六个类别都不在训练集里,将Query与Support Set逐一对比,相似度最高的就是预测结果
![](https://i-blog.csdnimg.cn/blog_migrate/2354815e74680da233ca6bf2e64c4e70.png)
![](https://i-blog.csdnimg.cn/blog_migrate/6c7e848d2d3607dbf47fd22a2ecf000d.png)
2.Triplet Loss
每次从训练集中选出3张图片,在这3个图片中选择一个记为xa, anchor(锚点),选出同类别的另一张图片,记作正样本x+,选出其他类别中的一张图片,记作负样本x-;
把三张图片输入卷积神经网络f提取特征向量f(xa), f(x+), f(x-),计算f(xa), f(x+)之间的二范数距离d+,和f(xa),f(x-)之间的二范数距离d-,d+应该很小,d-应该很大;
设置超参数α为margin,如果d-很大,d->d++α,那么损失函数为0,因为很好的区分开了两类图片,反之,损失函数为d++α-d-。
![](https://i-blog.csdnimg.cn/blog_migrate/6e4ad7ef6d6df75dab6214e8a5258f0a.png)
![](https://i-blog.csdnimg.cn/blog_migrate/cd8bec9453eab4243e9db113239c4153.png)
在预测时,把图片都变为特征向量,计算query与他们之间的距离,找出距离最小的
![](https://i-blog.csdnimg.cn/blog_migrate/02f2a720d209d7769daaaf4f74b1cc73.png)
总结
![](https://i-blog.csdnimg.cn/blog_migrate/568fdd50092b291ef0b055c4720cbb37.png)
三. Pretraining + Fine Tuning
大规模数据上做pretraining,小样本上fine tuning
1.Pretraining
![](https://i-blog.csdnimg.cn/blog_migrate/c57ad9376bce5ff92682ac25ed4b14c2.png)
![](https://i-blog.csdnimg.cn/blog_migrate/92fac023f952a0579d691d17dd9a5005.png)
神经网络的结构
用3-way 2-shot的SUpport Set做few-shot分类,用与训练的神经网络提取特征,将每个类别提取的两个特征向量求平均,归一化得到 μ1,μ2,μ3,
提取query的特征向量,归一化得到q,将 μ1,μ2,μ3堆叠起来,得到矩阵M,M与q相乘通过softmax函数得到输出p,显然μ1与q的内积是最大的,所以会将query识别为第一类。
2.Fine Tuning
上一过程中,我们假定的W=M,b=0,其实我们可以在Support Set上学习W和b,计算Support Set所有的pj和真实标签yj之间的CrossEntropy,并使之最小,加上Regularization防止过拟合。
Trick1在初始化分类器时,可以把W初始化为M,b=0
Trick2 Regularization使用Entropy Regularization
因为输出的q是类别的概率值,左边这种情况说明分类器无法判别query属于哪一类,这种情况的entropy很高;我们希望的情况是右边的这种情况,分类器认为query属于第二类