HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Large Scale Adversarial Representation Learning

Jeff Donahue; Karen Simonyan

Large Scale Adversarial Representation Learning

Abstract

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).

Code Repositories

rkorzeniowski/bigbigan-pytorch
pytorch
Mentioned in GitHub
LEGO999/BigBiGAN-TensorFlow2.0
tf
Mentioned in GitHub
LEGO999/BIgBiGAN
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
contrastive-learning-on-imagenet-1kResNet50 (4×)
ImageNet Top-1 Accuracy: 61.3
self-supervised-image-classification-onBigBiGAN (ResNet-50)
Number of Params: 25M
Top 1 Accuracy: 55.4%
Top 5 Accuracy: 77.4%
self-supervised-image-classification-onBigBiGAN (RevNet-50 ×4, BN+CReLU)
Number of Params: 86M
Top 1 Accuracy: 61.3%
Top 5 Accuracy: 81.9%
self-supervised-image-classification-onBigBiGAN (RevNet-50 ×4)
Number of Params: 86M
Top 1 Accuracy: 60.8%
Top 5 Accuracy: 81.4%
self-supervised-image-classification-onBigBiGAN (ResNet-50, BN+CReLU)
Number of Params: 24M
Top 1 Accuracy: 56.6%
Top 5 Accuracy: 78.6%
semi-supervised-image-classification-on-1BigBiGAN (RevNet-50 ×4, BN+CReLU)
Top 5 Accuracy: 55.2%
semi-supervised-image-classification-on-2BigBiGAN (RevNet-50 ×4, BN+CReLU)
Top 5 Accuracy: 78.8%

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