atlas

Research map / Architectures and efficiency / Attention and transformers

One Meta-tuned Transformer is What You Need for Few-shot Learning

ICML 2024 · Architectures and efficiency › Attention and transformers · approach: Efficient and robust vision transformers

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes MetaFormer, a ViT-based meta-tuned transformer for few-shot image classification that adds masked sample attention and patch-grained task attention to model sample and task relations, supporting both inductive and transductive protocols.

Problem

Few-shot image classification with pre-trained vision transformers; existing meta-learning methods model sample, patch, and task relationships separately with CNNs and task embeddings that carry background noise.

Approach

Builds a holistic attention transformer where self-attention spans patches, samples (Masked Sample Attention with a causal support/query mask and label infusion), and tasks (Patch-grained Task Attention with a dynamic knowledge pool), decoupling patch and sample attention for efficiency.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add PTA to MSA (ablation)MetaFormer-I · miniImageNet 1-shotaccuracy: 74.64 ± 0.76 -> 75.78 ± 0.71Helped
Add PTA to MSA (ablation)MetaFormer-I · miniImageNet 5-shotaccuracy: 87.53 ± 0.47 -> 90.02 ± 0.44Helped
Add MSA to PTA (ablation)MetaFormer-I · miniImageNet 1-shotaccuracy: 73.63 ± 0.75 -> 75.78 ± 0.71Helped
Add MSA to PTA (ablation)MetaFormer-I · miniImageNet 5-shotaccuracy: 87.76 ± 0.52 -> 90.02 ± 0.44Helped
Use decoupled instead of joint sample-patch attention (ablation)MetaFormer · miniImageNet 1-shotaccuracy: 73.35 ± 0.75 -> 73.21 ± 0.76No clear effect
Add temperature scaling to cosine distance (ablation)MetaFormer · miniImageNet 1-shotaccuracy: 64.22 ± 0.83 -> 74.14 ± 0.73Helped
Add patch-wise distance to temperature-scaled cosine distance (ablation)MetaFormer · miniImageNet 1-shotaccuracy: 74.14 ± 0.73 -> 75.78 ± 0.71Helped
Use patch-wise Mahalanobis distance instead of cosine distance (ablation)MetaFormer · miniImageNet 1-shotaccuracy: 75.78 ± 0.71 vs 71.97 ± 0.76Hurt
Add class labels by summation (ablation)MetaFormer · miniImageNet 1-shotaccuracy: 73.70 -> 74.64Helped
Add class labels by concatenation (ablation)MetaFormer · miniImageNet 1-shotaccuracy: 73.70 -> 74.59Helped
Reshuffle query sample order at inference (ablation)MetaFormer-A · CIFAR-FS 1-shotaccuracy: 85.76 ± 0.77 -> 88.34 ± 0.76Helped
Use MetaFormer-I instead of FewTUREViT-Small few-shot learning · miniImageNet 1-shotaccuracy: 68.02±0.88 -> 75.78±0.71Helped

Surprising findings

Limitations

Experimental setup

Datasets: miniImageNet tieredImageNet CIFAR-FS FC-100 Meta-Dataset

Benchmarks: miniImageNet tieredImageNet CIFAR-FS FC-100 CUB Cars Places Plantae CropDisease EuroSAT ISIC ChestX

Metrics: 5-way 1-shot accuracy 5-way 5-shot accuracy GFLOPs inference time

Baselines: FewTURE PMF SMKD HCTransformers protoLP EASY FEAT DeepEMD Tip-Adapter Zero-shot CLIP

Builds on: Vision Transformer DINO self-supervised pre-training FewTURE PMF Divided space-time attention

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy