核半隐式变分推断
机器学习
2024-05-30 v1 机器学习
摘要
半隐式变分推断(SIVI)通过层次方式定义的半隐式分布扩展了传统变分族。由于半隐式分布的密度难以处理,经典SIVI通常采用证据下界(ELBO)的替代目标,这会在训练中引入偏差。SIVI的最新进展SIVI-SM利用通过极小极大公式变得可处理的替代分数匹配目标,但需要额外的低层优化。在本文中,我们提出核SIVI(KSIVI),它是SIVI-SM的一个变体,通过核技巧消除了低层优化的需要。具体地,我们表明当在再生核希尔伯特空间(RKHS)上优化时,低层问题具有显式解。这样,上层目标变为核斯坦因散度(KSD),由于半隐式变分分布的层次结构,该散度易于通过随机梯度下降计算。我们推导了KSD目标的蒙特卡洛梯度估计量方差的上界,从而建立了KSIVI的新收敛保证。我们在合成分布和各种真实数据贝叶斯推断任务上展示了KSIVI的有效性和效率。
引用
@article{arxiv.2405.18997,
title = {Kernel Semi-Implicit Variational Inference},
author = {Ziheng Cheng and Longlin Yu and Tianyu Xie and Shiyue Zhang and Cheng Zhang},
journal= {arXiv preprint arXiv:2405.18997},
year = {2024}
}
备注
ICML 2024 camera ready