arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.13093cs.LGcs.ITmath.ITstat.ML

f-Divergence Variational Inference

University of Illinois at Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Anker Innovations(安克创新)

Neng Wan, Dapeng Li, Naira Hovakimyan

英文摘要:

This paper introduces the $f$-divergence variational inference ($f$-VI) that generalizes variational inference to all $f$-divergences. Initiated from minimizing a crafty surrogate $f$-divergence that shares the statistical consistency with the $f$-divergence, the $f$-VI framework not only unifies a number of existing VI methods, e.g. Kullback-Leibler VI, Rényi's $α$-VI, and $χ$-VI, but offers a standardized toolkit for VI subject to arbitrary divergences from $f$-divergence family. A general $f$-variational bound is derived and provides a sandwich estimate of marginal likelihood (or evidence). The development of the $f$-VI unfolds with a stochastic optimization scheme that utilizes the reparameterization trick, importance weighting and Monte Carlo approximation; a mean-field approximation scheme that generalizes the well-known coordinate ascent variational inference (CAVI) is also proposed for $f$-VI. Empirical examples, including variational autoencoders and Bayesian neural networks, are provided to demonstrate the effectiveness and the wide applicability of $f$-VI.

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

↑