Skip to main content
QUICK REVIEW

[论文解读] Versatile Auxiliary Classifier with Generative Adversarial Network (VAC+GAN), Multi Class Scenarios

Shabab Bazrafkan, Peter Corcoran|arXiv (Cornell University)|Jun 19, 2018
Anomaly Detection Techniques and Applications参考文献 11被引用 11
一句话总结

该论文提出VAC+GAN,一种用于在多类别场景中训练条件深度生成器的多功能框架,通过在判别器中并行插入分类器并反向传播分类损失至生成器,实现端到端训练。该方法使任意GAN变体均可生成类别特定样本,并提升类别分离度,在CIFAR-10上达到74.49%的分类准确率——优于ACGAN的71.89%——同时保持GAN架构与分类器设计的灵活性。

ABSTRACT

Conditional generators learn the data distribution for each class in a multi-class scenario and generate samples for a specific class given the right input from the latent space. In this work, a method known as "Versatile Auxiliary Classifier with Generative Adversarial Network" for multi-class scenarios is presented. In this technique, the Generative Adversarial Networks (GAN)'s generator is turned into a conditional generator by placing a multi-class classifier in parallel with the discriminator network and backpropagate the classification error through the generator. This technique is versatile enough to be applied to any GAN implementation. The results on two databases and comparisons with other method are provided as well.

研究动机与目标

  • 开发一种灵活的方法,用于训练多类别深度生成模型中的条件生成器。
  • 克服现有方法(如ACGAN)中分类器与判别器耦合所带来的架构与损失函数限制。
  • 通过将分类器与判别器解耦,使任意GAN变体均能支持条件生成。
  • 通过最大化类别分布之间的Jensen-Shannon散度,提升生成样本的类别间区分度。
  • 在多种数据集和GAN架构上通过定量比较验证该方法的有效性。

提出的方法

  • 在GAN框架中,于判别器网络旁并行添加一个多类别分类器。
  • 使用生成样本的分类交叉熵损失对分类器进行训练。
  • 将分类误差通过生成器反向传播,实现条件生成的端到端训练。
  • 生成器通过同时优化GAN损失与分类器损失,学习生成类别特定样本。
  • 该方法与GAN原始损失函数和架构解耦,兼容任意GAN变体。
  • 理论分析证明,该方法可增加类别分布之间的Jensen-Shannon散度,从而增强类别可分性。

实验结果

研究问题

  • RQ1能否开发一种通用的、与架构无关的方法,使任意GAN框架支持条件生成?
  • RQ2与ACGAN相比,所提出的VAC+GAN框架在生成样本的类别间区分度方面有何改进?
  • RQ3将分类器与判别器解耦在多大程度上提升了训练灵活性与性能?
  • RQ4VAC+GAN方法在CIFAR-10等多类别基准数据集上是否保持或提升了生成质量与分类准确率?
  • RQ5该方法能否有效应用于类别数量庞大且图像分辨率较高的数据集?

主要发现

  • 在200个训练周期后,VAC+GAN在CIFAR-10数据集上达到74.49%的分类准确率,优于ACGAN的71.89%。
  • VAC+GAN的混淆矩阵显示其在各类别上的预测性能优于ACGAN,表明类别间分离度得到改善。
  • 该方法成功增加了类别分布之间的Jensen-Shannon散度,证实了生成样本中类别独特性的增强。
  • VAC+GAN兼容任意GAN架构与损失函数,因分类器独立于判别器的结构与损失函数运行。
  • 该框架允许灵活选择分类器架构,而ACGAN则受限于判别器的设计。
  • 该方法通过将分类损失反向传播至生成器,实现了条件生成器的端到端训练,确保生成样本具备类别特异性。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。