摘要
arXiv:2505.00968v1 类别:交叉学科
摘要:树切片方法 recently emerged 作为传统切片 Wasserstein (SW) 距离的替代品,用基于树的度量空间替代了一维线条,并引入分裂机制对测度进行投影。这种方法增强了在切片最优传输中捕获积分域拓扑结构的能力,同时保持了较低的计算成本。在这一基础上,我们提出了一种新的非线性投影框架,用于树切片 Wasserstein (TSW) 距离,用通用投影取代了早期版本中的线性投影,同时确保关联的拉东变换的单射性,并保持结果度量的正确定义。通过设计适当的投影,我们为欧几里得空间和球体上的测度构建了高效的度量。最后,我们通过广泛的数值实验验证了我们提出的度量,这些实验包括欧几里得和球体数据集。应用包括梯度流、自监督学习和生成模型,其中我们的方法在最近的 SW 和 TSW 变体中表现出显著的改进。