Abstract:Humans possess remarkable ability to accurately classify new, unseen images after being exposed to only a few examples. Such ability stems from their capacity to identify common features shared between new and previously seen images while disregarding distractions such as background variations. However, for artificial neural network models, determining the most relevant features for distinguishing between two images with limited samples presents a challenge. In this paper, we propose an intra-task mutual attention method for few-shot learning, that involves splitting the support and query samples into patches and encoding them using the pre-trained Vision Transformer (ViT) architecture. Specifically, we swap the class (CLS) token and patch tokens between the support and query sets to have the mutual attention, which enables each set to focus on the most useful information. This facilitates the strengthening of intra-class representations and promotes closer proximity between instances of the same class. For implementation, we adopt the ViT-based network architecture and utilize pre-trained model parameters obtained through self-supervision. By leveraging Masked Image Modeling as a self-supervised training task for pre-training, the pre-trained model yields semantically meaningful representations while successfully avoiding supervision collapse. We then employ a meta-learning method to fine-tune the last several layers and CLS token modules. Our strategy significantly reduces the num- ber of parameters that require fine-tuning while effectively uti- lizing the capability of pre-trained model. Extensive experiments show that our framework is simple, effective and computationally efficient, achieving superior performance as compared to the state-of-the-art baselines on five popular few-shot classification benchmarks under the 5-shot and 1-shot scenarios

Supervised Contrastive Representation Embedding Based on Transformer for Few-Shot Classification

Feature Transformation for Few-Shot Learning

Few-Shot Learning Meets Transformer: Unified Query-Support Transformers for Few-Shot Classification

Boosting Few-Shot Classification with View-Learnable Contrastive Learning

Few-Shot Learning via Embedding Adaptation With Set-to-Set Functions

SPContrastNet: A Self-Paced Contrastive Learning Model for Few-Shot Text Classification

Learning Embedding Adaptation for Few-Shot Learning

Adaptive Feature Representation Based on Contrastive Learning for Few-Shot Classification

ContrastNet: A Contrastive Learning Framework for Few-Shot Text Classification

Supervised Contrastive Learning for Few-Shot Action Classification

Momentum Memory Contrastive Learning for Transfer-Based Few-Shot Classification

Learning a Few-shot Embedding Model with Contrastive Learning

Auto-view Contrastive Learning for Few-Shot Image Recognition

TATM: Task-Adaptive Token Matching for Few-Shot Transformer

Few-Shot Classification with Contrastive Learning

Unsupervised Few-Shot Image Classification Via One-Vs-all Contrastive Learning

Compare Network with Task-dependent Embeddings for Few-shot Learning

Intra-task Mutual Attention based Vision Transformer for Few-Shot Learning

Few-Shot Image Classification via Contrastive Self-Supervised Learning

Enhancing Few-Shot Image Classification With Cosine Transformer

Feature Transformation Network for Few-Shot Learning