FaceNet

  解决人脸检测的后续问题:face verification (is this the same person), recognition (who is this person) and clustering (find common people among these faces)。

一、传统的基于CNN的人脸识别方法为:

  1. 利用CNN的siamese网络来提取人脸特征
  2. 然后利用SVM等方法进行分类

二、FaceNet亮点:

  1. 利用DNN直接学习到从原始图片到欧氏距离空间的映射,从而使得在欧式空间里的距离的度量直接关联着人脸相似度;
  2. 引入triplet损失函数,使得模型的学习能力更高效。
  3. 结果示意图:



  这是一个简单的示例,其中图中的数字表示图像特征之间的欧式距离,可以看到,图像的类内距离明显的小于类间距离,阈值大约为1.1左右。

三、实现

  这篇文章中,最大的创新点应该是提出不同的损失函数,直接是优化特征本身,用特征空间上的点的距离来表示两张图像是否是同一类。网络结构如下:



  上图是文章中所采用的网络结构,上图步骤可以描述为:
  1、前面部分采用一个CNN结构提取特征
  2、CNN之后接一个L2标准化,这样图像的所有特征会被映射到一个超球面上
  3、再接入一个embedding层(嵌入函数),嵌入过程可以表达为一个函数,即把图像x通过函数f映射到d维欧式空间

  4、将嵌入层归一化到超球面上,例如可以使用:|| f(x) ||_2^2 = 1(或者用softmax也可以实现)
  5、接着,再去优化这些特征,而文章这里提出了一个新的损失函数,triplet损失函数(优化函数),而这也是文章最大的特点所在。

Triplet Loss(三元组损失函数):

  以下是Triplet损失函数的原理(Triplet翻译为三元组):
  思想:什么是Triplet Loss呢?故名思意,也就是有三张图片输入的Loss(之前的都是Double Loss或者是SingleLoss)。
  本文通过LDA思想训练分类模型,使得类内特征间隔小,类间特征间隔大。为了保证目标图像 与类内图片(正样本)特征距离小,与类间图片(负样本)特征距离大。需要Triplet损失函数来实现。



  根据上文,可以构建一个约束条件:

||f(x_i^a) - f(x_i^p)||_2^2 + \alpha < ||f(x_i^a) - f(x_i^n)||_2^2, \ \forall(f(x_i^a), f(x_i^p), f(x_i^n)) \in \tau \quad(1)

  其中,\tau:所有可能的三元组集合
  公式(1)中,\alpha决定了类间距的最小值,如下图所示,我们可以看到\alpha越大,类间距越大(本文\alpha=0.2


  把上式(1)写成损失(优化)函数,通过优化(减小)损失函数的值,来优化模型。损失函数为:

L = \sum_i^N[||f(x_i^a) - f(x_i^p)||_2^2 - ||f(x_i^a) - f(x_i^n)||_2^2 + \alpha]_+, \quad (2)

  其中,N:训练集中样本的个数
  从上面的两个公式中可以看到,如果一个三元组满足公式(1),也就意味着已经分类正确,不需要训练,如果把这种样本加入到loss里面,那么对网络的收敛帮助很小。所以,我们需要找那么hard triplets(也即是不满足公式(1)的三元组)来促进模型的训练。

四、Triplet Selection

  1、问题描述:
  为了确保模型快速收敛,选择违反公式1的约束条件的三元组是至关重要的。这意味着给定x_i^a,我们需要:
  (1)选择一个x_i^p(hard positive),使得argmax_{x_i^p}||f(x_i^a) - f(x_i^p)||_2^2
  (2)选择一个x_i^n(hard negtive),使得argmin_{x_i^n}||f(x_i^a) - f(x_i^n)||_2^2
  在整个训练集上穷举所有的三元组非常困难。而且,一些标注错误的图片和低质量的图片有可能主导hard positives和hard negatives,这反而会对训练产生不利影响。因此,我们有两种解决方案:
  2、解决方案:
  (1)离线更新三元组(每隔n步)。采用最近的网络模型的检测点 并 计算数据集的子集的argmin和argmax(局部最优)。
  (2)在线更新三元组。在mini-batch上 选择不好的正(类内)/负(类间)训练模型。
  本文采用在线的方式,采用这种方式需要较大的batch(本文为1800)。在每个batch中,首先选择40张属于同一类的图片,然后随机补充其他类的图片。
  3、实际采用的解决方案:
  (1)在实际训练中,作者选取一个batch中的所有positive的anchor对,而不是hardest positive,因为这种方式在训练中更稳定,而且训练的初始阶段收敛的也更快。(分析:单个数据更容易受到噪声和错误数据的干扰,而使用更多的数据相当于做了平滑操作,抗干扰性更强,函数的连续性更好)
  (2)虽然理论上需要选择最为困难的负样本,但是在实际中,这样会容易导致在训练中很快地陷入局部最优,或者说整个学习崩溃f(x)=0。为了避免这个问题,在选择negative的时候,我们选择满足以下公式的样本:||f(x_i^a) - f(x_i^p)||_2^2 < ||f(x_i^a) - f(x_i^n)||_2^2, \quad(3)
  我们把这种负样本叫做semi-hard。为什么叫semi-hard呢?我们把公式(3)和公式(1)作比较,发现只是少了一个\alpha,而\alpha代表的是最小类间距。也就是说,这个三元组的样本满足了类内距离小于类间距离得聚类要求,但是类间距离还没有足够远(应该要大于\alpha),所以叫semi-hard
  可以看出,这种方式和上面实际中选择hard positive的策略类似,不是选择单个的hardest negative的样本,而是所有满足公式(3)的样本都会被采用。

参考:https://www.cnblogs.com/lijie-blog/p/10168073.html,写的不错,所以偷个懒,本文中有些部分直接摘抄自这篇博客。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,311评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,339评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,671评论 0 342
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,252评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,253评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,031评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,340评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,973评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,466评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,937评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,039评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,701评论 4 323
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,254评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,259评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,485评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,497评论 2 354
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,786评论 2 345

推荐阅读更多精彩内容