atlas

Research map / Label-efficient and robust learning / Self-supervised and contrastive

No Train, all Gain: Self-Supervised Gradients Improve Deep Frozen Representations

NeurIPS 2024 · Label-efficient and robust learning › Self-supervised and contrastive · approach: Invariance-based self-supervised learning · 1 citations (OpenAlex)

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 FUNGI, which concatenates randomly projected self-supervised gradients (KL, DINO, SimCLR) from a frozen transformer's last attention projection with its embedding to improve kNN classification, retrieval, and in-context segmentation without training.

Problem

Frozen pretrained transformer embeddings are used directly for kNN classification and retrieval, but they may miss task-specific information; the paper asks whether self-supervised gradients can enrich these features without any training.

Approach

Compute per-sample gradients of self-supervised losses (KL-to-uniform, DINO, SimCLR) with respect to a late linear layer of a frozen backbone using a randomly initialized projection head, reduce them with binary random projections, L2-normalize, and concatenate with the embedding before PCA to form retrieval features.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add KL, DINO, and SimCLR gradients to embeddingsAugReg IN1K ViT-B/16 · 11-dataset image classification, full datasetmean kNN accuracy: 59.2 -> 63.6Helped
Add KL, DINO, and SimCLR gradients to embeddingsAugReg IN1K ViT-B/16 · 11-dataset image classification, 5-shotmean kNN accuracy: 33.9 -> 36.7Helped
Add gradient featuresDINO ViT-B/16 retrieval-based segmentation · Pascal VOC 2012, memory bank 1024 × 10⁴mIoU: 55.7 -> 67.0Helped
Add gradient featuresDINO ViT-B/16 retrieval-based segmentation · Pascal VOC 2012, data-efficient 1/64mIoU: 39.0 -> 46.8Helped
Add gradient featuresDINO ViT-B/16 retrieval-based segmentation · ADE20K, data-efficient 1/64mIoU: 11.3 -> 13.7Helped
Add KL and SimCLR gradients to embeddingsBERT Base · TREC, full datasettop-1 accuracy: 83.6 -> 86.8Helped
Add SimCLR gradients to KL-gradient featuresBERT Base · TREC, 5-shottop-1 accuracy: 27.6 -> 23.0Hurt
Add KL gradients to embeddingsSSAST · ESC-50, full datasettop-1 accuracy: 42.8 -> 47.0Helped
Add SimCLR gradients to embeddingsSSAST · SpeechCommands V2, full datasettop-1 accuracy: 27.4 -> 25.4Hurt
Use KL gradients instead of DINO or SimCLR gradientsSSAST · ESC-50 and SpeechCommands V2top-1 accuracy: ESC 50: 42.8 -> 47.0; SpeechCommands V2: 27.4 -> 29.9Mixed
Normalize features and use a projection head (ablation)KL gradients on DeIT ViT-B/16 · ImageNet-100top-1 kNN accuracy: 88.3 -> 89.1Helped
Normalize features and use a projection head (ablation)DINO gradients on DeIT ViT-B/16 · ImageNet-100top-1 kNN accuracy: 79.3 -> 90.1Helped

Surprising findings

Limitations

Experimental setup

Datasets: Cars CUB DTD EuroSAT CIFAR-100 CIFAR-10 Oxford Pets Food-101 ImageNet-1K FGVC Aircraft Flowers102 TREC

Benchmarks: Cars CUB DTD EuroSAT CIFAR-100 CIFAR-10 Oxford Pets Food-101 ImageNet-1K FGVC Aircraft Flowers102 TREC

Metrics: top-1 accuracy mIoU mAP mean accuracy

Baselines: DINO embeddings HummingBird end-to-end fine-tuning CLIP embeddings EVA-CLIP embeddings

Builds on: DINO SimCLR Head2Toe retrieval-based in-context scene understanding (Balazevic et al.) binary random projections

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