HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

MSG-GAN: Multi-Scale Gradients for Generative Adversarial Networks

Animesh Karnewar; Oliver Wang

MSG-GAN: Multi-Scale Gradients for Generative Adversarial Networks

Abstract

While Generative Adversarial Networks (GANs) have seen huge successes in image synthesis tasks, they are notoriously difficult to adapt to different datasets, in part due to instability during training and sensitivity to hyperparameters. One commonly accepted reason for this instability is that gradients passing from the discriminator to the generator become uninformative when there isn't enough overlap in the supports of the real and fake distributions. In this work, we propose the Multi-Scale Gradient Generative Adversarial Network (MSG-GAN), a simple but effective technique for addressing this by allowing the flow of gradients from the discriminator to the generator at multiple scales. This technique provides a stable approach for high resolution image synthesis, and serves as an alternative to the commonly used progressive growing technique. We show that MSG-GAN converges stably on a variety of image datasets of different sizes, resolutions and domains, as well as different types of loss functions and architectures, all with the same set of fixed hyperparameters. When compared to state-of-the-art GANs, our approach matches or exceeds the performance in most of the cases we tried.

Code Repositories

akanimax/msg-stylegan-tf
Official
tf
Mentioned in GitHub
denizalperacar/MSG-GAN
pytorch
Mentioned in GitHub
akanimax/BMSG-GAN
pytorch
Mentioned in GitHub
Iceland-Leo/Style-MSG-GAN
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-generation-on-celeba-hq-1024x1024MSG-StyleGAN
FID: 6.37
image-generation-on-ffhqMSG-StyleGAN
Clean-FID (70k): 6.51
image-generation-on-ffhq-1024-x-1024MSG-StyleGAN
FID: 5.8
image-generation-on-indian-celebs-256-x-256MSG-StyleGAN
FID: 28.44
image-generation-on-lsun-churches-256-x-256MSG-StyleGAN
Clean-FID (trainfull): 5.38 ± 0.03
FID: 5.2
image-generation-on-oxford-102-flowers-256-xMSG-StyleGAN
FID: 19.60

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
MSG-GAN: Multi-Scale Gradients for Generative Adversarial Networks | Papers | HyperAI