arXiv:2503.11249cs.LGcs.AI2025-03ICLR被引 10

提出球面上的树切片水石距离,高效度量球面分布间差异。

Spherical Tree-Sliced Wasserstein Distance

  • 用球面树结构替代传统切片线,捕捉球面拓扑信息。
  • 推导出球面水石距离的闭式解,计算效率高。
  • 适合几何深度学习、球面数据建模等场景使用。

切片最优传输(Sliced Optimal Transport, SOT)通过将高维分布投影到一维直线并利用一维最优传输的闭式解,显著降低计算负担。近期提出的树切片方法用更复杂的树结构替代直线,增强了对积分域拓扑信息的捕捉能力,同时保持低计算成本。受此启发,本文将树结构推广至球面支持的分布上,提出一种新的球面径向变换(spherical Radon transform),其积分域为球面树结构。借助该变换和球面树结构,我们推导出球面上最优传输问题的闭式表达式,从而构建了一种高效的球面测度度量——球面树切片水石距离(Spherical Tree-Sliced Wasserstein, STSW)。本文提供了全面的理论分析,证明了球面树的拓扑性质及变换的良定义性和单射性,进而获得正交不变的球面测度距离。最后,我们在梯度流和自监督学习等多个任务中进行广泛数值实验,验证了所提度量优于现有基准的表现。

原文摘要 · Abstract (English)

Sliced Optimal Transport (OT) simplifies the OT problem in high-dimensional spaces by projecting supports of input measures onto one-dimensional lines and then exploiting the closed-form expression of the univariate OT to reduce the computational burden of OT. Recently, the Tree-Sliced method has been introduced to replace these lines with more intricate structures, known as tree systems. This approach enhances the ability to capture topological information of integration domains in Sliced OT while maintaining low computational cost. Inspired by this approach, in this paper, we present an adaptation of tree systems on OT problems for measures supported on a sphere. As a counterpart to the Radon transform variant on tree systems, we propose a novel spherical Radon transform with a new integration domain called spherical trees. By leveraging this transform and exploiting the spherical tree structures, we derive closed-form expressions for OT problems on the sphere. Consequently, we obtain an efficient metric for measures on the sphere, named Spherical Tree-Sliced Wasserstein (STSW) distance. We provide an extensive theoretical analysis to demonstrate the topology of spherical trees and the well-definedness and injectivity of our Radon transform variant, which leads to an orthogonally invariant distance between spherical measures. Finally, we conduct a wide range of numerical experiments, including gradient flows and self-supervised learning, to assess the performance of our proposed metric, comparing it to recent benchmarks.

最优传输球面数据度量学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。