HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Self-Supervised GANs via Auxiliary Rotation Loss

Ting Chen; Xiaohua Zhai; Marvin Ritter; Mario Lucic; Neil Houlsby

Self-Supervised GANs via Auxiliary Rotation Loss

Abstract

Conditional GANs are at the forefront of natural image synthesis. The main drawback of such models is the necessity for labeled data. In this work we exploit two popular unsupervised learning techniques, adversarial training and self-supervision, and take a step towards bridging the gap between conditional and unconditional GANs. In particular, we allow the networks to collaborate on the task of representation learning, while being adversarial with respect to the classic GAN game. The role of self-supervision is to encourage the discriminator to learn meaningful feature representations which are not forgotten during training. We test empirically both the quality of the learned image representations, and the quality of the synthesized images. Under the same conditions, the self-supervised GAN attains a similar performance to state-of-the-art conditional counterparts. Finally, we show that this approach to fully unsupervised learning can be scaled to attain an FID of 23.4 on unconditional ImageNet generation.

Code Repositories

vandit15/Self-Supervised-Gans-Pytorch
pytorch
Mentioned in GitHub
google/compare_gan
Official
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-generation-on-celeba-hq-128x128SS-GAN (sBN)
FID: 24.36
image-generation-on-imagenet-128x128SS-GAN (sBN)
FID: 43.87
image-generation-on-lsun-bedroom-256-x-256SS-GAN (sBN)
FID: 13.3

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