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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | SVHN 1000 labels (test) | Error Rate3.45 | 69 | |
| Conditional Image Generation | CIFAR10 (test) | Fréchet Inception Distance17.9 | 66 | |
| Image Classification | SVHN 250 labels | Test Error Rate3.48 | 61 | |
| Image Classification | CIFAR-10 4,000 labels (test) | Test Error Rate10.01 | 57 | |
| Image Classification | SVHN 1,000 labels (train) | Error Rate (%)3.96 | 15 | |
| Image Classification | CIFAR10 4,000 labels (train) | Error Rate12.41 | 15 | |
| Image Classification | CIFAR-10 1k labels (test) | Test Error Rate15 | 9 | |
| Image Classification | CIFAR-10 2k labels (test) | Test Error Rate11.87 | 8 | |
| Image Classification | Tiny ImageNet 2,000 labels (test) | Error Rate15.4 | 7 | |
| Image Classification | SVHN 500 labels (test) | Error Rate3.61 | 5 |