CapsuleGAN:生成对抗胶囊网络
CapsuleGAN: Generative Adversarial Capsule Network
摘要:
我们提出了生成对抗胶囊网络(Capsule GAN),这是一个框架,它使用胶囊网络(CapsNets)而不是标准卷积神经网络(CNN)作为生成对抗网络(GAN)设置中的鉴别器,同时对图像数据进行建模。我们提供了设计CapsNet鉴别器和更新的GAN目标函数的指南,该目标函数包含CapsNet边际损失,用于训练Capsule GAN模型。我们表明,CapsuleGAN在MNIST和CIFAR-10数据集上建模图像数据分布方面优于卷积GAN,在生成对抗度量和半监督图像分类方面进行了评估。
英文摘要:
We present Generative Adversarial Capsule Network (CapsuleGAN), a framework that uses capsule networks (CapsNets) instead of the standard convolutional neural networks (CNNs) as discriminators within the generative adversarial network (GAN) setting, while modeling image data. We provide guidelines for designing CapsNet discriminators and the updated GAN objective function, which incorporates the CapsNet margin loss, for training CapsuleGAN models. We show that CapsuleGAN outperforms convolutional-GAN at modeling image data distribution on MNIST and CIFAR-10 datasets, evaluated on the generative adversarial metric and at semi-supervised image classification.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。