arXiv:2412.08824cs.LGstat.ML2024-12被引 7

拆解五因素影响,给出高效流变分推断的实用配方

Disentangling impact of capacity, objective, batchsize, estimators, and step-size on flow VI

  • 逐项分析容量、目标函数等五大因素对性能的影响
  • 提出可媲美甚至超越HMC的流变分推断方案
  • 针对常见后验难题设计可精确采样的合成基准

基于归一化流的变分推断(flow VI)是一种有前景的近似推断方法,但其性能在不同研究中表现不一。众多算法选择会影响flow VI的表现。本文通过逐步分析,分离出容量、目标函数、梯度估计器、梯度估计次数(批大小)和步长这五个关键因素的影响。每一步均在控制其他变量的前提下,结合前序洞察与大规模并行计算进行评估。为实现高保真评估,我们构建了一个包含常见后验病态的合成基准,并支持精确采样。研究给出了各因素的具体建议,并提出一个可匹配或超越主流无迹哈密顿蒙特卡洛(HMC)方法的flow VI使用配方。

原文摘要 · Abstract (English)

Normalizing flow-based variational inference (flow VI) is a promising approximate inference approach, but its performance remains inconsistent across studies. Numerous algorithmic choices influence flow VI's performance. We conduct a step-by-step analysis to disentangle the impact of some of the key factors: capacity, objectives, gradient estimators, number of gradient estimates (batchsize), and step-sizes. Each step examines one factor while neutralizing others using insights from the previous steps and/or using extensive parallel computation. To facilitate high-fidelity evaluation, we curate a benchmark of synthetic targets that represent common posterior pathologies and allow for exact sampling. We provide specific recommendations for different factors and propose a flow VI recipe that matches or surpasses leading turnkey Hamiltonian Monte Carlo (HMC) methods.

变分推断流模型贝叶斯推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。