在机器学习中,经常要度量两个对象的相似度,例如k-最近邻算法,即通过度量数据的相似度而进行分类。在无监督学习中,K-Means算法是一种聚类算法,它通过欧几里得距离计算指定的数据点与聚类中心的距离。在推荐系统中,也会用到相似度的计算(当然还有其他方面的度量)。

本文中,将介绍业务实践中最常用的几种相似度的度量方法。

  1. 基于相似性的度量
  • 皮尔逊相关系数
  • 斯皮尔曼秩相关系数
  • 肯德尔秩相关系数
  • 余弦相似度
  • 雅卡尔相似度
  1. 基于距离的度量
  • 欧几里得距离
  • 曼哈顿距离

1. 基于相似性的度量

1.1 皮尔逊相关系数

皮尔逊相关系数度量两个随机变量之间的线性关系,包括相关程度和方向。这两个随机变量可以是连续型,也可以是离散型。

计算公式如下:

HanLP 做相似度分析 相似度检测算法_机器学习

其中,HanLP 做相似度分析 相似度检测算法_神经网络_02
皮尔逊相关系数的范围在 $ [-1,1] $, -1意味着两者负相关,1代表正相关。

HanLP 做相似度分析 相似度检测算法_神经网络_03

用python实现皮尔逊相关系数的计算:

import numpy as np
from scipy.stats import pearsonr
import matplotlib.pyplot as plt。

# 设置随机数的种子
np.random.seed(42)
# 创建随机数
x = np.random.randn(15)
y = x + np.random.randn(15)

# 作图
plt.scatter(x, y)
plt.plot(np.unique(x), np.poly1d(np.polyfit(x, y, 1))(np.unique(x)))
plt.xlabel('x')
plt.ylabel('y')
plt.show()

HanLP 做相似度分析 相似度检测算法_深度学习_04

# 计算皮尔逊相关系数
corr, _ = pearsonr(x, y)
print('Pearsons correlation: %.3f' % corr)

输出:Pearsons correlation: 0.810

1.2 斯皮尔曼秩相关系数

斯皮尔曼秩相关系数(Spearman’s correlation)常用于非参数统计。所谓非参数统计,即概率分布不依靠参数(常见的正态分布、二项分布等都是参数统计)。最常见的非参数统计,是统计数据的秩,而不是原始的值。斯皮尔曼秩相关系数的计算方法,与皮尔逊相关系数类似,区别就在于斯皮尔曼相关系数中使用的是数据的秩。

为了计算斯皮尔曼相关系数,我们首先需要将每个原始数据转化为等级数据,建立如下映射关系:

HanLP 做相似度分析 相似度检测算法_机器学习_05HanLP 做相似度分析 相似度检测算法_深度学习_06
HanLP 做相似度分析 相似度检测算法_机器学习_07HanLP 做相似度分析 相似度检测算法_HanLP 做相似度分析_08

例如,原始数据为 [0, -5, 4, 7],则对应的等级数据为[2, 1, 3, 4]。

用以下方法计算斯皮尔曼相关系数:

HanLP 做相似度分析 相似度检测算法_HanLP 做相似度分析_09

其中 HanLP 做相似度分析 相似度检测算法_相似度_10

斯皮尔曼秩相关系数可以度量两个量之间的非线性相似度,这是和皮尔逊相关系数的重要区别。它的取值范围从-1到+1。

下面的图显示了了皮尔逊相关系数和斯皮尔曼相关系数之间的差异。

HanLP 做相似度分析 相似度检测算法_HanLP 做相似度分析_11

图中的数据是呈非线性单调,如果用皮尔逊相关系数——它度量线性关系,得到的相关系数是 0.88 ,而斯皮尔曼秩相关系数是 1 。

在研究数据的相似度时,根据经验,建议分别计算皮尔逊相关系数和斯皮尔曼秩相关系数。

在Python中,实现斯皮尔曼秩相关系数的方法如下:

from scipy.stats import spearmanr
# 计算斯皮尔曼秩相关系数
corr, _ = spearmanr(x, y)
print(‘Spearmans correlation: %.3f’ % corr)

输出结果:Spearmans correlation: 0.836

1.3 肯德尔秩相关系数

肯德尔秩相关系数与斯皮尔曼相关系数类似,都是非参数统计中度量相似度的方法,都依据原始数据的等级数据进行计算,不是原始数据。

肯德尔秩相关系数的值也是在 -1 和 +1 之间,其中 -1 表示两个变量之间的强负相关,1 表示两个变量之间的强正相关。

相对于斯皮尔曼秩相关系数,肯德尔秩相关系数具有统计学上的优势,如果样本量比较大了,对肯德尔秩相关系数的影响较小,特别是手工计算时,体现出了优势。但是,如果用程序实现计算,从算法的角度看,斯皮尔曼秩相关系数的时间复杂度是 HanLP 做相似度分析 相似度检测算法_神经网络_12 ,肯德尔秩相关系数的时间复杂度是 HanLP 做相似度分析 相似度检测算法_HanLP 做相似度分析_13

