HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Learning from Simulated and Unsupervised Images through Adversarial Training

Ashish Shrivastava; Tomas Pfister; Oncel Tuzel; Josh Susskind; Wenda Wang; Russ Webb

Learning from Simulated and Unsupervised Images through Adversarial Training

Abstract

With recent progress in graphics, it has become more tractable to train models on synthetic images, potentially avoiding the need for expensive annotations. However, learning from synthetic images may not achieve the desired performance due to a gap between synthetic and real image distributions. To reduce this gap, we propose Simulated+Unsupervised (S+U) learning, where the task is to learn a model to improve the realism of a simulator's output using unlabeled real data, while preserving the annotation information from the simulator. We develop a method for S+U learning that uses an adversarial network similar to Generative Adversarial Networks (GANs), but with synthetic images as inputs instead of random vectors. We make several key modifications to the standard GAN algorithm to preserve annotations, avoid artifacts, and stabilize training: (i) a 'self-regularization' term, (ii) a local adversarial loss, and (iii) updating the discriminator using a history of refined images. We show that this enables generation of highly realistic images, which we demonstrate both qualitatively and with a user study. We quantitatively evaluate the generated images by training models for gaze estimation and hand pose estimation. We show a significant improvement over using synthetic images, and achieve state-of-the-art results on the MPIIGaze dataset without any labeled real data.

Code Repositories

rickyhan/SimGAN-Captcha
tf
Mentioned in GitHub
adnanalam53/cycleGAN
tf
Mentioned in GitHub
shinseung428/simGAN_NYU_Hand
tf
Mentioned in GitHub
rvorias/uvHolographics
Mentioned in GitHub
mjdietzx/SimGAN
tf
Mentioned in GitHub
ajdillhoff/simgan-pytorch
pytorch
Mentioned in GitHub
AlexHex7/SimGAN_pytorch
pytorch
Mentioned in GitHub
ashkanpakzad/atn
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-to-image-translation-on-cityscapesSimGAN
Class IOU: 0.04
Per-class Accuracy: 10%
Per-pixel Accuracy: 20%
image-to-image-translation-on-cityscapes-1SimGAN
Class IOU: 0.07
Per-class Accuracy: 11%
Per-pixel Accuracy: 47%

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
Learning from Simulated and Unsupervised Images through Adversarial Training | Papers | HyperAI