Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Return of Unconditional Generation: A Self-supervised Representation Generation Method

About

Unconditional generation -- the problem of modeling data distribution without relying on human-annotated labels -- is a long-standing and fundamental challenge in generative models, creating a potential of learning from large-scale unlabeled data. In the literature, the generation quality of an unconditional method has been much worse than that of its conditional counterpart. This gap can be attributed to the lack of semantic information provided by labels. In this work, we show that one can close this gap by generating semantic representations in the representation space produced by a self-supervised encoder. These representations can be used to condition the image generator. This framework, called Representation-Conditioned Generation (RCG), provides an effective solution to the unconditional generation problem without using labels. Through comprehensive experiments, we observe that RCG significantly improves unconditional generation quality: e.g., it achieves a new state-of-the-art FID of 2.15 on ImageNet 256x256, largely reducing the previous best of 5.91 by a relative 64%. Our unconditional results are situated in the same tier as the leading class-conditional ones. We hope these encouraging observations will attract the community's attention to the fundamental problem of unconditional generation. Code is available at https://github.com/LTH14/rcg.

Tianhong Li, Dina Katabi, Kaiming He• 2023

Related benchmarks

TaskDatasetResultRank
Class-conditional Image GenerationImageNet 256x256
Inception Score (IS)215.5
441
Class-conditional Image GenerationImageNet 256x256 (train)
IS300.7
305
Image GenerationImageNet 256x256
FID3.49
243
Class-conditional generationImageNet 256 x 256 1k (val)
FID2.12
67
Class-unconditional image generationImageNet 256x256
FID2.15
25
Unconditional Image GenerationImageNet 256x256 (train)
FID2.15
21
Unconditional Image GenerationImageNet (val)
FID3.44
12
Showing 7 of 7 rows

Other info

Code

Follow for update