$\lambda$-geodesic convexity of the squared sliced-Wasserstein distance

技术摘要：平方切片 Wasserstein 距离的 $\lambda$-测地凸性

问题陈述
本文探讨了最优传输理论中的一个基本局限性：对于固定的目标测度 $\nu$，平方 Wasserstein 距离 $W_2^2(\cdot, \nu)$ 在维度 $d \geq 2$ 时，在概率测度空间 $\mathcal{P}_2(\mathbb{R}^d)$ 上通常不是测地凸的。虽然该函数在维度 $d=1$ 时是 2-测地凸的，但在高维情况下缺乏凸性，这使得梯度流的分析和解的唯一性变得复杂。

为了缓解高维 Wasserstein 距离的计算成本，研究者引入了切片 Wasserstein 距离（$SW_2$）。它通过对测度在直线上的所有投影的一维 Wasserstein 距离进行平均来定义。本研究的核心问题在于：在具有有界密度的测度集合上，平方切片 Wasserstein 距离 $F_\nu(\rho) = \frac{1}{2}SW_2^2(\rho, \nu)$ 是否具有类似于一维情况下的 $\lambda$-测地凸性性质。

方法论
作者结合了最优传输理论、几何分析和泛函分析，以建立 $\lambda$-测地凸性。其方法论分为三个主要阶段：

1. 梯度流识别： 论文首先确定了切片 Wasserstein 流（SWF）——一种密度 $\rho$ 向目标 $\nu$ 演化的各向同性连续时间演化过程——对应于泛函 $F_\nu$ 的 Wasserstein 梯度下降。驱动该流的速度场 $v[\rho]$ 被明确推导为 $F_\nu$ 的一阶变分的负梯度。
2. 速度 Lipschitz 界限： 核心技术策略依赖于引理 3.1，该引理指出，如果速度场 $v[\rho] + \frac{1}{d}\text{id}$ 是 Lipschitz 连续的且 Lipschitz 常数为 $L$，那么 $F_\nu$ 是 $\lambda$-测地凸的，其中 $\lambda = \frac{1}{d} - L$。因此，问题转化为推导在有界密度测度集合上速度场 $v[\rho]$ 的一致 Lipschitz 界限。
3. 投影分析： 为了界定 Lipschitz 常数，作者分析了一维投影中最优传输映射的导数。他们将传输映射 $T_\vartheta$ 的导数与源密度投影 $\rho_\vartheta$ 和目标密度投影 $\nu_\vartheta$ 的比值联系起来。
   • 情况 1（强假设）： 最初，他们考虑目标测度 $\nu$ 的投影在一致有下界的情况（$\nu_\vartheta \geq c_0 > 0$）。这允许直接估计，但被证明是一个非常苛刻的条件，基本上将存在性限制在维度 $d \leq 3$ 内。
   • 情况 2（一般假设）： 作者随后推导出了一个更稳健的结果（定理 4.4），该结果仅要求目标测度 $\nu$ 满足特定的几何条件（假设 4.6）。该条件规定了 $\nu$ 在其支撑集边界附近的密度下界，即 $\nu_\vartheta(r) \geq \kappa (\min\{N_\vartheta(r), 1-N_\vartheta(r)\})^\alpha$，其中 $N_\vartheta$ 是投影的累积分布。

主要贡献与结果

• 定理 4.4（主要结果）： 论文证明，对于满足假设 (4.6) 的目标测度 $\nu$ 以及集合 $K = \{\rho \in \mathcal{P}_2(\mathbb{R}^d) : \|\rho\|_{L^\infty} \leq M\}$ 中的源测度，泛函 $F_\nu$ 在 $K$ 上是 $\lambda$-测地凸的。其凸性模数由下式给出：
  $$\lambda = \frac{1}{d} - \frac{c(d)}{(1 - \frac{\alpha d}{d-1})\kappa} M^{1/d}$$
  其中 $c(d)$ 是仅取决于维度的常数，而 $\kappa$ 和 $\alpha$ 是刻画目标测度 $\nu$ 的假设 (4.6) 中的参数。作者指出，$\lambda$ 通常是非正的，并且随着密度界 $M$ 的增加而减小，这表明凸性在高密度测度下会退化。
• 目标测度的几何条件： 论文对假设 (4.6) 进行了详细分析，证明了该假设被满足于以下类型的测度：在光滑、一致凸区域上密度远离零的测度（示例 4.10），以及在支撑集边界附近具有特定幂律行为的测度（示例 4.8）。
• 切片 Wasserstein 流的唯一性： 作为 $\lambda$-测地凸性结果的直接推论，作者确立了切片 Wasserstein 流的适定性。具体而言，推论 1.1 指出，如果 $\nu$ 满足 (4.6) 且初始数据 $\rho_0$ 在 $L^\infty$ 中，则存在唯一的解序列 $\{\rho_t\}_{t \geq 0}$ 满足流方程 (1.3)，且其 $L^\infty$ 范数保持局部有界。
• 维度约束： 论文阐明了“投影一致有下界”这一条件的几何局限性，通过备注 4.2 表明，此类测度本质上仅存在于维度 $d \leq 3$ 的情况下。

意义与主张
论文声称，其主要意义在于将测地凸性理论扩展到切片 Wasserstein 距离，由于其计算效率高，该度量在机器学习和数据科学领域被广泛使用。通过在现实的密度界限下证明 $\lambda$-测地凸性，作者使得将成熟的 Wasserstein 梯度流理论（开发于如 [AGS05] 和 [San17] 等著作中）应用于切片 Wasserstein 流成为可能。

具体而言，作者强调该结果：

1. 保证了 SWF 解的唯一性，这是此前在这些一般条件下未曾建立的性质。
2. 暗示了离散 Jordan-Kinderlehrer-Otto (JKO) 方案向连续流解的收敛速率。
3. 允许在未来涉及 $SW_2$ 的研究中使用流交换技术（flow interchange technique）。

作者对该凸性结果的范围保持了审慎的态度，明确指出 $\lambda$ 通常是负的（意味着在标准意义下是弱凸性甚至凹性），并且当密度界 $M$ 增大时，凸性模数会恶化。他们并未提出新的算法或实验验证，而是为严谨分析现有切片传输方法的行为提供了必要的理论基础。