k-means聚类算法原理简介
概要
K-means算法是最普及的聚类算法,也是一个比较简单的聚类算法。
算法接受一个未标记的数据集,然后将数据聚类成不同的组,同时,k-means算法也是一种无监督学习。
算法思想
k-means算法的思想比较简单,假设我们要把数据分成K个类,大概可以分为以下几个步骤:
1.随机选取k个点,作为聚类中心;
2.计算每个点分别到k个聚类中心的聚类,然后将该点分到最近的聚类中心,这样就行成了k个簇;
3.再重新计算每个簇的质心(均值);
4.重复以上2~4步,直到质心的位置不再发生变化或者达到设定的迭代次数。
算法流程图解
下面我们通过一个具体的例子来理解这个算法(我这里用到了Andrew Ng的机器学习教程中的图):
假设我们首先拿到了这样一个数据,要把它分成两类:
我们人眼当然可以很快的分辨出来,可以在两个聚类间找到一条合理的分界线,
那么用k-means算法来解决这个问题会是怎样的呢?
首先我们随机选取两个点作为聚类中心(因为已经明确是分为两类):
接下来就可以开始计算每个点到红点和蓝点的距离了,离红点近就标记为红色,离蓝点近就标记为蓝色。结果为下图:
很明显,这样完全不是我们想要的结果,接下来我们进行第三步,重新计算聚类中心的位置。
红X和蓝X都向中间靠拢了一点。
我们可以看到,聚类中心发生改变后,其他点离两个聚类中心的距离也跟随着发生了变化。
然后我们重复第二步,根据每个点到两个聚类中心的距离远近来进行重新分类,
离红X近的归为红类,离蓝X近的归为蓝类。
之前站错了队伍的一些点重新进行了调整,现在的分类离我们的目标越来越近了,
但还没有达到最佳的分类效果。接下来继续重复上面的步骤,重新计算聚类中心的位置,
再重新分类,不断迭代,直至聚类中心的位置不再变化(变化范围达到设定值)或达到迭代次数为止。
这样我们就利用k-means算法把这个数据很好的分为两类啦。
我们可以看到,在整个过程中,我们都没有去监督算法,告诉他具体是分错了还是对了,
只是在开始的时候告诉他要把这个数据分成多少类,然后后面的操作都是由他自己完成,
完全没有人为的让他进行分类的学习,也没有帮助他纠正错误,所以k-means算法也是一种无监督学习方法。
相信看到这里你对k-means算法的原理也有了一个大概的了解啦。
代价函数(Distortion function)
要是k-means最后的分类结果最好,也就是要是K-均值最小化,
是要最小化所有的数据点与其所关联的聚类中心点之间的距离之和,
因此我们可以设计 K-均值的代价函数(又称 畸变函数 Distortion function)为:
其中μ c (i)代表与 x (i) 最近的聚类中心点。
我们的的优化目标也就是要找出使得代价函数最小的 c (1) ,c (2) ,…,c (m) 和μ 1 ,μ 2 ,…,μ k 。
我们再回顾一下刚才给出的 K-means算法的迭代过程,我们知道,第一个步骤(根据聚类中心分类)
是用于减小 c (i) 引起的代价,而第二个步骤(重新定位聚类中心)则是用于减小μ i 引起的代价。
所以迭代的过程一定会是每一次迭代都在减小代价函数,如果发生迭代之后代价函数反而增加,则很可能是出现了错误。
如何选取k值
对于一个给定没有分类的数据集,最后具体应该分为多少类呢?
这确实是一个问题,比如我们前面的那个例子,通过人眼观察,很明显可以分为两类,那么我们选取K值为2,
可以得到一个比较好的聚类结果。那如果K选择3或者其他值行不行呢?当然也可以。
比如下图,有一个数据经过统计之后发现随着K值的增加其畸变函数在不断变小,但是我们发现在k=3时,
畸变函数随着k值变化的幅度显著降低,在k>3之后所带来的好处并不是特别明显,
所以我们可以选择k=3作为我们的聚类数目。由于其形状像我们人类的肘部,我们也称其为“肘部法则”。
但是实际应用中,k值的变换规律都不是和上图一样存在突变点,多数情况如下图所示:
随着k值的增大,其畸变函数也随着不断减小,根本就不存在明显的拐点。
那么这种情况,K值的选择主要还是根据经验以及利用k-means聚类的目的来决定。
聚类中心的初始化
前面提到代价函数的建立,可以方便我们来对k-means算法的结果进行优化,
方便我们察觉出算法迭代过程中的收敛问题,是否达到局部最小化,或者检查算法迭代过程中是否出现问题。
而通过k值的选取也可能使我们的代价函数尽量的减小,以得到更好的聚类效果。那么还有什么其他优化的方法呢?
我们还可以通过选取更优的聚类中心来优化聚类效果。
上面的例图是一个简单的二分类问题,并且不同类之间的界限也比较明显,最后我们得到的聚类结果也相对比较理想。但实际上聚类中心选择的不同,最终的聚类结果肯定也是会不一样的。
比如对于下面这张图:
我把初始的聚类中心选择在这两个不同的位置,最后导致的分类结果和迭代次数都会不一样。
聚类中心的选取主要还是以随机为主,并且初始的时候最好是选择数据中的点。
对于多分类,甚至还会出现下面这种情况:
最后很可能仅仅实现了局部最优,把本来不是一类的多个类分为了一类,或者把本来是一类的分成了多个类,
这些都是有可能的。那么对于这种情况怎么办呢?
对于聚类数目K值较小(K<10)的情况下,我们可以多次随机选取不同聚类中心,最后比较各自迭代完成后的畸变函数值,
畸变函数越小,则说明聚类效果更优。但是在k值较大的情况下,比如上百类甚至上千万类,
这时候重新选取不同的聚类中心可能就没有很好的效果了。