arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1807.09356cs.LGstat.ML
收藏

迭代摊销推理

Iterative Amortized Inference

California Institute of Technology(加州理工学院);Disney Research(迪士尼研究院)

Joseph Marino, Yisong Yue, Stephan Mandt

摘要:

推理模型是将变分推理扩展到深度潜在变量模型的关键组件,最值得注意的是变分自动编码器(VAE)中的编码器网络。通过用学习模型替换传统的基于优化的推理,推理在数据示例上被摊销,因此计算效率更高。然而,标准的推理模型仅限于直接从数据映射到近似的后验估计。这些模型未能达到完全优化的近似后验估计,导致摊销缺口。我们的目标是通过提出迭代推理模型来缩小这一差距,该模型通过重复编码梯度来学习执行推理优化。我们的方法概括了VAE的标准推理模型,并提供了一些实证研究结果,包括自上而下的推理技术。我们展示了迭代推理模型的推理优化能力,并表明它们在几个基准数据集的图像和文本上优于标准推理模型。

英文摘要:

Inference models are a key component in scaling variational inference to deep latent variable models, most notably as encoder networks in variational auto-encoders (VAEs). By replacing conventional optimization-based inference with a learned model, inference is amortized over data examples and therefore more computationally efficient. However, standard inference models are restricted to direct mappings from data to approximate posterior estimates. The failure of these models to reach fully optimized approximate posterior estimates results in an amortization gap. We aim toward closing this gap by proposing iterative inference models, which learn to perform inference optimization through repeatedly encoding gradients. Our approach generalizes standard inference models in VAEs and provides insight into several empirical findings, including top-down inference techniques. We demonstrate the inference optimization capabilities of iterative inference models and show that they outperform standard inference models on several benchmark data sets of images and text.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