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

TASK-SENSITIVE DISCRIMINATIVE MUTUAL ATTENTION NETWORK FOR FEW-SHOT LEARNING

TAAN: Task-Aware Attention Network for Few-shot Classification

Task-Oriented Channel Attention for Fine-Grained Few-Shot Classification

Task-Related Saliency for Few-Shot Image Classification

Task Discrepancy Maximization for Fine-grained Few-Shot Classification

A Simple Task-aware Contrastive Local Descriptor Selection Strategy for Few-shot Learning between inter class and intra class

Spatial Attention Network for Few-Shot Learning

Multi-Scale Adaptive Task Attention Network for Few-Shot Learning.

TALDS-Net: Task-Aware Adaptive Local Descriptors Selection for Few-shot Image Classification

DLA-MatchNet for Few-Shot Remote Sensing Image Scene Classification

Task Encoding with Distribution Calibration for Few-Shot Learning

Task Adaptive Feature Distribution Based Network for Few-shot Fine-grained Target Classification

Task-wise Attention Guided Part Complementary Learning for Few-Shot Image Classification

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

Channel-spatial attention network for fewshot classification

Few-shot Classification with Task-Adaptive Semantic Feature Learning

Few-Shot Learning Based on Deep Learning for Image Classification

Multi-attention mutual information distributed framework for few-shot learning

Cross Attention Network for Few-shot Classification

Multi-instance attention network for few-shot learning