UMAP:比t-SNE更好的降维算法

论文题目:UMAP: Uniform Manifold Approximation and Projection for Dimension Reduction
作者:Leland McInnes; John Healy; James Melville
时间:December 7, 2018

UMAP (Uniform Manifold Approximation and Projection) 算法是一种创新的降维流形学习算法。Ideas 来自于拓扑数据分析。

论文整体思路

  • UMAP uses local manifold approximations and patches together their local fuzzy simplicial set representations to construct a topological representation of the high dimensional data.
  • a similar process can be used to construct an equivalent topological representation.
  • then minimize the cross-entropy between the two topological representations.

背景补充

单纯形 (Simplices)

从几何上讲,单纯形是构建 k 维对象的一种非常简单的方法。 k 维单纯形称为 k-单纯形。它是由+1个独立点的凸包构成的。因此,0-单纯形是一 个点,1-单纯形是一条线段 (在两个零单纯形之间),2-单纯形是一个三角形 (其中三个1-单纯形作为“面”),而3-单纯形是一个四面体 (四个2-单纯形作为“面”)。

UMAP:比t-SNE更好的降维算法_第1张图片
单纯形

单纯复形 (Simplicial complex)

A simplicial complex is a set of simplices that satisfies the following conditions:

  1. Every face of a simplex from is also in .
  2. The non-empty intersection of any two simplices is a face of both .

(The convex hull of any nonempty subset of the n + 1 points that define an n-simplex is called a face of the simplex. )

单纯复形 K 是单纯形的集合。任意单纯形的任何面也在 K 中(确保所有面都存在),并且 K 中任何两个单纯形的交集都是这两个单纯形的面。

UMAP:比t-SNE更好的降维算法_第2张图片
单纯复形

理论应用

假设一个有限的数据样本集来自一个拓扑空间,如果要了解该空间的拓扑,需要先生成该空间的开覆盖 (open cover)。

若数据实际处于 metric space,一种近似开覆盖的方法是对每个数据点形成固定半径的球。

举例,有一批数据位于假设的流形中。如图所示:

UMAP:比t-SNE更好的降维算法_第3张图片
Test data set of a noisy sine wave

如果固定一个半径,就可以把覆盖的开放集想象成一个个圆(考虑二维)。

UMAP:比t-SNE更好的降维算法_第4张图片
A basic open cover of the test data

构造单纯复形。 具体是指构造Čech 复形。

具体方法是:

让覆盖 (cover) 中的每个集合都是0-单纯形;如果两个集合具有非空交集,则在它们之间创建1-单纯形;如果三个这样的集合的三重交集都非空,则在集合之间创建2-单纯形;以此类推。

然后,我们可以将0-、1-和2-单纯形的单纯复形描绘为点、线和三角形。如图所示:

UMAP:比t-SNE更好的降维算法_第5张图片
A simplicial complex built from the test data

通过这样的方法捕获拓扑结构。

一些改进

理论可行,但在实际数据上的效果不佳。

问题1:如何为开覆盖选择正确的球半径?

半径太小 ---> 单纯复形分裂为许多相连的组件。
半径太大 ---> 单纯复形变成维度很高的单纯形,无法捕捉流形结构。

解决方案:

Assumption: Data is uniformly distributed on the manifold.

UMAP:比t-SNE更好的降维算法_第6张图片
Open balls over uniformly_distributed_data

数据在流行上均匀分布,就不难选择一个好的半径。

数据均匀分布的假设在流形学习的其他地方也有。比如:拉普拉斯映射要证明有效,就要假设数据是均匀分布的。

但是,现实中的数据不会是均匀分布的。问题反过来:假设数据均匀分布在流形上,我们能得到关于流形的什么信息?

即:距离的概念在流形上是不一样的 - 空间扭曲:在数据看起来更稀疏或更密集的地方来拉伸或缩小。

引入黎曼几何。
在流形上定义一个 Riemannian metric 让假设为真。
在黎曼空间里数据点是均匀分布的,但是放到欧几里得空间就变成非均匀分布了。

球的大小在不同的度量下是不一样的。在黎曼度量下球的 size 是一样的,但放在欧几里德空间就不一样了。

UMAP:比t-SNE更好的降维算法_第7张图片
Open balls of radius one with a locally varying metric

通过假设数据是均匀分布的,可以通过使用一些标准的黎曼几何来计算每个点的局部距离概念(其近似值)。围绕一个点的单位球延伸到该点的第 k 个最近的邻居,其中 k 是我们用来近似局部距离的样本大小。

