Large Scale Adversarial Representation Learning

NeurIPS 2019  ·  Jeff Donahue, Karen Simonyan ·

Adversarially trained generative models (GANs) have recently achieved compelling image synthesis results. But despite early successes in using GANs for unsupervised representation learning, they have since been superseded by approaches based on self-supervision. In this work we show that progress in image generation quality translates to substantially improved representation learning performance. Our approach, BigBiGAN, builds upon the state-of-the-art BigGAN model, extending it to representation learning by adding an encoder and modifying the discriminator. We extensively evaluate the representation learning and generation capabilities of these BigBiGAN models, demonstrating that these generation-based models achieve the state of the art in unsupervised representation learning on ImageNet, as well as in unconditional image generation. Pretrained BigBiGAN models -- including image generators and encoders -- are available on TensorFlow Hub (https://tfhub.dev/s?publisher=deepmind&q=bigbigan).

PDF Abstract NeurIPS 2019 PDF NeurIPS 2019 Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Self-Supervised Image Classification ImageNet BigBiGAN (ResNet-50) Top 1 Accuracy 55.4% # 119
Top 5 Accuracy 77.4% # 35
Self-Supervised Image Classification ImageNet BigBiGAN (RevNet-50 ×4, BN+CReLU) Top 1 Accuracy 61.3% # 113
Top 5 Accuracy 81.9% # 31
Number of Params 86M # 35
Self-Supervised Image Classification ImageNet BigBiGAN (ResNet-50, BN+CReLU) Top 1 Accuracy 56.6% # 118
Top 5 Accuracy 78.6% # 33
Self-Supervised Image Classification ImageNet BigBiGAN (RevNet-50 ×4) Top 1 Accuracy 60.8% # 114
Top 5 Accuracy 81.4% # 32
Semi-Supervised Image Classification ImageNet - 10% labeled data BigBiGAN (RevNet-50 ×4, BN+CReLU) Top 5 Accuracy 78.8% # 47
Contrastive Learning imagenet-1k ResNet50 (4×) ImageNet Top-1 Accuracy 61.3 # 10
Semi-Supervised Image Classification ImageNet - 1% labeled data BigBiGAN (RevNet-50 ×4, BN+CReLU) Top 5 Accuracy 55.2% # 32

Methods