【DBSCAN——KD树优化】

原博文:亲测可行。最后对比测试时间,与原博主的结果有些大(DBSCAN:67.0119s、DBSCAN_KD-Tree:8.8526s)

DBSCAN(Density-Based Spatial Clustering of Applications with Noise)聚类算法,是一种基于高密度连通区域的、基于密度的聚类算法,能够将具有足够高密度的区域划分为簇(Cluster),并在具有噪声的数据中发现任意形状的簇。DBSCAN算法通过距离定义出一个密度函数,计算出每个样本附近的密度,从而根据每个样本附近的密度值来找出那些样本相对比较集中的区域,这些区域就是我们要找的簇。

1. DBSCAN算法的基本原理

其它聚类方法大都是基于对象之间的距离进行聚类,聚类结果是球状的簇。DBSCAN 算法利用簇的高密度连通性,寻找被低密度区域分离的高密度区域,可以发现任意形状的簇,其基本思想是:对于一个簇中的每个对象,在其给定半径的领域中包含的对象不能少于某一给定的最小数目。

DBSCAN算法中有两个重要参数:εε表示定义密度时的邻域半径,MM 表示定义核心点时的阈值。

考虑数据集合X={x(1),x(2),...,x(n)}X={x(1),x(2),...,x(n)},引入以下概念与记号。

1. εε邻域

设x∈Xx∈X,称

Nε(x)={y∈X:d(y,x)≤ε}Nε(x)={y∈X:d(y,x)≤ε}


为XX的εε邻域。显然,x∈Nε(x)x∈Nε(x)。

为了简单起见,节点x(i)x(i)与其下标ii一一对应,引入记号

Nε(i)={j:d(y(j),x(i))≤ε;y(j),x(i)∈X}Nε(i)={j:d(y(j),x(i))≤ε;y(j),x(i)∈X}

2. 密度

设x∈Xx∈X,称ρ(x)=|Nε(x)|ρ(x)=|Nε(x)|为xx的密度。密度是一个整数,且依赖于半径εε。

3. 核心点

设x∈Xx∈X,若ρ(x)≥Mρ(x)≥M,则称xx为XX的核心点。记由XX中所有核心点构成的集合为XcXc,并记Xnc=X−XcXnc=X−Xc表示XX中所有非核心点构成的集合。

4. 边界点

若x∈Xncx∈Xnc,且∃y∈X∃y∈X,满足y∈Nε(x)⋂Xcy∈Nε(x)⋂Xc,即XX的非核心点xx的εε邻域中存在核心点,则称xx 为XX的边界点。记由XX中所有边界点构成的集合为XbdXbd。

此外,边界点也可以这么定义,若x∈Xncx∈Xnc,且xx落在某个核心点的εε邻域内,则称xx为XX的边界点。一个边界点可能同时落入一个或多个核心点的εε 邻域内。

5. 噪声点

记Xnoi=X−(Xc⋃Xbd)Xnoi=X−(Xc⋃Xbd),若x∈Xnoix∈Xnoi,则称xx为噪音点。

至此,我们严格给出了核心点、边界点和噪音点的数学定义,且满足X=Xc⋃Xbd⋃XnoiX=Xc⋃Xbd⋃Xnoi.

图1:核心点、边界点和噪声点

直观地说,核心点对应稠密区域内部的点,边界点对应稠密区域边缘的点,而噪音点对应稀疏区域中的点。

数据集通过聚类形成的子集是簇。核心点位于簇的内部,它确定无误地属于某个特定的簇;噪音点是数据集中的干扰数据,它不属于任何一个簇;边界点是一类特殊的点,它位于一个或几个簇的边缘地带,可能属于一个簇,也可能属于另外一个簇,其归属并不明确。

6. 直接密度可达

设x,y∈Xx,y∈X. 若满足x∈Xcx∈Xc,则称yy是xx从直接密度可达的。

7. 密度可达