每个点都有自己唯一的距离函数,我们可以简单地选择半径为1的球作为局部距离函数!

于是我们发现这不就是 k-近邻图吗?

这意味着数据集中的每个点都被赋予了到其k个最近邻居的每个点的边-这是我们使用半径为1的球的局部变化度量的有效结果。

对 k 的拓扑解释,k 的选择决定了我们希望在多大程度上局部地估计黎曼度量。选择很小的k意味着我们想要一个非常局部的解释,它将更准确地捕捉到黎曼度量的精细细节结构和变化。选择较大的 k 意味着我们的估计将基于更大的区域,可以有更多数据来估计。

基于黎曼度量的另一个好处

我们实际上有一个与每个点相关联的局部度量空间,并且可以有意义地测量距离,因此可以根据边上的点之间的距离(根据局部度量)来加权可能生成的图的边。

用模糊拓扑的知识来理解:在一个覆盖的开集合不再是是和否的概念,而是介于 0 和 1 的模糊值。点在给定半径的球里的确定性会随着我们离开球的中心的移动而衰减。我们可以把这样一个模糊的 cover 想象成如图所示的样子:

UMAP:比t-SNE更好的降维算法_第8张图片
Fuzzy open balls of radius one with a locally varying metric

问题2:更高的维度上的数据很多点是完全孤立的。

解决方法:局部连接。

Assumption:
The manifold is locally connected.

UMAP:比t-SNE更好的降维算法_第9张图片
Local connectivity and fuzzy open sets

由于维数诅咒,高维空间的数据,距离更大,但彼此也可能更相似。这意味着到第一个最近邻居的距离可能相当大,但到第十个最近邻居的距离通常只会稍微大一点(相对而言)。局部连接的约束确保我们关注最近邻居之间的距离差异,而不是绝对距离(这显示邻居之间的差别很小)。

问题3:局部度量标准不兼容性。

每个点都有与其相关联的局部度量,并且从点a的角度来看,从点a到点b的距离可能是1.5,但是从点b的角度来看,从点b到点a的距离可能只有0.6。

基于图形的直觉,可认为这是具有不同权重的有向边,如下图所示:

UMAP:比t-SNE更好的降维算法_第10张图片
Edges with incompatible weights

解决方法:

将权重为 a 和 b 的两条不一致的边合并在一起,那么我们应该有一条具有组合权重 的边。考虑这一点的方法是,权重实际上是边 (1-单纯形) 存在的概率。然后,组合权重是至少一条边存在的概率。

如果我们应用这个过程将所有模糊单纯集合并在一起,我们最终得到一个单一的模糊单纯复合体,我们可以再次将其视为一个加权图。在计算方面,我们只是将边权重组合公式应用于整个图 (非边的权重为0)。最后,我们得到了类似这样的东西。

UMAP:比t-SNE更好的降维算法_第11张图片
Graph with combined edge weights

因此,最终只是构造了一个加权图。但是背后有强大的数学知识在支撑。

因此,假设我们现在有了数据的模糊拓扑表示 (数学上说它将捕获数据背后的流形的拓扑),我们如何将其转换为低维表示呢?

找到一个低维表示

目的:希望低维表示具有尽可能相似的模糊拓扑结构。
第一个问题是如何确定低维表示的模糊拓扑结构;
第二个问题是如何找到一个好的拓扑结构。

第一个问题可遵循以上寻找数据模糊拓扑结构的过程。

数据位于的流形是我们试图嵌入的低维欧几里德空间。因此,流形上的距离是相对于全局坐标系的标准欧几里得距离,不再是变化的度量。

第二个问题,“如何找到一个好的低维表示”,即衡量模糊拓扑结构的匹配程度,最终可以转化为一个优化问题。

采用的优化方法是交叉熵函数。

回顾之前的权重处理方法,我们将权重解释为单纯形存在的概率。由于我们正在比较的两个拓扑结构共享相同的0-单纯形,可以想象我们正在比较由1-单纯形索引的两个概率向量。假设这些都是伯努利变量 (最终单纯形要么存在,要么不存在,而且概率是伯努利分布的参数),这里正确的选择是交叉熵。

如果所有可能的 1-单纯形的集合是 ,并且有权重函数使得 是高维情况下 的权重, 是低维情况下 权重,则交叉熵是:

从图的角度来看,可将最小化交叉熵视为一种 力有向图布局算法 (force directed graph layout algorithm)。

