中文

DVAE#:具有松弛玻尔兹曼先验的离散变分自编码器

机器学习 2018-10-17 v4 机器学习

摘要

玻尔兹曼机是强大的分布,已被证明是变分自编码器(VAEs)中二值潜变量的有效先验。然而,先前训练离散 VAE 的方法使用了证据下界而非更紧的重要性加权界。我们提出两种将玻尔兹曼机松弛为连续分布的方法,以允许使用重要性加权界进行训练。这些松弛基于广义重叠变换和高斯积分技巧。在 MNIST 和 OMNIGLOT 数据集上的实验表明,这些松弛优于先前具有玻尔兹曼先验的离散 VAE。复现这些结果的实现可在 https://github.com/QuadrantAI/dvae 获取。

关键词

引用

@article{arxiv.1805.07445,
  title  = {DVAE#: Discrete Variational Autoencoders with Relaxed Boltzmann Priors},
  author = {Arash Vahdat and Evgeny Andriyash and William G. Macready},
  journal= {arXiv preprint arXiv:1805.07445},
  year   = {2018}
}

备注

Neural Information Processing Systems (NIPS) 2018