在机器学习中,经常要度量两个对象的相似度,例如k-最近邻算法,即通过度量数据的相似度而进行分类。在无监督学习中,K-Means算法是一种聚类算法,它通过欧几里得距离计算指定的数据点与聚类中心的距离。在推荐系统中,也会用到相似度的计算(当然还有其他方面的度量)。
本文中,将介绍业务实践中最常用的几种相似度的度量方法。
- 基于相似性的度量
- 皮尔逊相关系数
- 斯皮尔曼秩相关系数
- 肯德尔秩相关系数
- 余弦相似度
- 雅卡尔相似度
- 基于距离的度量
- 欧几里得距离
- 曼哈顿距离
1. 基于相似性的度量
1.1 皮尔逊相关系数
皮尔逊相关系数度量两个随机变量之间的线性关系,包括相关程度和方向。这两个随机变量可以是连续型,也可以是离散型。
计算公式如下:
其中,。
皮尔逊相关系数的范围在 $ [-1,1] $, -1意味着两者负相关,1代表正相关。
用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()
# 计算皮尔逊相关系数
corr, _ = pearsonr(x, y)
print('Pearsons correlation: %.3f' % corr)
输出:Pearsons correlation: 0.810
1.2 斯皮尔曼秩相关系数
斯皮尔曼秩相关系数(Spearman’s correlation)常用于非参数统计。所谓非参数统计,即概率分布不依靠参数(常见的正态分布、二项分布等都是参数统计)。最常见的非参数统计,是统计数据的秩,而不是原始的值。斯皮尔曼秩相关系数的计算方法,与皮尔逊相关系数类似,区别就在于斯皮尔曼相关系数中使用的是数据的秩。
为了计算斯皮尔曼相关系数,我们首先需要将每个原始数据转化为等级数据,建立如下映射关系:
,
,
例如,原始数据为 [0, -5, 4, 7],则对应的等级数据为[2, 1, 3, 4]。
用以下方法计算斯皮尔曼相关系数:
其中 。
斯皮尔曼秩相关系数可以度量两个量之间的非线性相似度,这是和皮尔逊相关系数的重要区别。它的取值范围从-1到+1。
下面的图显示了了皮尔逊相关系数和斯皮尔曼相关系数之间的差异。
图中的数据是呈非线性单调,如果用皮尔逊相关系数——它度量线性关系,得到的相关系数是 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 表示两个变量之间的强正相关。
相对于斯皮尔曼秩相关系数,肯德尔秩相关系数具有统计学上的优势,如果样本量比较大了,对肯德尔秩相关系数的影响较小,特别是手工计算时,体现出了优势。但是,如果用程序实现计算,从算法的角度看,斯皮尔曼秩相关系数的时间复杂度是 ,肯德尔秩相关系数的时间复杂度是
计算肯德尔秩相关系数的第一步与前述斯皮尔曼秩相关系数一样,也是要得到原始数据的等级数据,然后依据下面的公式计算:
其中
在Python中实现肯德尔相关系数的计算,方法如下:
from scipy.stats import kendalltau
corr, _ = kendalltau(x, y)
print(‘Kendalls tau: %.3f’ % corr)
输出结果:Kendalls tau: 0.695
1.4 余弦相似度
余弦相似度计算两个向量或者随机变量之间夹角的余弦,公式如下:
下图显示了余弦函数的特点,从中可知,余弦函数的取值在 -1 到 +1 之间。如果向量指向相同的方向,余弦相似度是+1。如果向量指向相反的方向,余弦相似度为-1。
余弦相似度在文本分析中很常见。它用于确定文档之间的相似程度,而不考虑文档的大小。
下面的程序中演示了在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 为两个有限样本集合,雅卡尔相似度定义为:
下图以可视化方式解释了定义式的含义。
我们可以看到,雅卡尔相似度是交集的大小除以样本集的并集的大小。
余弦相似度和雅卡尔相似度都是度量文本相似度的常用方法,但雅卡尔相似度在计算上成本较高,因为它要将一个文档的所有词汇匹配到另一个文档。实践证明,雅卡尔相似度在检测重复项方面很有用——集合运算的特点。
用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 欧几里得距离
欧几里德距离是两个向量之间的直线距离。
设两个向量 和
与前述的余弦相似度和雅卡尔相似度相比,欧几里得距离很少用于NLP中,它更适用于计算连续型变量间的距离。要注意,欧几里得距离与度量单位有关,所以,在计算的时候,首先要进行单位统一。
在Python中计算欧几里得距离的基本方法是:
from scipy.spatial import distance
dst = distance.euclidean(x,y)
print(‘Euclidean distance: %.3f’ % dst)
输出结果:Euclidean distance: 3.273
2.2 曼哈顿距离
曼哈顿距离,也叫做城市街区距离,与欧几里得距离不同,它是从一个向量到另一个向量的距离。可以想象,当你不能穿过建筑物时,这个度量可以用来计算两点之间的距离。
计算曼哈顿距离的公式如下:
下图中的绿线表示欧几里德距离,而紫线表示曼哈顿距离。
在许多机器学习应用中,欧几里德距离是首选的度量标准。然而,对于高维数据来说,曼哈顿距离更可取,因为它产生了更有说服力的结果。
在Python中实现曼哈顿距离的方法是:
from scipy.spatial import distance
dst = distance.cityblock(x,y)
print(‘Manhattan distance: %.3f’ % dst)
输出结果:Manhattan distance: 10.468
总结
本文概述了在实践中度量相似度的常见方法。在实际的问题中,没有简单的“如果. … 那么 … ”流程图来选择使用哪一种相似性度量方法。我们首先需要了解和研究数据。然后,要想为给定的数据科学问题找到量化相似性的正确方法,始终需要针对每一个具体案例做出具体决策。
注:本文内容是以2021年5月份即将出版的《机器学习数学基础》(作者:齐伟)中的某章节内容为基础编写而成。更详细的相关内容,请参阅此书。