第一项, 在点 跨度之间提供吸引力,当在高维空间存在较大的权重时。最小化这一项时, 要尽可能大,这时点之间的距离是尽可能小的。

第二项, 在 两段之间提供排斥力,当 较小时。通过使 尽可能小来最小化这一项。

UMAP 算法

总结:构建UMAP算法包括:第一阶段包括构建模糊拓扑表示,如上所述。第二阶段是简单地优化低维表示,使其具有尽可能接近的模糊拓扑表示,并用交叉熵来度量。

构建初始模糊拓扑表示

在实践中,由于模糊集隶属度逐渐衰减到几乎为零,我们只需要计算每个点的最近邻域的隶属度。归根结底,这意味着我们需要一种快速高效地计算(近似)最近邻居的方法,在高维空间中也是如此。

优化低维嵌入

在实践中,UMAP 使用 的曲线族。

由于拓扑表示的拉普拉斯算子是流形的Laplace-Beltrami算子的近似,我们可以使用谱嵌入初始化低维表示。

数学表达

UMAP can ultimately be described in terms of, construction of, and operations on weighted graphs.

高维空间的分布

用找最近邻的算法,得到每个 的 k 最近邻集合

对于每个 ,确定 和 :

分布的计算

低维空间的分布

where and for default UMAP hyperparameters.

交叉熵作为代价函数

UMAP:比t-SNE更好的降维算法_第12张图片
交叉熵函数

Algorithm description

UMAP:比t-SNE更好的降维算法_第13张图片

总而言之,UMAP算法相对简单(参见算法1)。

如算法1所述,UMAP算法采用四个超参数:

  1. ,近似本地度量时要考虑的邻居数量;

  2. ,目标嵌入维度;

  3. min_dist,嵌入空间中闭合点之间的期望间隔;

  4. 次迭代,即优化低维表示时要使用的训练迭代的数目。

下面将更详细地描述用于构造局部模糊单纯集、确定谱嵌入以及关于模糊集交叉熵优化嵌入的各个函数。

UMAP:比t-SNE更好的降维算法_第14张图片

算法 2 描述了局部模糊单纯集。我们可以通过寻找n个最近邻域,在流形上生成适当的归一化距离,然后通过函子FinSing将有限度量空间转换为单纯集,在这种情况下,单纯集转化为负距离的指数,从而构造出局部于给定点 的模糊单纯集。

UMAP:比t-SNE更好的降维算法_第15张图片

如算法 3 所示。我们不是直接使用到第 n 个最近邻居的距离作为归一化,而是使用KNN距离的平滑版本,它将 1-单纯形的模糊集合的基数固定为固定值。为此,我们在实证实验的基础上选择了log2(N)。

UMAP:比t-SNE更好的降维算法_第16张图片

算法4:通过将全局模糊拓扑表示的1-骨架视为加权图,并在对称归一化拉普拉斯算子上使用标准谱方法来执行谱嵌入。

UMAP的最后一个主要组成部分是通过最小化模糊集交叉熵来优化嵌入。回想一下,关于给定的隶属函数 和 ,模糊集交叉熵由下式给出

通过最小化模糊集合的 cross entropy 来最优化低维嵌入。

UMAP:比t-SNE更好的降维算法_第17张图片

算法 5 通过随机梯度下降来优化整个过程。

实验结果

1. 对全局结构和局部结构的双重捕捉

UMAP:比t-SNE更好的降维算法_第18张图片

从图中几种降维算法的比较中可发现:UMAP成功地反映了Laplacian特征映射和 PCA(特别是对于MNIST和Fashion-MNIST)所很好地表示的大部分大规模全局结 构,同时也保留了类似于t-SNE和LargeVis的局部精细结构。

2. UMAP 运行时间感人!

UMAP:比t-SNE更好的降维算法_第19张图片

3. 受样本大小约束较小

UMAP:比t-SNE更好的降维算法_第20张图片

4. UMAP 在数万维范围内仍然表现良好

UMAP:比t-SNE更好的降维算法_第21张图片

UMAP、t-SNE、FIT-SNE和Largevis相对于数据环境维度的运行时性能缩放。当环境维度增加到几千维以上时,t-SNE、Fit-SNE和LargeVis的计算成本都会急剧增加,而 UMAP 在数万维范围内仍然表现良好。

推荐阅读

How Exactly UMAP Works

Why UMAP is Superior Over t-SNE

你可能感兴趣的:(UMAP:比t-SNE更好的降维算法)