Gromov-Wasserstein Distance 用于数据降维 — 文献综述
本文系统梳理 Gromov-Wasserstein(GW)距离及其变体在数据降维(Dimensionality Reduction)领域的最新进展。核心路线:从经典 GW 出发,经过半松弛 GW(semi-relaxed GW)到分布缩减(Distributional Reduction),并结合嵌入学习、有监督扩展和流匹配等方向。
背景:为什么 GW 适合降维?
降维的核心目标是:给定高维数据 $X \subset \mathbb{R}^D$,找一个低维嵌入 $Z \subset \mathbb{R}^d$($d \ll D$),使得 $Z$ 尽可能”保持” $X$ 的某种结构。
GW 距离天然适合这个任务,因为它度量两个度量空间之间的差异——而不要求空间维度相同、特征可比。这正是降维的设定:高维空间和低维空间维度不同、特征不可比,但我们可以要求它们的内部点间距离结构一致。
标准 GW 距离: \(\text{GW}(P, Q) = \min_{\pi \in \Pi(\mu,\nu)} \sum_{i,j,k,l} \|d_X(x_i, x_k) - d_Y(y_j, y_l)\|^2 \, \pi_{ij} \pi_{kl}\)
Group 1:核心 GW 降维方法
1.1 Eufrazio et al. [arXiv:2501.13732] — 基于 GW 的降维
将降维重写为 GW 对齐问题:
\[\min_{Z} \text{GW}(\mu_X, \nu_Z)\]- 高维数据 $\mu_X$ 和低维嵌入 $\nu_Z$ 的 GW 距离最小化
- 超越传统 MDS:MDS 固定点到点的对应关系($\pi = I$),GW 同时学习对应关系和嵌入位置
- 梯度下降交替优化 $\pi$ 和 $Z$
与经典算法的关系:
| 方法 | GW 视角 |
|---|---|
| MDS | 固定 $\pi = I$,只优化 Z |
| Isomap | 用测地距离替换欧氏距离,其他同 MDS |
| GW-DR (本文) | 不固定 $\pi$,同时学对应+嵌入 |
1.2 Van Assel et al. [arXiv:2310.03398] — 半松弛 GW 插值聚类与降维
semi-relaxed GW (srGW):
\[\text{srGW}(D_X, D_Z, \mu, \nu) = \min_{\pi \in \Pi(\mu, \nu)} \sum_{i,j,k,l} |d_X(x_i, x_k)^2 - d_Z(z_j, z_l)^2| \, \pi_{ij} \pi_{kl}\]关键特性:目标分布的权重 $\nu$ 可以学习(部分松弛)。
| 嵌入样本数 | 行为 |
|---|---|
| $n_{emb} = n_{input}$ | 退化为经典降维 |
| $n_{emb} < n_{input}$ | 样本缩减——多点到一点 |
| 维度无约束 | 运输计划 $\pi$ 给出硬聚类 |
| 中间状态 | 聚类 + 降维同时做 |
一个参数 $\lambda$ 连续控制从”纯降维”到”纯聚类”的过渡。
1.3 Van Assel et al. [arXiv:2402.02239] — 分布缩减(Distributional Reduction)
系统化扩展,提出统一框架:
\[\boxed{\text{分布缩减} = \text{降维} + \text{聚类 统一在 GW 框架下}}\]- 学一个缩减表示 $Y = {y_1, \ldots, y_m}$($m \ll n$)和权重 $b \in \mathbb{R}^m$
- $m = n$ → 经典 MDS
- $m \ll n$ + 低维 Y → 降维
- $m \ll n$ + 维度无约束 → 聚类
- 支持多尺度分析:选择不同 $m$ 在不同分辨率下看数据
数学本质:在 Wasserstein 空间中学一个低维多面体(simplex)近似。
1.4 Clark, Needham, Weighill [arXiv:2405.15959] — 推广到任意度量空间
传统降维:$\mathbb{R}^D \to \mathbb{R}^d$
本文:$\mathbb{R}^D \to$ 任意度量空间(圆 $S^1$、图、树、流形)
\[\min_{Z \subset \mathcal{M}} \text{srGW}(D_X, D_Z)\]- 建立 srGW 与 Gromov-Hausdorff 距离的理论联系
- srGW 是 GH 的”可计算松弛”
- 应用例子:政治选区重划嵌入圆环 $S^1$,判断是否 gerrymandering
Group 2:嵌入 / 表征学习
2.1 Bai et al. [arXiv:2410.16669] — 线性部分 GW 嵌入 (LPGW)
痛点:GW 复杂度 $O(n^4)$,每次新样本都要重算。
LPGW:
\[\text{LPGW}(P, Q) = \|\phi(P) - \phi(Q)\|\]- $\phi$ 是显式特征映射,将概率测度映射到特征向量
- 选参考分布 $\mu_0$,对每个分布找 $\mu_0$ 到 $P$ 的最优部分运输
- 运输方向作为特征
好处:$K$ 个分布间距离计算从 $O(K^2)$ 降到 $O(K)$
2.2 Eufrazio et al. [arXiv:2604.23912] — Bary-GWMDS
多视图数据降维:
\[\min_{Z, \pi^{(1)}, \ldots, \pi^{(V)}} \sum_{v=1}^V \lambda_v \cdot \text{GW}(D^{(v)}, D_Z)\]- GW 重心:找嵌入 $Z$ 使其到各视图的 GW 距离加权和最小
- 每个视图”结构投票”,通过 GW 对齐后取共识
- Mean-GWMDS-C:聚类版,约束嵌入数量
2.3 Eufrazio et al. [arXiv:2604.02610] — 保结构多视图嵌入
两种策略: - Mean-GWMDS:平均距离矩阵后 GW 嵌入(简单但损失非线性结构) - Multi-GWMDS:每个视图独立 GW 嵌入,再从候选 Z 中选代表性嵌入
后者更精细:$\min_{Z} \sum_v \text{GW}(D_Z^{(v)}, D_Z)$
2.4 Xu et al. [arXiv:1901.06003] — GW 图匹配与节点嵌入
联合框架:同时做图匹配 + 节点嵌入
\[\min_{\pi, Z_1, Z_2} \text{GW}(D_{G_1}, D_{G_2}) + \lambda \cdot \text{reg}(Z_1, Z_2, \pi)\]- 固定 Z 优化 $\pi$ → 图匹配
- 固定 $\pi$ 优化 $Z$ → 节点嵌入
- 交替近端点法求解
| node2vec | GW 方法 | |
|---|---|---|
| 图数 | 单个图 | 两个不同图 |
| 对齐 | 不需要 | 自动学出跨图对应 |
| 应用 | 单个图分析 | 跨图匹配+嵌入 |
Group 3:扩展方向
3.1 Ryner & Karlsson [arXiv:2207.12279] — GW 型反馈正交化
不是直接最小化 GW,而是用 GW 结构做反馈:
- 扩散映射:数据点用 Markov 转移概率表示
- GW 型反馈:迭代修改转移概率矩阵使其正交化(簇间概率→0,簇内→1)
- 证明对某些参数全局收敛到唯一不动点
3.2 Ramesh et al. [arXiv:2605.27619] — 有监督分布缩减 (SDR)
将分布缩减推广到有监督:
\[\min_{Y, b, \pi} \underbrace{\text{FGW}(D_X, D_Y, \pi)}_{\text{保结构}} + \beta \cdot \underbrace{\text{MI}(Y, \text{label})}_{\text{标签依赖}}\]- Fused GW (FGW):GW 中集成标签特征距离
- 互信息最大化:嵌入携带标签信息
- 学到的距离可用于非平稳核函数设计(高斯过程)
3.3 Cai et al. [arXiv:2510.23015] — Coupled Flow Matching (CPFM)
Flow Matching + GW 做可控降维+重建:
两个耦合流: - 潜伏流 $y_t$:先验 → 嵌入 - 数据流 $x_t$:嵌入条件 → 原始数据
GW 保证嵌入空间几何忠实反映原始数据结构:不只是压缩,而是”压缩后的空间几何 ≈ 原始空间几何”。
用户可指定保留哪些语义维度,残余信息存在流网络权重中可恢复。
整体逻辑关系
┌─────────────────────┐
│ GW 距离 — 基础 │
└────┬────────┬───────┘
│ │
┌────────────┤ ├──────────────┐
▼ ▼ ▼ ▼
┌──────────┐ ┌──────────┐ ┌──────┐ ┌──────────┐
│ 纯降维 │ │聚类+降维 │ │嵌入 │ │ 扩展方向 │
│ #1 Eufrazio│ │#2 VanA │ │#5 Bai│ │ #9 反馈 │
│ #4 Clark │ │#3 VanA │ │#6-7E │ │#10 有监督 │
└──────────┘ │#10 Ramesh │ │#8 Xu │ │#11 流匹配 │
└──────────┘ └──────┘ └──────────┘
参考文献
- Eufrazio et al. A dimensionality reduction technique based on the Gromov-Wasserstein distance. arXiv:2501.13732
- Van Assel et al. Interpolating between Clustering and Dimensionality Reduction with Gromov-Wasserstein. arXiv:2310.03398
- Van Assel et al. Distributional Reduction: Unifying Dimensionality Reduction and Clustering with Gromov-Wasserstein. arXiv:2402.02239
- Clark, Needham, Weighill. Generalized Dimension Reduction Using Semi-Relaxed Gromov-Wasserstein Distance. arXiv:2405.15959
- Bai et al. Linear Partial Gromov-Wasserstein Embedding. arXiv:2410.16669
- Eufrazio et al. Gromov-Wasserstein Methods for Multi-View Relational Embedding and Clustering. arXiv:2604.23912
- Eufrazio et al. Structure-Preserving Multi-View Embedding Using Gromov-Wasserstein Optimal Transport. arXiv:2604.02610
- Xu et al. Gromov-Wasserstein Learning for Graph Matching and Node Embedding. arXiv:1901.06003
- Ryner & Karlsson. Orthogonalization of data via Gromov-Wasserstein type feedback for clustering and visualization. arXiv:2207.12279
- Ramesh et al. Supervised Distributional Reduction via Optimal Transport and Dependence Maximization. arXiv:2605.27619
- Cai et al. Coupled Flow Matching. arXiv:2510.23015