论文 | matching net 《Matching Networks for One Shot Learning》

一 写在前面

未经允许,不得转载,谢谢~~

这是做小样本图像分类的文章,文章是2016年发在nips上的,但是现在为止还是作为很多one-shot小样本领域的competitor进行比较。

二 内容介绍

这里就按照个人的理解简单总结一下文章内容了

2.1 one shot learning 小样本学习

目前的网络依赖大量的标注数据,但是人能够只通过几张图像就快速学习到一个新概念。

由这一点受到启发,引出小样本学习的概念,即测试阶段,对于没有见过的类别只有少数几个标注样本的情况下,如何快速的学习这个概念 ,进行识别。 (可能会有理解或者总结不到位的地方,望见谅~)

2.2 本文主要工作

主要有两个部分的创新之处:

  • modeling level: 提出matching net,利用attention以及memory来获取快速学习新概念的能力;
  • training procedure:在训练时采取与测试一样的原则(只有少量的标准样本)

2.2.1 training procedure

这个地方的创新之处在于不是简单的利用所有的标注数据进行模型的训练,然后再测试阶段进行测试。而是直接在训练的时候就去模仿测试时只有少量标注样本的情况,提出episode的概念。

就是每个episode会包含一个support set(充当训练数据),和一个测试集合batch(充当测试数据)。其中每个support set都是随机生成的,比较有名的就是N-way-K-shot的模式。在所有的类别中随机选择N个类别,然后每个类别又随机选择K个样本作为支持集(k通常小,1-5)。

这样的好处是模拟测试阶段只有少量标注数据的情况,就是测试阶段怎么使用,训练的时候就怎么训练。

其实就是meta-learning的概念。

那我们就知道对于每个episode,网络的优化目标一定是要所有batch中测试样本产生的loss最小。

  • S 表示Support set,其中包含n*k个标注样本{x_i,y_i}
  • B 表示测试集合Batch
  • 那么现在问题就转换给定S的情况下,假设有一个测试样本x,如何得到x属于y的概率;

2.2.2 model architecture

这里来解决给定S的情况下,假设有一个测试样本x,如何得到x属于y的概率的问题;

用数学表示其实就是:


这里作者给出的计算方法是:


  • 其中a表示attention机制;
  • 不管a的情况下,其实可以看成是支持向量集S中{x_i,y_i}所有样本的线性结合,只是给他们赋予了不同的权重。
  • 示意图:


对于a作者采用的也是最简单的方式,就是用x^样本与x样本之间的特征(文中成为embedding)的cosine距离的softmax值进行计算。


  • c表示cosine距离;
  • f和g其实就如上图所示,表示特征提取器;

整篇文章到这里都挺简单的,没有什么特别复杂的地方,设计的方法也比较自然合理。

但是文章对对于f和g这2个特征提取器的设计确实是下了一番功夫。

g特征提取器

  • 首先一般的都是g(x)这样的函数表示,及输出只与输入x有关;

  • 但是作者觉得这样不够,x的特征除了与x本身有关外,还应该与support set中其他的样本也相关;

  • 所以他将原本无序的集合看成是有序的,然后用双向LSTM进行建模;

  • 最后得到的g表示:




  • 其中g'(x), h(i-1), c(i-1), h(i+1), c(i+1)分别表示原始状态,上一时刻的隐含状态,上一时刻的记忆状态,下一时刻的隐含状态,下一时刻的记忆状态;

h特征提取器

  • 对于每个测试样本,使用注意力LSTM来获取测试样本的特征;
  • 最终得到的h表示:


  • 其中rc分别表示读数状态和记忆状态;
  • 最后一步得到的h即为测试样本的特征;

三 写在最后

整体的文章思路挺自然易懂的,但是g和h的特征提取部分确实不是很容易吸收,需要有耐性去看,个人感觉自己的基础功还不是很扎实,对这部分的解读不够透彻。

这里也推荐两篇我觉得写得不错的博客给大家,尤其是【平价数据】One Shot Learning
)这篇真得写的很棒。

参考资料:

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

推荐阅读更多精彩内容