Aligning Visual Prototypes with BERT Embeddings for Few-Shot Learning
This work addresses the challenge of recognizing new image categories from few examples, offering incremental improvements in metric-based few-shot learning.
The paper tackles few-shot learning by using class names with BERT embeddings to improve visual prototype alignment, achieving consistent state-of-the-art improvements on datasets like miniImageNet, CUB, and tieredImageNet.
Few-shot learning (FSL) is the task of learning to recognize previously unseen categories of images from a small number of training examples. This is a challenging task, as the available examples may not be enough to unambiguously determine which visual features are most characteristic of the considered categories. To alleviate this issue, we propose a method that additionally takes into account the names of the image classes. While the use of class names has already been explored in previous work, our approach differs in two key aspects. First, while previous work has aimed to directly predict visual prototypes from word embeddings, we found that better results can be obtained by treating visual and text-based prototypes separately. Second, we propose a simple strategy for learning class name embeddings using the BERT language model, which we found to substantially outperform the GloVe vectors that were used in previous work. We furthermore propose a strategy for dealing with the high dimensionality of these vectors, inspired by models for aligning cross-lingual word embeddings. We provide experiments on miniImageNet, CUB and tieredImageNet, showing that our approach consistently improves the state-of-the-art in metric-based FSL.