arXiv:2504.02544v1 宣言类型: cross 摘要: 我们提出了傅里叶截断 Wasserstein (FSW) 嵌入 - 这是一种将多集和 $\mathbb{R}^d$ 上的测度嵌入到欧几里得空间的新型方法。我们提出的嵌入在分布之间近似保留了截断 Wasserstein 距离，从而提供了几何上有意义的表示，更好地捕捉了输入结构。此外，它在测度上是单注的，在多集上是双唇同步的 - 这使其与基于和池或最大池的常见方法相比具有显著优势，后者在证明上不满足双唇同步性，并且在许多情况下甚至不是单注的。这些保证所需的目标维数几乎是最优的，约为 $2 N d$，其中 $N$ 是输入多集的最大大小。 此外，我们证明了不可能以双唇同步的方式将 $\mathbb{R}^d$ 上的分布嵌入到欧几里得空间。因此，我们的嵌入的度量属性，在某种意义上是最好的。 通过数值实验，我们证明了我们的方法能够产生优越的多集表示，从而在实际学习任务中提高了性能。具体来说，我们展示了 (a) 一个简单的 FSW 嵌入与 MLP 的组合在学习（非截断的）Wasserstein 距离时达到了最先进的性能；以及 (b) 用 FSW 嵌入替换最大池化可以显著增强 PointNet 对参数减少的鲁棒性，即使在参数减少 40 倍后，仅出现轻微的性能下降。