Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Triple Generative Adversarial Networks

About

We propose a unified game-theoretical framework to perform classification and conditional image generation given limited supervision. It is formulated as a three-player minimax game consisting of a generator, a classifier and a discriminator, and therefore is referred to as Triple Generative Adversarial Network (Triple-GAN). The generator and the classifier characterize the conditional distributions between images and labels to perform conditional generation and classification, respectively. The discriminator solely focuses on identifying fake image-label pairs. Under a nonparametric assumption, we prove the unique equilibrium of the game is that the distributions characterized by the generator and the classifier converge to the data distribution. As a byproduct of the three-player mechanism, Triple-GAN is flexible to incorporate different semi-supervised classifiers and GAN architectures. We evaluate Triple-GAN in two challenging settings, namely, semi-supervised learning and the extreme low data regime. In both settings, Triple-GAN can achieve excellent classification results and generate meaningful samples in a specific class simultaneously. In particular, using a commonly adopted 13-layer CNN classifier, Triple-GAN outperforms extensive semi-supervised learning methods substantially on more than 10 benchmarks no matter data augmentation is applied or not.

Chongxuan Li, Kun Xu, Jiashuo Liu, Jun Zhu, Bo Zhang• 2019

Related benchmarks

TaskDatasetResultRank
Image ClassificationSVHN 1000 labels (test)
Error Rate3.45
69
Conditional Image GenerationCIFAR10 (test)
Fréchet Inception Distance17.9
66
Image ClassificationSVHN 250 labels
Test Error Rate3.48
61
Image ClassificationCIFAR-10 4,000 labels (test)
Test Error Rate10.01
57
Image ClassificationSVHN 1,000 labels (train)
Error Rate (%)3.96
15
Image ClassificationCIFAR10 4,000 labels (train)
Error Rate12.41
15
Image ClassificationCIFAR-10 1k labels (test)
Test Error Rate15
9
Image ClassificationCIFAR-10 2k labels (test)
Test Error Rate11.87
8
Image ClassificationTiny ImageNet 2,000 labels (test)
Error Rate15.4
7
Image ClassificationSVHN 500 labels (test)
Error Rate3.61
5
Showing 10 of 16 rows

Other info

Code

Follow for update