CVJun 9, 2021

Generative Models as a Data Source for Multiview Representation Learning

arXiv:2106.05258v3150 citationsHas Code
Originality Incremental advance
AI Analysis

This addresses the challenge of data scarcity, privacy, or unwieldiness in machine learning by enabling representation learning from generative models, though it is incremental as it builds on existing contrastive methods.

The paper tackles the problem of learning visual representations from generative models instead of real datasets, showing that representations trained on samples from an off-the-shelf image generator can rival or outperform those learned directly from real data, with performance depending on sampling and training strategies.

Generative models are now capable of producing highly realistic images that look nearly indistinguishable from the data on which they are trained. This raises the question: if we have good enough generative models, do we still need datasets? We investigate this question in the setting of learning general-purpose visual representations from a black-box generative model rather than directly from data. Given an off-the-shelf image generator without any access to its training data, we train representations from the samples output by this generator. We compare several representation learning methods that can be applied to this setting, using the latent space of the generator to generate multiple "views" of the same semantic content. We show that for contrastive methods, this multiview data can naturally be used to identify positive pairs (nearby in latent space) and negative pairs (far apart in latent space). We find that the resulting representations rival or even outperform those learned directly from real data, but that good performance requires care in the sampling strategy applied and the training method. Generative models can be viewed as a compressed and organized copy of a dataset, and we envision a future where more and more "model zoos" proliferate while datasets become increasingly unwieldy, missing, or private. This paper suggests several techniques for dealing with visual representation learning in such a future. Code is available on our project page https://ali-design.github.io/GenRep/.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes