摘要
arXiv:2504.02544v2 宣告类型: replace-cross
摘要: 我们提出了傅里叶剪切 Wasserstein (FSW) 插值方法——一种将 $\mathbb{R}^d$ 上的多元集和测度嵌入欧几里得空间的新方法。我们提出的嵌入方法约在分布之间保留剪切 Wasserstein 距离,从而提供了几何上有意义的表示,更好地捕获输入结构。此外,该方法在测度上是单射的,并且在多元集上是双唇距的——与基于和池或最大池的流行方法相比,这是一个显著的优势,这些流行方法在理论上证明不是双唇距的,并且在许多情况下甚至不是单射的。这些保证所需的输出维度接近最优:大约 $2 N d$,其中 $N$ 是输入多元集的最大大小。
此外,我们证明了不可能以双唇距的方式将 $\mathbb{R}^d$ 上的分布嵌入到欧几里得空间。因此,我们的嵌入的度量属性可以说是最好的。
通过数值实验,我们证明了该方法生成了优于现有方法的多元集表示,从而在实际学习任务中提高了性能。具体来说,我们展示了以下两点:(a) 简单组合 FSW 嵌入与多层感知器 (MLP) 能够实现 Wasserstein 距离(非剪切)的最新性能;(b) 使用 FSW 嵌入替换最大池化使得 PointNet 在参数减少至多 40 倍的情况下显著更具鲁棒性,即使在这种情况下性能下降也很小。