4 个月前

双辅助分类器生成对抗网络

双辅助分类器生成对抗网络

摘要

条件生成模型在过去几年中取得了显著进展。其中一种流行的条件模型是辅助分类器生成对抗网络(AC-GAN),它通过扩展生成对抗网络(GAN)的损失函数引入一个辅助分类器,从而生成具有高辨别性的图像。然而,随着类别数量的增加,AC-GAN生成样本的多样性往往会降低,这限制了其在大规模数据上的应用能力。本文从理论上识别了低多样性问题的根源,并提出了一种实用的解决方案来解决该问题。我们指出,AC-GAN中的辅助分类器强加了完美的可分性,当类分布的支持集有显著重叠时,这种可分性是有害的。为了解决这一问题,我们提出了双辅助分类器生成对抗网络(TAC-GAN),该网络进一步引入了一个新的参与者,与GAN中的其他参与者(生成器和判别器)进行交互。理论上,我们证明了TAC-GAN可以有效地最小化生成数据分布与真实数据分布之间的差异。广泛的实验结果表明,我们的TAC-GAN能够在模拟数据上成功复制真实的数据分布,并且在实际数据集上显著提高了类条件图像生成的多样性。

代码仓库

Ram81/AC-VAEGAN-PyTorch
pytorch
GitHub 中提及
pranavbudhwant/ACVAEGAN
pytorch
GitHub 中提及
batmanlab/twin_ac
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
conditional-image-generation-on-cifar-100TAC-GAN
FID: 7.22
Inception Score: 9.34

用 AI 构建 AI

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

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

Hyper Newsletters

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