StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets

1 Feb 2022  ·  Axel Sauer, Katja Schwarz, Andreas Geiger ·

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.

PDF Abstract

Results from the Paper


 Ranked #1 on Image Generation on CIFAR-10 (NFE metric)

     Get a GitHub badge
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Image Generation CIFAR-10 StyleGAN-XL FID 1.85 # 10
Image Generation CIFAR-10 StyleGAN-XL (DINOv2) NFE 1 # 1
FD 204.60 # 4
Image Generation FFHQ 1024 x 1024 StyleGAN-XL FID 2.02 # 2
Image Generation FFHQ 256 x 256 StyleGAN-XL FID 2.19 # 3
Image Generation FFHQ 256 x 256 StyleGAN-XL (DINOv2) FD 240.07 # 3
Precision 0.77 # 5
Recall 0.43 # 3
Image Generation FFHQ 512 x 512 StyleGAN-XL FID 2.41 # 2
Image Generation ImageNet 128x128 StyleGAN-XL FID 1.81 # 2
Image Generation ImageNet 256x256 StyleGAN-XL FID 2.30 # 12
Image Generation ImageNet 32x32 StyleGAN-XL FID 1.10 # 1
Image Generation ImageNet 512x512 StyleGAN-XL FID 2.40 # 8
Image Generation ImageNet 64x64 StyleGAN-XL Inception Score 4.06 # 9
FID 1.51 # 4
NFE 1 # 1
Image Generation Pokemon 1024x1024 StyleGAN-XL FID 25.47 # 1
Image Generation Pokemon 256x256 StyleGAN-XL FID 23.97 # 1

Methods