Abstract. 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.
Jaiswal et al. propose the CapsuleGAN REF , a framework that uses capsule networks to replace the standard convolutional neural networks as discriminators.
CapsuleGAN: Generative Adversarial Capsule Network
ECCV Workshops
Computer Science Mathematics