HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets

Axel Sauer Katja Schwarz Andreas Geiger

StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets

Abstract

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.

Code Repositories

autonomousvision/stylegan-xl
pytorch
Mentioned in GitHub
autonomousvision/stylegan_xl
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-generation-on-ffhq-1024-x-1024StyleGAN-XL
FID: 2.02
image-generation-on-ffhq-256-x-256StyleGAN-XL (DINOv2)
FD: 240.07
Precision: 0.77
Recall: 0.43
image-generation-on-ffhq-256-x-256StyleGAN-XL
FID: 2.19
image-generation-on-ffhq-512-x-512StyleGAN-XL
FID: 2.41
image-generation-on-imagenet-128x128StyleGAN-XL
FID: 1.81
image-generation-on-imagenet-256x256StyleGAN-XL
FID: 2.30
image-generation-on-imagenet-32x32StyleGAN-XL
FID: 1.10
image-generation-on-imagenet-512x512StyleGAN-XL
FID: 2.40
image-generation-on-imagenet-64x64StyleGAN-XL
FID: 1.51
NFE: 1
image-generation-on-pokemon-1024x1024StyleGAN-XL
FID: 25.47
image-generation-on-pokemon-256x256StyleGAN-XL
FID: 23.97

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp