3 个月前

三重生成对抗网络

三重生成对抗网络

摘要

我们提出了一种统一的博弈论框架,用于在有限监督条件下实现分类与条件图像生成。该框架被建模为一个三玩家极小极大博弈,包含生成器、分类器和判别器,因此被称为三重生成对抗网络(Triple-GAN)。其中,生成器和分类器分别用于刻画图像与标签之间的条件分布,以实现条件生成与分类任务;而判别器则专注于识别虚假的图像-标签配对。在非参数假设下,我们证明了该博弈的唯一均衡状态为:生成器与分类器所刻画的分布均收敛至真实数据分布。作为三玩家机制的副产品,Triple-GAN具有良好的灵活性,可方便地融合多种半监督分类器与GAN架构。我们在两个具有挑战性的场景下对Triple-GAN进行了评估,即半监督学习与极端低数据量情形。在两种设置下,Triple-GAN均能实现优异的分类性能,并同时生成特定类别下具有语义意义的样本。特别地,在采用广泛使用的13层CNN分类器时,Triple-GAN在超过10个基准数据集上显著优于众多现有的半监督学习方法,无论是否使用数据增强技术。

代码仓库

taufikxu/Triple-GAN
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
semi-supervised-image-classification-on-cifarTriple-GAN-V2 (CNN-13)
Percentage error: 10.01
semi-supervised-image-classification-on-cifarTriple-GAN-V2 (CNN-13, no aug)
Percentage error: 12.41
semi-supervised-image-classification-on-cifarTriple-GAN-V2 (ResNet-26)
Percentage error: 6.54
semi-supervised-image-classification-on-cifar-11Triple-GAN-V2 (ResNet-26)
Accuracy: 91.59
semi-supervised-image-classification-on-cifar-11Triple-GAN-V2 (CNN-13, no aug)
Accuracy: 81.81
semi-supervised-image-classification-on-cifar-11Triple-GAN-V2 (CNN-13)
Accuracy: 85.00
semi-supervised-image-classification-on-svhnTriple-GAN-V2 (CNN-13, no aug)
Accuracy: 96.04
semi-supervised-image-classification-on-svhnTriple-GAN-V2 (CNN-13)
Accuracy: 96.55
semi-supervised-image-classification-on-svhn-1Triple-GAN-V2 (CNN-13)
Accuracy: 96.52
semi-supervised-image-classification-on-svhn-1Triple-GAN-V2 (CNN-13, no aug)
Accuracy: 95.81
semi-supervised-image-classification-on-svhn-3Triple-GAN-V2 (CNN-13, no aug)
Accuracy: 96.16
semi-supervised-image-classification-on-svhn-3Triple-GAN-V2 (CNN-13)
Accuracy: 96.39

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
三重生成对抗网络 | 论文 | HyperAI超神经