中文

CapsuleGAN:生成对抗胶囊网络

机器学习 2018-10-03 v7 机器学习

摘要

我们提出生成对抗胶囊网络(CapsuleGAN),一种在生成对抗网络(GAN)框架下对图像数据建模时使用胶囊网络(CapsNets)替代标准卷积神经网络(CNNs)作为判别器的框架。我们提供了设计CapsNet判别器以及融合了CapsNet间隔损失的更新后GAN目标函数的指导方针,用于训练CapsuleGAN模型。我们表明,在MNIST和CIFAR-10数据集上,根据生成对抗度量以及半监督图像分类评估,CapsuleGAN在建模图像数据分布上优于卷积-GAN。

关键词

引用

@article{arxiv.1802.06167,
  title  = {CapsuleGAN: Generative Adversarial Capsule Network},
  author = {Ayush Jaiswal and Wael AbdAlmageed and Yue Wu and Premkumar Natarajan},
  journal= {arXiv preprint arXiv:1802.06167},
  year   = {2018}
}

备注

To appear in Proceedings of ECCV Workshop on Brain Driven Computer Vision (BDCV) 2018