Pushing the Limits of Simple Pipelines for Few-Shot Learning: External Data and Fine-Tuning Make a Difference
Shell Xu Hu, Da Li, Jan Stühmer, Minyoung Kim, Timothy M. Hospedales
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/hushell/pmf_cvpr22OfficialIn paperpytorch★ 199
Abstract
Few-shot learning (FSL) is an important and topical problem in computer vision that has motivated extensive research into numerous methods spanning from sophisticated meta-learning methods to simple transfer learning baselines. We seek to push the limits of a simple-but-effective pipeline for more realistic and practical settings of few-shot image classification. To this end, we explore few-shot learning from the perspective of neural network architecture, as well as a three stage pipeline of network updates under different data supplies, where unsupervised external data is considered for pre-training, base categories are used to simulate few-shot tasks for meta-training, and the scarcely labelled data of an novel task is taken for fine-tuning. We investigate questions such as: (1) How pre-training on external data benefits FSL? (2) How state-of-the-art transformer architectures can be exploited? and (3) How fine-tuning mitigates domain shift? Ultimately, we show that a simple transformer-based pipeline yields surprisingly good performance on standard benchmarks such as Mini-ImageNet, CIFAR-FS, CDFSL and Meta-Dataset. Our code and demo are available at https://hushell.github.io/pmf.
Tasks
Benchmark Results
| Dataset | Model | Metric | Claimed | Verified | Status |
|---|---|---|---|---|---|
| CIFAR-FS 5-way (1-shot) | P>M>F (P=DINO-ViT-base, M=ProtoNet) | Accuracy | 84.3 | — | Unverified |
| CIFAR-FS 5-way (5-shot) | P>M>F (P=DINO-ViT-base, M=ProtoNet) | Accuracy | 92.2 | — | Unverified |
| Meta-Dataset | P>M>F (P=DINO-ViT-base, M=ProtoNet) | Accuracy | 84.75 | — | Unverified |
| Mini-Imagenet 5-way (1-shot) | P>M>F (P=DINO-ViT-base, M=ProtoNet) | Accuracy | 95.3 | — | Unverified |
| Mini-Imagenet 5-way (5-shot) | P>M>F (P=DINO-ViT-base, M=ProtoNet) | Accuracy | 98.4 | — | Unverified |