Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Relative representations enable zero-shot latent space communication

About

Neural networks embed the geometric structure of a data manifold lying in a high-dimensional space into latent representations. Ideally, the distribution of the data points in the latent space should depend only on the task, the data, the loss, and other architecture-specific constraints. However, factors such as the random weights initialization, training hyperparameters, or other sources of randomness in the training phase may induce incoherent latent spaces that hinder any form of reuse. Nevertheless, we empirically observe that, under the same data and modeling choices, the angles between the encodings within distinct latent spaces do not change. In this work, we propose the latent similarity between each sample and a fixed set of anchors as an alternative data representation, demonstrating that it can enforce the desired invariances without any additional training. We show how neural architectures can leverage these relative representations to guarantee, in practice, invariance to latent isometries and rescalings, effectively enabling latent space communication: from zero-shot model stitching to latent space comparison between diverse settings. We extensively validate the generalization capability of our approach on different datasets, spanning various modalities (images, text, graphs), tasks (e.g., classification, reconstruction) and architectures (e.g., CNNs, GCNs, transformers).

Luca Moschella, Valentino Maiorca, Marco Fumero, Antonio Norelli, Francesco Locatello, Emanuele Rodol\`a• 2022

Related benchmarks

TaskDatasetResultRank
Image ClassificationMNIST
Accuracy50
417
Image ClassificationFashion MNIST
Accuracy63
240
Text ClassificationTREC
Accuracy36
207
Image ClassificationCIFAR10
Accuracy80
137
Text ClassificationAGNews
Accuracy39
119
Text ClassificationDBpedia (DBP)
Accuracy16
110
Image ClassificationImageNet-100--
87
Word Embedding RetrievalWord Embeddings (test)
MRR0.9
80
Text-to-Image RetrievalFlickr30k (1K)
R@126.7
63
Image MatchingCOCO (val)
Matching Accuracy52.5
48
Showing 10 of 19 rows

Other info

Follow for update