Feb, 2024

图上概率测度的广义Sobolev输运

TL;DR我们研究了基于图度量空间支持的度量问题的最优传输(OT)问题。最近,Le等人(2022)利用图结构提出了OT的一个变种,即Sobolev传输(ST),它可以通过快速计算得到一个闭式表达式。然而,ST的定义本质上与$L^p$几何结构耦合在一起,使得利用ST处理其他先验结构变得非常复杂。与之相反,经典OT通过修改底层成本函数可以灵活适应各种几何结构,其中一个重要的实例是Orlicz-Wasserstein(OW)。与标准的$p$阶Wasserstein相比,OW通过利用Orlicz几何结构推进了某些机器学习方法,但由于其二级优化形式,OW的计算带来了新的挑战。在这项工作中,我们利用Orlicz结构的特定类凸函数提出了广义Sobolev传输(GST),GST包含ST作为其特例,并可以用于超过$L^p$几何的先验结构。与OW相连,我们展示了计算GST只需要简单地解决一个一元优化问题,而不是OW中复杂的二级优化问题。我们进行了实证研究,证明GST的计算速度比OW快几个数量级。此外,我们提供了GST在文档分类和拓扑数据分析中的优势的初步证据。