An improved transformer model based on global and local features for few-shot learning with noise samples

Yingjian Jiang
DOI: https://doi.org/10.1117/12.3004881
2023-10-09
Abstract:It is a challenging problem to identify new categories with few support samples. In the current few-shot learning tasks, it is usually assumed that the support set is accurate, but in real scenarios, the support set may still have noisy samples. Therefore, how to learn a reliable model on the noisy support set is the direction of current research. To address the situation in few-shot learning task where the support set contains noisy samples, we propose a Transformer model for noisy few-shot learning that considers both global and local features. Specifically, we consider the difference between local features and global features between noise samples and support set samples, then align the local features of the samples, and use the attention mechanism of the transformer model to reduce the interference of noise samples to the model. To evaluate the effectiveness of our proposed models, we extensively test these methods on noisy versions of MiniImageNet and TieredImageNet. Our results show that the model performs competitively in the presence of label noise.
Computer Science,Engineering
What problem does this paper attempt to address?