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

StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets

About

Computer graphics has experienced a recent surge of data-centric approaches for photorealistic and controllable content creation. StyleGAN in particular sets new standards for generative modeling regarding image quality and controllability. However, StyleGAN's performance severely degrades on large unstructured datasets such as ImageNet. StyleGAN was designed for controllability; hence, prior works suspect its restrictive design to be unsuitable for diverse datasets. In contrast, we find the main limiting factor to be the current training strategy. Following the recently introduced Projected GAN paradigm, we leverage powerful neural network priors and a progressive growing strategy to successfully train the latest StyleGAN3 generator on ImageNet. Our final model, StyleGAN-XL, sets a new state-of-the-art on large-scale image synthesis and is the first to generate images at a resolution of $1024^2$ at such a dataset scale. We demonstrate that this model can invert and edit images beyond the narrow domain of portraits or specific object classes.

Axel Sauer, Katja Schwarz, Andreas Geiger• 2022

Related benchmarks

TaskDatasetResultRank
Image GenerationCIFAR-10 (test)
FID1.85
471
Class-conditional Image GenerationImageNet 256x256
Inception Score (IS)265.1
441
Image GenerationImageNet 256x256 (val)
FID2.3
307
Class-conditional Image GenerationImageNet 256x256 (train)
IS265.1
305
Class-conditional Image GenerationImageNet 256x256 (val)
FID2.3
293
Image GenerationImageNet 256x256
FID2.3
243
Image GenerationImageNet 512x512 (val)
FID-50K2.4
184
Class-conditional Image GenerationImageNet 256x256 (train val)
FID2.3
178
Unconditional Image GenerationCIFAR-10
FID1.85
171
Class-conditional Image GenerationImageNet 256x256 (test)
FID2.3
167
Showing 10 of 54 rows

Other info

Follow for update