【论文笔记】Rethinking Few-Shot Image

2022-03-02  本文已影响0人  huyuanda

一、写在前面

本文是 MIT CSAIL & Google Research 在2020年关于 Few-Shot Learning的又一篇力作,受 ICLR 2020 的经典文章 A baseline for few-shot image classification 启发,提出了如下假设:

Embeddings are the most critical factor to the performance of few-shot learning/meta learning algorithms; better embeddings will lead to better few-shot testing performance (even with a simple linear classier)

二、主要内容

2.1 backgrounds

meta-learning 大致分为两个主要思路:

few-shot classification tasks 也有两个思路:

2.2 related work

2.3 motivation

Few-shot classification中一直有这样一个问题:meta-learning 和 representation-learning 哪个表现更好。最近的文章发现仅用常规的模型做 fine-tuning,就可以达到与最好的meta-learning方法差不多的效果。因此本文探索在Few-shot classification中如何学习好的 representation。

三、文章方法

3.1 key idea

正如题目所说,A Good Embedding is All You Need。本文提出一个简单的few-shot classification baseline,学习 good learned representations 可以干掉一众 meta-learning algorithms

3.2 methods

具体怎么学习 good learned representations ?

  1. meta-training阶段,在组合训练集(可以理解为大规模数据集)上训练一个图像分类任务得到 embedding model,固定参数。
  1. meta-testing 阶段,将所有的 support images 和 query images 输入至上一步训练好的embedding model 得到对应的 embeddings (与组合训练集可以不同分布)
  2. 然后用support images embeddings 训练一个简单的线性模型 base learner
  3. 最后将 query images embeddings 输入至线性模型中得到分类结果


四、写在最后

这篇工作给 few-shot-classification打开了一个大门,验证了representation的重要性。此外,作者没有全盘否认meta-learning,只是说在 few-shot-classification任务中本文的方法比meta-learning好。

上一篇 下一篇

猜你喜欢

热点阅读