LGCVFeb 1, 2022

StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets

arXiv:2202.00273v2681 citations
AI Analysis

This addresses the challenge of scaling generative models to diverse datasets for broader applications in computer graphics, though it is incremental as it builds on existing StyleGAN and Projected GAN methods.

The paper tackled the problem of StyleGAN's performance degradation on large, diverse datasets like ImageNet by identifying the training strategy as the main limitation, and introduced StyleGAN-XL, which sets a new state-of-the-art for large-scale image synthesis, generating images at 1024x1024 resolution.

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 Implementations2 repos
Foundations

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

Your Notes