中文

PixelVAE++:基于离散先验的改进 PixelVAE

计算机视觉与模式识别 2019-08-28 v1 机器学习 机器学习

摘要

为自然图像构建强大的生成模型是一项具有挑战性的任务。PixelCNN 模型能很好地捕捉图像中的细节与局部信息,但感受野有限。具有因子化解码器的变分自编码器可轻易捕捉全局信息,但往往无法忠实重建细节。PixelVAE 结合了两种模型的优点,构建了能够学习局部与全局结构的生成模型。本文引入 PixelVAE++,一种具有三类潜变量并以 PixelCNN++ 作为解码器的 VAE。我们提出一种新颖架构,将解码器的一部分复用为编码器。我们在 MNIST 与 Omniglot 等二值数据集上取得了最先进的性能,并在潜变量模型中于 CIFAR-10 上取得最先进性能,同时保持潜变量的信息性。

关键词

引用

@article{arxiv.1908.09948,
  title  = {PixelVAE++: Improved PixelVAE with Discrete Prior},
  author = {Hossein Sadeghi and Evgeny Andriyash and Walter Vinci and Lorenzo Buffoni and Mohammad H. Amin},
  journal= {arXiv preprint arXiv:1908.09948},
  year   = {2019}
}