ViTGAN: Training GANs with Vision Transformers
Authors: Kwonjoon Lee, Huiwen Chang, Lu Jiang, Han Zhang, Zhuowen Tu, Ce Liu
AI-generated Key Points
- Vision Transformers (ViTs) have demonstrated impressive performance in image recognition tasks with fewer vision-specific biases compared to traditional Convolutional Neural Networks (CNNs).
- Kwonjoon Lee and colleagues investigate whether ViTs can also be used for image generation tasks by integrating the ViT architecture into Generative Adversarial Networks (GANs).
- Existing regularization methods for GANs do not interact well with self-attention mechanisms in ViTs, leading to instability issues during training.
- The authors introduce novel regularization techniques specifically designed for training GANs with ViTs named ViTGAN.
- ViTGAN achieves comparable performance to state-of-the-art CNN-based StyleGAN2 on three popular datasets: CIFAR-10, CelebA, and LSUN bedroom.
- Empirical results demonstrate that their method produces high-quality images with sharp details and diverse styles.
- Combining Vision Transformers and Generative Adversarial Networks could be a promising direction for future research in image generation tasks.
Abstract: Recently, Vision Transformers (ViTs) have shown competitive performance on image recognition while requiring less vision-specific inductive biases. In this paper, we investigate if such observation can be extended to image generation. To this end, we integrate the ViT architecture into generative adversarial networks (GANs). We observe that existing regularization methods for GANs interact poorly with self-attention, causing serious instability during training. To resolve this issue, we introduce novel regularization techniques for training GANs with ViTs. Empirically, our approach, named ViTGAN, achieves comparable performance to state-of-the-art CNN-based StyleGAN2 on CIFAR-10, CelebA, and LSUN bedroom datasets.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.