English

A Baseline for Few-Shot Image Classification

Machine Learning 2020-10-23 v5 Computer Vision and Pattern Recognition Machine Learning

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.

Keywords

Cite

@article{arxiv.1909.02729,
  title  = {A Baseline for Few-Shot Image Classification},
  author = {Guneet S. Dhillon and Pratik Chaudhari and Avinash Ravichandran and Stefano Soatto},
  journal= {arXiv preprint arXiv:1909.02729},
  year   = {2020}
}
R2 v1 2026-06-23T11:07:25.239Z