中文

VarGrad:一种用于变分推断的低方差梯度估计器

机器学习 2020-10-30 v2 机器学习 统计理论 统计理论

摘要

我们基于带留一控制变量的得分函数方法,分析了变分推断中 ELBO 的一个无偏梯度估计器的性质。我们证明,该梯度估计器可通过一个新的损失获得,该损失定义为精确后验与变分近似之间对数比的方差,我们称之为 log-variance loss\textit{log-variance loss}(对数方差损失)。在某些条件下,对数方差损失的梯度等于(负)ELBO 的梯度。我们从理论上证明,由于其与对数方差损失的联系而被称为 VarGrad\textit{VarGrad} 的该梯度估计器,在某些设定下比得分函数方法具有更低的方差,且留一控制变量系数接近最优系数。我们通过实验证明,在离散 VAE 上,与其他最先进的估计器相比,VarGrad 提供了更有利的方差-计算权衡。

关键词

引用

@article{arxiv.2010.10436,
  title  = {VarGrad: A Low-Variance Gradient Estimator for Variational Inference},
  author = {Lorenz Richter and Ayman Boustati and Nikolas Nüsken and Francisco J. R. Ruiz and Ömer Deniz Akyildiz},
  journal= {arXiv preprint arXiv:2010.10436},
  year   = {2020}
}