Apr, 2024

通过熵正则化扩展均场变分推断的理论与计算

TL;DR我们提出了一种新的变分推断方法 $\Xi$- 变分推断 ($\Xi$-VI),通过熵正则化扩展了朴素均值场。$\Xi$-VI 与熵最优输运问题密切相关,并从计算高效的 Sinkhorn 算法中受益。我们展示了 $\Xi$- 变分后验能够有效恢复真实的后验依赖关系,其中依赖关系由正则化参数进行加权。我们分析了参数空间维度对 $\Xi$- 变分近似精度的影响以及它对计算的影响,从而粗略刻画了 $\Xi$-VI 中的统计计算权衡。我们还研究了 $\Xi$-VI 的经验性质,并建立了关于一致性、渐近正态性、高维渐近性和算法稳定性的结果。我们提供了使用该方法实现多项式时间近似推断的充分条件。最后,我们在模拟和真实数据上展示了 $\Xi$-VI 相对于均值场变分推断的实际优势。