计算肯德尔秩相关系数的第一步与前述斯皮尔曼秩相关系数一样,也是要得到原始数据的等级数据,然后依据下面的公式计算:

HanLP 做相似度分析 相似度检测算法_深度学习_14

其中

HanLP 做相似度分析 相似度检测算法_深度学习_15

在Python中实现肯德尔相关系数的计算,方法如下:

from scipy.stats import kendalltau
corr, _ = kendalltau(x, y)
print(‘Kendalls tau: %.3f’ % corr)

输出结果:Kendalls tau: 0.695

1.4 余弦相似度

余弦相似度计算两个向量或者随机变量之间夹角的余弦,公式如下:

HanLP 做相似度分析 相似度检测算法_HanLP 做相似度分析_16

下图显示了余弦函数的特点,从中可知,余弦函数的取值在 -1 到 +1 之间。如果向量指向相同的方向,余弦相似度是+1。如果向量指向相反的方向,余弦相似度为-1。

HanLP 做相似度分析 相似度检测算法_神经网络_17

HanLP 做相似度分析 相似度检测算法_相似度_18

余弦相似度在文本分析中很常见。它用于确定文档之间的相似程度,而不考虑文档的大小。

下面的程序中演示了在Python语言中实现余弦相似度的方法。

from sklearn.metrics.pairwise import cosine_similarity
cos_sim = cosine_similarity(x.reshape(1,1),y.reshape(1,-1))
print('Cosine similarity: %.3f' % cos_sim)

输出结果:Cosine similarity: 0.773

1.5 雅卡尔相似度

雅卡尔相似度(Jaccard similarity),又称为“雅卡尔指数”(Jaccard index)、“并交比”(Intersection over Union),是用于比较两个集合相似性的统计量。设 A、B 为两个有限样本集合,雅卡尔相似度定义为:

HanLP 做相似度分析 相似度检测算法_相似度_19

下图以可视化方式解释了定义式的含义。

HanLP 做相似度分析 相似度检测算法_深度学习_20

我们可以看到,雅卡尔相似度是交集的大小除以样本集的并集的大小。

余弦相似度和雅卡尔相似度都是度量文本相似度的常用方法,但雅卡尔相似度在计算上成本较高,因为它要将一个文档的所有词汇匹配到另一个文档。实践证明,雅卡尔相似度在检测重复项方面很有用——集合运算的特点。

用Python实现雅卡尔相似度的计算过程:

from sklearn.metrics import jaccard_score
A = [1, 1, 1, 0]
B = [1, 1, 0, 1]
jacc = jaccard_score(A,B)
print(‘Jaccard similarity: %.3f’ % jacc)

输出:Jaccard similarity: 0.500

2. 基于距离的度量

2.1 欧几里得距离

欧几里德距离是两个向量之间的直线距离。

设两个向量 HanLP 做相似度分析 相似度检测算法_HanLP 做相似度分析_21HanLP 做相似度分析 相似度检测算法_深度学习_22

HanLP 做相似度分析 相似度检测算法_神经网络_23

与前述的余弦相似度和雅卡尔相似度相比,欧几里得距离很少用于NLP中,它更适用于计算连续型变量间的距离。要注意,欧几里得距离与度量单位有关,所以,在计算的时候,首先要进行单位统一。

在Python中计算欧几里得距离的基本方法是:

from scipy.spatial import distance
dst = distance.euclidean(x,y)
print(‘Euclidean distance: %.3f’ % dst)

输出结果:Euclidean distance: 3.273

2.2 曼哈顿距离

曼哈顿距离,也叫做城市街区距离,与欧几里得距离不同,它是从一个向量到另一个向量的距离。可以想象,当你不能穿过建筑物时,这个度量可以用来计算两点之间的距离。

计算曼哈顿距离的公式如下:

HanLP 做相似度分析 相似度检测算法_HanLP 做相似度分析_24

下图中的绿线表示欧几里德距离,而紫线表示曼哈顿距离。

HanLP 做相似度分析 相似度检测算法_HanLP 做相似度分析_25

在许多机器学习应用中,欧几里德距离是首选的度量标准。然而,对于高维数据来说,曼哈顿距离更可取,因为它产生了更有说服力的结果。

在Python中实现曼哈顿距离的方法是:

from scipy.spatial import distance
dst = distance.cityblock(x,y)
print(‘Manhattan distance: %.3f’ % dst)

输出结果:Manhattan distance: 10.468

总结

本文概述了在实践中度量相似度的常见方法。在实际的问题中,没有简单的“如果. … 那么 … ”流程图来选择使用哪一种相似性度量方法。我们首先需要了解和研究数据。然后,要想为给定的数据科学问题找到量化相似性的正确方法,始终需要针对每一个具体案例做出具体决策。

注:本文内容是以2021年5月份即将出版的《机器学习数学基础》(作者:齐伟)中的某章节内容为基础编写而成。更详细的相关内容,请参阅此书。