设p(1),p(2),...,p(m)∈Xp(1),p(2),...,p(m)∈X,其中m≥2m≥2。若它们满足:p(i+1)p(i+1) 是从p(i)p(i)直接密度可达的,其中i=1,2,...,m−1i=1,2,...,m−1,则称p(m)p(m) 是从p(1)p(1) 中密度可达的。

7.1. 当m=2m=2时,密度可达即为直接密度可达。实际上,密度可达是直接密度可达的传递闭包。

7.2. 密度可达关系不具有对称性。若p(m)p(m)是从p(1)p(1)密度可达的,那么p(1)p(1) 不一定是从p(m)p(m)密度可达的。根据上述定义可知,p(1),p(2),...,p(m−1)p(1),p(2),...,p(m−1)必须为核心点,而p(m)p(m)可以是核心点,也可以是边界点。当p(m)p(m)是边界点时,p(1)p(1)一定不是从p(m)p(m)密度可达的。

8. 密度相连

设x,y,z∈Xx,y,z∈X,若yy和zz均是从xx密度可达的,则称yy和zz是密度相连的。显然,密度相连具有对称性。

9. 簇(cluster)

非空集合C⊂XC⊂X,如果CC满足:对于x,y∈Xx,y∈X

  • 若x∈Cx∈C,且yy是从xx密度可达的,则y∈Cy∈C,
  • 若x∈Cx∈C,y∈Cy∈C,则x,yx,y是密度相连的。

则称CC是XX的一个簇。

DBSCAN 算法基于以下一个基本事实:对于任一核心点xx,数据集XX中所有从xx 密度可达的数据点可以构成一个完整的簇CC,且x∈Cx∈C。其核心思想描述如下:从某个选定的核心点出发,不断向密度可达的区域扩张,从而得到一个包含核心点和边界点的最大化区域,区域中任意两点密度相连。

2. DBSCAN算法的实现

《数据挖掘概念与技术》给出的算法伪代码如下:

考虑数据集合X={x(1),x(2),...,x(n)}X={x(1),x(2),...,x(n)}。DBSCAN算法的目标是将数据集合XX分成KK个簇及噪声点集合,其中KK也是由算法得到,为此,引入簇的标记数组

