HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Let Go of Your Labels with Unsupervised Transfer

Gadetsky Artyom ; Jiang Yulun ; Brbic Maria

Let Go of Your Labels with Unsupervised Transfer

Abstract

Foundation vision-language models have enabled remarkable zero-shottransferability of the pre-trained representations to a wide range ofdownstream tasks. However, to solve a new task, zero-shot transfer stillnecessitates human guidance to define visual categories that appear in thedata. Here, we show that fully unsupervised transfer emerges when searching forthe labeling of a dataset that induces maximal margin classifiers inrepresentation spaces of different foundation models. We present TURTLE, afully unsupervised method that effectively employs this guiding principle touncover the underlying labeling of a downstream dataset without any supervisionand task-specific representation learning. We evaluate TURTLE on a diversebenchmark suite of 26 datasets and show that it achieves new state-of-the-artunsupervised performance. Furthermore, TURTLE, although being fullyunsupervised, outperforms zero-shot transfer baselines on a wide range ofdatasets. In particular, TURTLE matches the average performance of CLIPzero-shot on 26 datasets by employing the same representation space, spanning awide range of architectures and model sizes. By guiding the search for theunderlying labeling using the representation spaces of two foundation models,TURTLE surpasses zero-shot transfer and unsupervised prompt tuning baselines,demonstrating the surprising power and effectiveness of unsupervised transfer.

Code Repositories

mlbio-epfl/turtle
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
image-clustering-on-birdsnapTURTLE (CLIP + DINOv2)
Accuracy: 68.1
image-clustering-on-caltech-101TURTLE (CLIP + DINOv2)
Accuracy: 89.8
image-clustering-on-cifar-10TURTLE (CLIP + DINOv2)
ARI: 0.989
Accuracy: 0.995
NMI: 0.985
image-clustering-on-cifar-100TURTLE (CLIP + DINOv2)
ARI: 0.834
Accuracy: 0.898
NMI: 0.915
image-clustering-on-clevr-countsTURTLE (CLIP + DINOv2)
Accuracy: 24.0
image-clustering-on-country211TURTLE (CLIP + DINOv2)
Accuracy: 11.1
image-clustering-on-dtdTURTLE (CLIP + DINOv2)
Accuracy: 57.3
image-clustering-on-eurosatTURTLE (CLIP + DINOv2)
Accuracy: 96.6
image-clustering-on-fer2013TURTLE (CLIP + DINOv2)
Accuracy: 36.2
image-clustering-on-fgvc-aircraftTURTLE (CLIP + DINOv2)
Accuracy: 36.5
image-clustering-on-flowers-102TURTLE (CLIP + DINOv2)
Accuracy: 99.6
image-clustering-on-food-101TURTLE (CLIP + DINOv2)
Accuracy: 92.2
image-clustering-on-gtsrbTURTLE (CLIP + DINOv2)
Accuracy: 48.4
image-clustering-on-hateful-memesTURTLE (CLIP + DINOv2)
Accuracy: 54.2
image-clustering-on-imagenetTURTLE (CLIP + DINOv2)
ARI: 62.5
Accuracy: 72.9
NMI: 88.2
image-clustering-on-kinetics-700TURTLE (CLIP + DINOv2)
Accuracy: 43.0
image-clustering-on-kittiTURTLE (CLIP + DINOv2)
Accuracy: 39.4
image-clustering-on-mnistTURTLE (CLIP + DINOv2)
Accuracy: 97.8
image-clustering-on-oxford-iiit-petsTURTLE (CLIP + DINOv2)
Accuracy: 92.3
image-clustering-on-pcamTURTLE (CLIP + DINOv2)
Accuracy: 52.0
image-clustering-on-rendered-sst2TURTLE (CLIP + DINOv2)
Accuracy: 51.6
image-clustering-on-resisc45TURTLE (CLIP + DINOv2)
Accuracy: 89.6
image-clustering-on-stanford-carsTURTLE (CLIP + DINOv2)
Accuracy: 0.646
image-clustering-on-stl-10TURTLE (CLIP + DINOv2)
ARI: 0.994
Accuracy: 0.997
NMI: 0.993
image-clustering-on-sun397TURTLE (CLIP + DINOv2)
Accuracy: 67.9
image-clustering-on-ucf101TURTLE (CLIP + DINOv2)
Accuracy: 82.3
unsupervised-image-classification-on-cifar-10TURTLE (CLIP + DINOv2)
Accuracy: 99.5
unsupervised-image-classification-on-imagenetTURTLE (CLIP + DINOv2)
ARI: 62.5
Accuracy (%): 72.9
unsupervised-image-classification-on-mnistTURTLE (CLIP + DINOv2)
Accuracy: 97.8
unsupervised-image-classification-on-stl-10TURTLE (CLIP + DINOv2)
Accuracy: 99.7

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Let Go of Your Labels with Unsupervised Transfer | Papers | HyperAI