Revisiting Local Descriptor Based Image-To-Class Measure for Few-Shot Learning

By Wenbin Li, Jinglin Xu, Jing Huo, Yang Gao, Jiebo Luo. Computer Vision and Pattern Recognition, 2019

591 citations recorded by the source corpus.

Few-shot learning in image classification aims to learn a classifier to classify images when only few training examples are available for each class. Recent work has achieved promising classification performance, where an image-level feature based measure is usually used. In this paper, we argue that a measure at such a level may not be effective enough in light of the scarcity of examples in few-shot learning. Instead, we think a local descriptor based image-to-class measure should be taken, inspired by its surprising success in the heydays of local invariant features. Specifically, building upon the recent episodic training mechanism, we propose a Deep Nearest Neighbor Neural Network (DN4 in short) and train it in an end-to-end manner. Its key difference from the literature is the replacement of the image-level feature based measure in the final layer by a local descriptor based image-to-class measure. This measure is conducted online via a k-nearest neighbor search over the deep local descriptors of convolutional feature maps. The proposed DN4 not only learns the optimal deep local descriptors for the image-to-class measure, but also utilizes the higher efficiency of such a measure in the case of example scarcity, thanks to the exchangeability of visual patterns across the images in the same class. Our work leads to a simple, effective, and computationally efficient framework for few-shot learning. Experimental study on benchmark datasets consistently shows its superiority over the related state-of-the-art, with the largest absolute improvement of 17% over the next best. The source code can be available from https://github.com/WenbinLee/DN4.git.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Radiology, Nuclear Medicine and Imaging

View Paper: https://www.semanticscholar.org/paper/821c0d99eec8e2ee312e08b5605924e284b7cc7c

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Guosheng Lin
  2. Chi Zhang
  3. Chunhua Shen
  4. Xilin Chen
  5. Shiguang Shan
  6. Yu-Wing Tai
  7. Minsu Cho
  8. Yujun Cai

In Computer Vision and Pattern Recognition

  1. Dacheng Tao
  2. Luc Van Gool
  3. Chunhua Shen
  4. Xiaogang Wang
  5. Yi Yang
  6. Ming-Hsuan Yang
  7. Trevor Darrell
  8. Bernt Schiele

In Radiology, Nuclear Medicine and Imaging

  1. Chi Zhang
  2. Guosheng Lin
  3. Shiguang Shan
  4. Evgeniya Ustinova
  5. Bo Jiang
  6. Ivan V. Oseledets
  7. Victor S. Lempitsky
  8. Chunhua Shen

Citations in the corpus