用非线性投影改进树切片Wasserstein距离,提升对复杂数据结构的建模能力。
Tree-Sliced Wasserstein Distance with Nonlinear Projection
- 采用非线性投影替代传统线性投影,增强对测度分布拓扑结构的捕捉能力
- 在欧氏空间和球面上构建高效度量,数值实验验证优于现有SW与TSW方法
- 适用于生成模型、自监督学习等场景,特别适合高维复杂数据分布建模
树切片方法作为传统切片Wasserstein(SW)距离的替代方案,将一维直线投影替换为基于树结构的度量空间,并引入分裂机制来投影测度。该方法在保持低计算成本的同时,提升了对积分域拓扑结构的捕捉能力。本文提出一种新型非线性投影框架用于树切片Wasserstein(TSW)距离,以一般投影取代早期版本中的线性投影,同时保证对应Radon变换的单射性并维持度量的良定义性。通过设计合适的投影方式,我们在欧氏空间和球面上构建了高效的测度度量。最后,通过大量数值实验验证了所提度量在欧氏与球面数据集上的有效性。应用涵盖梯度流、自监督学习和生成模型,结果表明本方法显著优于近期的SW与TSW变体。
原文摘要 · Abstract (English)
Tree-Sliced methods have recently emerged as an alternative to the traditional Sliced Wasserstein (SW) distance, replacing one-dimensional lines with tree-based metric spaces and incorporating a splitting mechanism for projecting measures. This approach enhances the ability to capture the topological structures of integration domains in Sliced Optimal Transport while maintaining low computational costs. Building on this foundation, we propose a novel nonlinear projectional framework for the Tree-Sliced Wasserstein (TSW) distance, substituting the linear projections in earlier versions with general projections, while ensuring the injectivity of the associated Radon Transform and preserving the well-definedness of the resulting metric. By designing appropriate projections, we construct efficient metrics for measures on both Euclidean spaces and spheres. Finally, we validate our proposed metric through extensive numerical experiments for Euclidean and spherical datasets. Applications include gradient flows, self-supervised learning, and generative models, where our methods demonstrate significant improvements over recent SW and TSW variants.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。