作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程

• 人工智能及识别技术 • 上一篇    下一篇

结合mean-shift与MST的K-means聚类算法

(安徽大学计算智能与信号处理教育部重点实验室,合肥 230039)   

  1. (安徽大学计算智能与信号处理教育部重点实验室,合肥 230039)
  • 收稿日期:2012-10-29 出版日期:2013-12-15 发布日期:2013-12-13
  • 作者简介:徐 沁(1983-),女,博士研究生,主研方向:机器视觉,模式识别;罗 斌,教授、博士
  • 基金资助:
    国家自然科学基金资助项目(61073116, 61211130309)

K-means Clustering Algorithm Combined with mean-shift and Minimum Spanning Tree

(Key Laboratory of Intelligence Computing and Signal Processing, Ministry of Education, Anhui University, Hefei 230039, China)   

  1. (Key Laboratory of Intelligence Computing and Signal Processing, Ministry of Education, Anhui University, Hefei 230039, China)
  • Received:2012-10-29 Online:2013-12-15 Published:2013-12-13

摘要: 针对初始点选择不当导致K-means陷入局部最小值问题,提出一种结合自适应mean-shift与最小生成树(MST)的K-means聚类算法。将数据对象投影到主成分分析(PCA)子空间,给出自适应mean-shift算法,并在PCA子空间内将数据向密度大的区域聚集,再利用MST与图连通分量算法,找出数据的类别数和类标签,据此计算原始空间的密度峰值,并将其作为K-means聚类的初始中心点。对K-means的目标函数、聚类精度和运行时间进行比较,结果表明,该算法在较短的运行时间内能给出较优的全 局解。

关键词: 聚类分析, K-means算法, 初始中心, Mean-Shift算法, 主成分分析, 最小生成树

Abstract: Given an inappropriate set of initial clustering centroids, K-means algorithm can get trapped in a local minimum. To remedy this, this paper proposes a K-means clustering algorithm combined with adaptive mean-shift and Minimum Spanning Tree(MST). The original data set is projected into Principal Component Analysis(PCA) subspace. An adaptive Mean-shift is proposed and run in the PCA subspace to let the data move to dense regions, and via the MST and graph connected component algorithm, it finds the number of clusters and the cluster indicators. According to the indicators, the density peaks are computed in the full space and taken as the initial centroids for K-means clustering. Experimental results show that the proposed algorithm can provide better global solution and higher clustering accuracy within a shorter period of execution time.

Key words: clustering analysis, K-means algorithm, initial centroid, Mean-Shift algorithm, Principal Component Analysis(PCA), Minimum Spanning Tree(MST)

中图分类号: