【十大經典數據挖掘算法】系列
1. 引言
k-means與kNN雖然都是以k打頭,但卻是兩類算法——kNN為監督學習中的分類算法,而k-means則是非監督學習中的聚類算法;二者相同之處:均利用近鄰信息來標注類別。
聚類是數據挖掘中一種非常重要的學習流派,指將未標注的樣本數據中相似的分為同一類,正所謂“物以類聚,人以群分”嘛。k-means是聚類算法中最為簡單、高效的,核心思想:由用戶指定k個初始質心(initial centroids),以作為聚類的類別(cluster),重復迭代直至算法收斂。
2. 基本算法
在k-means算法中,用質心來表示cluster;且容易證明k-means算法收斂等同於所有質心不再發生變化。基本的k-means算法流程如下:
選取k個初始質心(作為初始cluster);
repeat:
對每個樣本點,計算得到距其最近的質心,將其類別標為該質心所對應的cluster;
重新計算k個cluser對應的質心;
until 質心不再發生變化
對於歐式空間的樣本數據,以平方誤差和(sum of the squared error, SSE)作為聚類的目標函數,同時也可以衡量不同聚類結果好壞的指標:
表示樣本點\(x\)到cluster \(C_i\) 的質心 \(c_i\) 距離平方和;最優的聚類結果應使得SSE達到最小值。
下圖中給出了一個通過4次迭代聚類3個cluster的例子:
k-means存在缺點:
-
k-means是局部最優的,容易受到初始質心的影響;比如在下圖中,因選擇初始質心不恰當而造成次優的聚類結果(SSE較大):
-
同時,k值的選取也會直接影響聚類結果,最優聚類的k值應與樣本數據本身的結構信息相吻合,而這種結構信息是很難去掌握,因此選取最優k值是非常困難的。
3. 優化
為了解決上述存在缺點,在基本k-means的基礎上發展而來二分 (bisecting) k-means,其主要思想:一個大cluster進行分裂后可以得到兩個小的cluster;為了得到k個cluster,可進行k-1次分裂。算法流程如下:
初始只有一個cluster包含所有樣本點;
repeat:
從待分裂的clusters中選擇一個進行二元分裂,所選的cluster應使得SSE最小;
until 有k個cluster
上述算法流程中,為從待分裂的clusters中求得局部最優解,可以采取暴力方法:依次對每個待分裂的cluster進行二元分裂(bisect)以求得最優分裂。二分k-means算法聚類過程如圖:
從圖中,我們觀察到:二分k-means算法對初始質心的選擇不太敏感,因為初始時只選擇一個質心。
4. 參考資料
[1] Pang-Ning Tan, Michael Steinbach, Vipin Kumar, Introduction to Data Mining.
[2] Xindong Wu, Vipin Kumar, The Top Ten Algorithms in Data Mining.