Selecting Relevant Features from a Multi-domain Representation for Few-shot Classification
About
Popular approaches for few-shot classification consist of first learning a generic data representation based on a large annotated dataset, before adapting the representation to new classes given only a few labeled samples. In this work, we propose a new strategy based on feature selection, which is both simpler and more effective than previous feature adaptation approaches. First, we obtain a multi-domain representation by training a set of semantically different feature extractors. Then, given a few-shot learning task, we use our multi-domain feature bank to automatically select the most relevant representations. We show that a simple non-parametric classifier built on top of such features produces high accuracy and generalizes to domains never seen during training, which leads to state-of-the-art results on MetaDataset and improved accuracy on mini-ImageNet.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Few-shot classification | MiniImagenet | 5-way 5-shot Accuracy79.25 | 98 | |
| Image Classification | Mini-Imagenet (test) | Acc (5-shot)81.19 | 75 | |
| Few-shot classification | Meta-Dataset (test) | Omniglot93.1 | 48 | |
| Few-shot classification | Meta-Dataset | Avg Seen Accuracy75.6 | 45 | |
| Few-shot classification | Meta-Dataset 1.0 (test) | ILSVRC Accuracy57.2 | 42 | |
| Few-shot Image Classification | Meta-Dataset (test) | Omniglot Accuracy95.8 | 40 | |
| Few-shot Image Classification | Aircraft (test) | Mean Accuracy85.5 | 28 | |
| Few-shot classification | Quick Draw Meta-Dataset (test) | Mean Accuracy81.8 | 10 | |
| Few-shot classification | Fungi Meta-Dataset (test) | Mean Accuracy64.3 | 10 | |
| Few-shot classification | Omniglot Meta-Dataset (test) | Mean Accuracy94.1 | 10 |