A Baseline for Few-Shot Image Classification

Guneet S. Dhillon, Pratik Chaudhari, Avinash Ravichandran, Stefano Soatto
9/6/2019
cs.LGcs.CVstat.ML

Official code: amazon-science/few-shot-baseline (28 stars, Apache-2.0, last commit 2021-08-25).

Abstract

Fine-tuning a deep network trained with the standard cross-entropy loss is a strong baseline for few-shot learning. When fine-tuned transductively, this outperforms the current state-of-the-art on standard datasets such as Mini-ImageNet, Tiered-ImageNet, CIFAR-FS and FC-100 with the same hyper-parameters. The simplicity of this approach enables us to demonstrate the first few-shot learning results on the ImageNet-21k dataset. We find that using a large number of meta-training classes results in high few-shot accuracies even for a large number of few-shot classes. We do not advocate our approach as the solution for few-shot learning, but simply use the results to highlight limitations of current benchmarks and few-shot protocols. We perform extensive studies on benchmark datasets to propose a metric that quantifies the "hardness" of a few-shot episode. This metric can be used to report the performance of few-shot algorithms in a more systematic way.

View on arXivView PDF

Code Implementations(1)

Apache-2.0
288PythonAug 25, 20215 years agoApache-2.0

Similar Papers

Cite this paper

@article{dhillon2019baseline,
  title  = {A Baseline for Few-Shot Image Classification},
  author = {Guneet S. Dhillon and Pratik Chaudhari and Avinash Ravichandran and Stefano Soatto},
  year   = {2019},
  eprint = {1909.02729},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/1909.02729}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/1909.02729.svg)](https://codeofpaper.com/paper/1909.02729)

Discussion