mi={j,−1,若x(i)属于第j个簇;若x(i)为噪声点mi={j,若x(i)属于第j个簇;−1,若x(i)为噪声点


DBSCAN算法的目标就是生成标记数组mi,i=1,...,nmi,i=1,...,n.

为了保证可以更有效地实现算法1中第3句随机选择一个unvisited对象pp,设计了一个数据结构visitlist,其中包含两个列表visitedlist和unvisitedlist,分别用于存储已访问的点和未访问的点,每次从unvisitedlist 中取点可以保证每次取到的点都是未访问过的点,实现代码如下:

代码1:visitlist数据结构

复制代码

 1 # visitlist类用于记录访问列表
 2 # unvisitedlist记录未访问过的点
 3 # visitedlist记录已访问过的点
 4 # unvisitednum记录访问过的点数量
 5 class visitlist:
 6     def _init_(self, count=0):
 7         self.unvisitedlist=[i for i in range(count)]
 8         self.visitedlist=list()
 9         self.unvisitednum=count
10 
11     def visit(self, pointId):
12         self.visitedlist.append(pointId)
13         self.unvisitedlist.remove(pointId)
14         self.unvisitednum -= 1

复制代码

DBSCAN算法实现代码如下:

代码2:DBSCAN算法实现

复制代码

 1 import numpy as np
 2 import matplotlib.pyplot as plt
 3 import math
 4 import random
 5 
 6 def  dist(a, b):
 7     # 计算a,b两个元组的欧几里得距离
 8     return math.sqrt(np.power(a-b, 2).sum())
 9 
10 def my_dbscanl(dataSet, eps, minPts):
11     # numpy.ndarray的 shape属性表示矩阵的行数与列数
12     nPoints = dataSet.shape[0]
13     # (1)标记所有对象为unvisited
14     # 在这里用一个类vPoints进行买现
15     vPoints = visitlist(count=nPoints)
16     # 初始化簇标记列表C,簇标记为 k
17     k = -1
18     C = [-1 for i in range(nPoints)]
19     while(vPoints.unvisitednum > 0):
20         # (3)随机上选择一个unvisited对象p
21         P = random.choice(vPoints.unvisitedlist)
22         # (4)标记p为visited
23         vPoints.visit(p)
24         # (5)if p的$\varepsilon$-邻域至少有MinPts个对象
25         # N是p的$\varepsilon$-邻域点列表
26         N = [i for i in range(nPoints) if dist(dataSet[i], dataSet[p])<= eps]
27         if  len(N) >= minPts:
28             # (6)创建个新簇C,并把p添加到C
29             # 这里的C是一个标记列表,直接对第p个结点进行赋植
30             k += 1
31             C[p]=k
32             # (7)令N为p的ε-邻域中的对象的集合
33             # N是p的$\varepsilon$-邻域点集合
34             # (8) for N中的每个点p'
35             for p1 in N:
36                 # (9) if p'是unvisited
37                 if p1 in vPoints.unvisitedlist:
38                     # (10)标记p’为visited
39                     vPoints.visit(p1)
40                     # (11) if p'的$\varepsilon$-邻域至少有MinPts个点,把这些点添加到N
41                     # 找出p'的$\varepsilon$-邻域点,并将这些点去重添加到N
42                     M=[i for i in range(nPoints) if dist(dataSet[i], \
43                         dataSet[p1]) <= eps]
44                     if len(M) >= minPts:
45                         for i in M:
46                             if i not in N:
47                                 N.append(i)
48                     # (12) if p'还不是任何簇的成员,把P'添加到C
49                     # C是标记列表,直接把p'分到对应的簇里即可
50                     if  C[p1] == -1:
51                         C[p1]= k
52         # (15)else标记p为噪声
53         else:
54             C[p]=-1
55 
56     # (16)until没有标t己为unvisitedl内对象
57     return C

复制代码

 利用sklearn生成数据集,共2500条数据,并利用matplotlib画出散点图,代码如下:

代码3:生成数据集

复制代码

 1 import numpy as np
 2 import matplotlib.pyplot as plt
 3 from sklearn import datasets
 4 
 5 X1, Y1 = datasets.make_circles(n_samples=2000, factor=0.6, noise=0.05,
 6                                random_state=1)
 7 X2, Y2 = datasets.make_blobs(n_samples=500, n_features=2, centers=[[1.5,1.5]],
 8                              cluster_std=[[0.1]], random_state=5)
 9 
10 X = np.concatenate((X1, X2))
11 plt.figure(figsize=(12, 9), dpi=80)
12 plt.scatter(X[:,0], X[:,1], marker='.')
13 plt.show()

复制代码

 

图2:数据集散点图

设置参数Eps=0.1, MinPts=10,聚类结果如下图:

 

图3:聚类结果

3. 利用KD树进行优化

KD树(K-Dimensional Tree),是一种分割k维数据空间的数据结构,是二叉搜索树在多维条件下的推广。主要应用于多维空间关键数据的搜索。KD树的介绍见:https://www.jianshu.com/p/ffe52db3e12b,不赘述。

利用scipy实现KD树的构造和查询,对代码2的算法进行改进,代码如下:

代码4:DBSCAN算法的优化实现

复制代码

 1 import numpy as np
 2 import matplotlib.pyplot as plt
 3 import math
 4 import random
 5 from scipy.spatial import KDTree
 6 
 7 def my-dbscan2(dataSet, eps, minPts):
 8     # numpy.ndarray的 shape属性表示矩阵的行数与列数
 9     # 行数即表小所有点的个数
10     nPoints = dataSet.shape[0]
11     # (1) 标记所有对象为unvisited
12     # 在这里用一个类vPoints进行实现
13     vPoints = visitlist(count=nPoints)
14     # 初始化簇标记列表C,簇标记为 k
15     k = -1
16     C = [-1 for i in range(nPoints)]
17     # 构建KD-Tree,并生成所有距离<=eps的点集合
18     kd = KDTree(X)
19     while(vPoints.unvisitednum>0):
20         # (3) 随机选择一个unvisited对象p
21         p = random.choice(vPoints.unvisitedlist)
22         # (4) 标t己p为visited
23         vPoints.visit(p)
24         # (5) if p 的$\varepsilon$-邻域至少有MinPts个对象
25         # N是p的$\varepsilon$-邻域点列表
26         N = kd.query_ball_point(dataSet[p], eps)
27         if len(N) >= minPts:
28             # (6) 创建个一个新簇C,并把p添加到C
29             # 这里的C是一个标记列表,直接对第p个结点进行赋值
30             k += 1
31             C[p] = k
32             # (7) 令N为p的$\varepsilon$-邻域中的对象的集合
33             # N是p的$\varepsilon$-邻域点集合
34             # (8) for N中的每个点p'
35             for p1 in N:
36                 # (9) if p'是unvisited
37                 if p1 in vPoints.unvisitedlist:
38                     # (10) 标记p'为visited
39                     vPoints.visit(p1)
40                     # (11) if p'的$\varepsilon$-邻域至少有MinPts个点,把这些点添加到N
41                     # 找出p'的$\varepsilon$-邻域点,并将这些点去重新添加到N
42                     M = kd.query_ball_point(dataSet[p1], eps)
43                     if len(M) >= minPts:
44                         for i in M:
45                             if i not in N:
46                                 N.append(i)
47                     # (12) if p'还不是任何簇的成员,把p'添加到c
48                     # C是标记列表,直接把p'分到对应的簇里即可
49                     if C[p1] == -1
50                         C[p1] = k
51         # (15) else标记p为噪声
52         else:
53             C[p1] = -1
54 
55     # (16) until没有标记为unvisited的对象
56     return C

复制代码

 以代码3中生成的2500条数据作为测试,比较优化前后的算法性能

复制代码

1 import time
2 start = time.time()
3 C1 = my_dbscanl(X, 0.1, 10)
4 end = time.time()
5 print "`运行时间`:", end - start
6 plt.scatter(X[:, 0], X[:, 1], c=C1, marker='.')
7 plt.show()
8 >>> `运行时间:`29.1249849796

复制代码

图4:优化前算法结果

复制代码

1 import time
2 start = time.time()
3 C2 = my_dbscan2(X, 0.1, 10)
4 end = time.time()
5 print "运行时间:", end - start
6 plt.scatter(X[:, 0], X[:, 1], c=C2, marker='.')
7 plt.show()
8 >>> 运行时间:4.72340583801

复制代码

 

图5:优化后算法结果

可以看到优化后的算法运行时间从29.12s降到了4.72s,优化的效果非常明显。

4. 后记

上文仅仅是对DBSCAN算法的思想与实现进行了简略摘要,是学习算法的一个过程。算法的学习还比较粗劣和浅层,在实践应用中上述代码并不实用。如果需要使用DBSCAN的算法求解聚类问题,建议使用sklearn自带的DBSCAN函数。以代码3中生成数据为例:

复制代码

 1 # DBSCAN eps = 0.1, MinPts = 10
 2 import time
 3 from sklearn.cluster import DBSCAN
 4 start = time.time()
 5 C = DBSCAN(eps=0.1, min_pts=10).
 6 end = time.time()
 7 print "运行时间:", end - start
 8 plt.scatter(X[:, 0], X[:, 1], c=C, marker='.')
 9 plt.show()
10 >>> 运行时间:0.0240921974182

猜你喜欢

转载自blog.csdn.net/u011511601/article/details/81631893