arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.09016cs.DSmath.STstat.MLstat.TH
收藏

等周约束下的快速并行采样

Fast parallel sampling under isoperimetry

Nima Anari, Sinho Chewi, Thuy-Duong Vuong

AI总结:

本文提出基于 Langevin 算法的并行采样方法,在等周条件下以近最优梯度复杂度获得 KL 或 TV 距离保证,并由此得到有向欧拉回路和非对称行列式点过程的 RNC 采样器。

摘要:

我们展示了如何通过并行化 Langevin(相应地,欠阻尼 Langevin)算法,从满足对数 Sobolev 不等式且具有光滑对数密度的 $\mathbb R^d$ 上的分布 $π$ 中进行并行采样。我们证明,我们的算法输出的样本来自一个在 Kullback–Leibler(KL)散度(相应地,全变差(TV)距离)意义上与 $π$ 接近的分布 $\hatπ$,同时仅使用 $\log(d)^{O(1)}$ 轮并行和总计 $\widetilde{O}(d)$(相应地 $\widetilde O(\sqrt d)$)次梯度评估。这构成了首个具有 TV 距离保证的并行采样算法。对于我们的主要应用,我们展示了如何将我们算法的 TV 距离保证与先前工作相结合,从而为超立方体 $\{\pm 1\}^n$ 上在指数倾斜下封闭且具有有界协方差的离散分布族获得 RNC 采样到计数的归约。因此,我们获得了用于有向欧拉回路和非对称行列式点过程的 RNC 采样器,解决了先前工作中提出的开放问题。

英文摘要:

We show how to sample in parallel from a distribution $π$ over $\mathbb R^d$ that satisfies a log-Sobolev inequality and has a smooth log-density, by parallelizing the Langevin (resp. underdamped Langevin) algorithms. We show that our algorithm outputs samples from a distribution $\hatπ$ that is close to $π$ in Kullback--Leibler (KL) divergence (resp. total variation (TV) distance), while using only $\log(d)^{O(1)}$ parallel rounds and $\widetilde{O}(d)$ (resp. $\widetilde O(\sqrt d)$) gradient evaluations in total. This constitutes the first parallel sampling algorithms with TV distance guarantees. For our main application, we show how to combine the TV distance guarantees of our algorithms with prior works and obtain RNC sampling-to-counting reductions for families of discrete distribution on the hypercube $\{\pm 1\}^n$ that are closed under exponential tilts and have bounded covariance. Consequently, we obtain an RNC sampler for directed Eulerian tours and asymmetric determinantal point processes, resolving open questions raised in prior works.

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

↑