arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1906.01437cs.DScs.CCcs.LGstat.COstat.ML
收藏

最优运输问题的熵正则化算法的效率

On the Efficiency of Entropic Regularized Algorithms for Optimal Transport

University of California, Berkeley(加州大学伯克利分校);University of Texas, Austin(德克萨斯大学奥斯汀分校)

Tianyi Lin, Nhat Ho, Michael I. Jordan

摘要:

我们提出了几个新的复杂性结果的熵正则化算法,近似解决最多$n$原子的两个离散概率措施之间的最佳运输(OT)问题。首先,我们改进了Sinkhorn的贪婪变体\textit{Greenkhorn}的复杂性界,从$\widetilde{O}(n^2\varepsilon^{-3})$到$\widetilde{O}(n^2\varepsilon^{-2})$。值得注意的是,我们的结果可以匹配Sinkhorn最知名的复杂性界,并有助于澄清为什么Greenkhorn显着优于Sinkhorn在实践中的行/列更新方面观察到的~ \citet{Altschuler-2017-Near}。其次,我们提出了一个新的算法,我们称之为\textit{APDAMD}和推广的自适应原始-对偶加速梯度下降(APDAGD)算法~ \citep{Dvurechensky-2018-Computational}与预先指定的镜像映射$ϕ$。我们证明了APDAMD达到了$\widetilde{O}(n^2\sqrtδ\varepsilon^{-1})$的复杂性界,其中$δ>0$代表$ϕ$的正则性。另外,通过一个反例证明了以前对APDAGD证明的$\widetilde{O}(\min\{n^{9/4}\varepsilon^{-1}, n^2\varepsilon^{-2}\})$的复杂性界是无效的,并给出了$\widetilde{O}(n^{5/2}\varepsilon^{-1})$的精确复杂性界.进一步,我们通过对估计序列的上诉,提出了Sinkhorn的\textit{deterministic}加速变体,并证明了$\widetilde{O}(n^{7/3}\varepsilon^{-4/3})$的复杂度界.因此,我们看到Sinkhorn的加速变体在$1/\varepsilon$和APDAGD方面优于Sinkhorn和Greenkhorn,在$n$方面优于加速交替最小化(AAM)~ \citep{Guminov-2021-Combination}。最后,我们对合成数据和实际数据进行了实验,数值结果表明了Greenkhorn,APDAMD和加速Sinkhorn在实际中的有效性。

英文摘要:

We present several new complexity results for the entropic regularized algorithms that approximately solve the optimal transport (OT) problem between two discrete probability measures with at most $n$ atoms. First, we improve the complexity bound of a greedy variant of Sinkhorn, known as \textit{Greenkhorn}, from $\widetilde{O}(n^2\varepsilon^{-3})$ to $\widetilde{O}(n^2\varepsilon^{-2})$. Notably, our result can match the best known complexity bound of Sinkhorn and help clarify why Greenkhorn significantly outperforms Sinkhorn in practice in terms of row/column updates as observed by~\citet{Altschuler-2017-Near}. Second, we propose a new algorithm, which we refer to as \textit{APDAMD} and which generalizes an adaptive primal-dual accelerated gradient descent (APDAGD) algorithm~\citep{Dvurechensky-2018-Computational} with a prespecified mirror mapping $ϕ$. We prove that APDAMD achieves the complexity bound of $\widetilde{O}(n^2\sqrtδ\varepsilon^{-1})$ in which $δ>0$ stands for the regularity of $ϕ$. In addition, we show by a counterexample that the complexity bound of $\widetilde{O}(\min\{n^{9/4}\varepsilon^{-1}, n^2\varepsilon^{-2}\})$ proved for APDAGD before is invalid and give a refined complexity bound of $\widetilde{O}(n^{5/2}\varepsilon^{-1})$. Further, we develop a \textit{deterministic} accelerated variant of Sinkhorn via appeal to estimated sequence and prove the complexity bound of $\widetilde{O}(n^{7/3}\varepsilon^{-4/3})$. As such, we see that accelerated variant of Sinkhorn outperforms Sinkhorn and Greenkhorn in terms of $1/\varepsilon$ and APDAGD and accelerated alternating minimization (AAM)~\citep{Guminov-2021-Combination} in terms of $n$. Finally, we conduct the experiments on synthetic and real data and the numerical results show the efficiency of Greenkhorn, APDAMD and accelerated Sinkhorn in practice.

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

↑