Abstract:While humans can excel at image classification tasks by comparing a few images, existing metric-based few-shot classification methods are still not well adapted to novel tasks. Performance declines rapidly when encountering new patterns, as feature embeddings cannot effectively encode discriminative properties. Moreover, existing matching methods inadequately utilize support set samples, focusing only on comparing query samples to category prototypes without exploiting contrastive relationships across categories for discriminative features. In this work, we propose a method where query samples select their most category-representative features for matching, making feature embeddings adaptable and category-related. We introduce a category alignment mechanism (CAM) to align query image features with different categories. CAM ensures features chosen for matching are distinct and strongly correlated to intra-and inter-contrastive relationships within categories, making extracted features highly related to their respective categories. CAM is parameter-free, requires no extra training to adapt to new tasks, and adjusts features for matching when task categories change. We also implement a cross-validation-based feature selection technique for support samples, generating more discriminative category prototypes. We implement two versions of inductive and transductive inference and conduct extensive experiments on six datasets to demonstrate the effectiveness of our algorithm. The results indicate that our method consistently yields performance improvements on benchmark tasks and surpasses the current state-of-the-art methods.

Selective Model for Few Shot Classification

Selectively Augmented Attention Network for Few-Shot Image Classification

Channel-Spatial Support-Query Cross-Attention for Fine-Grained Few-Shot Image Classification

Attributes-Guided and Pure-Visual Attention Alignment for Few-Shot Recognition

Cross Attention Network for Few-shot Classification

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

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

Few-Shot Image Classification Based on Asymmetric Convolution and Attention Mechanism

Self-Adaptive Embedding for Few-Shot Classification by Hierarchical Attention.

Few-Shot Classification with Cross-Feature Fusion.

Channel-spatial attention network for fewshot classification

Focus Your Attention when Few-Shot Classification

Cross Attention with Deep Local Features for Few-Shot Image Classification

Selective, Structural, Subtle: Trilinear Spatial-Awareness for Few-Shot Fine-Grained Visual Recognition

Category Alignment Mechanism for Few-Shot Image Classification

Class-relevant Patch Embedding Selection for Few-Shot Image Classification

Improved fine-grained image classification in few-shot learning based on channel-spatial attention and grouped bilinear convolution

Feature Transformation Network for Few-Shot Learning

VSA: Adaptive Visual and Semantic Guided Attention on Few-Shot Learning

Multi-scale Feature Modification Network for FewShot Image Classification

Attribute- and attention-guided few-shot classification