基于GSA算法改进的K均值聚类

为改善K均值聚类存在的对初始聚心敏感、全局搜索能力弱和凭经验确定聚类数等不足,提出一种基于GSA算法的改进K均值聚类。采用粒子编码策略,把聚类中心集合视作种群粒子,引入GSA搜索聚类质量最好的初始聚类中心,设均方误差为适应度函数,引导全局搜索方向,设置种群成熟度因子避免算法陷入局部最优,引入聚类质量评价指标获取最佳聚类数。通过在4种UCI数据集上做仿真测试,验证了改进后K均值聚类具有较高的正确率和更好的稳定性。

Improved K-means clustering based on GSA algorithm

To improve the shortcomings of K-means clustering,such as sensitivity to initial concentration,weak global search ability and empirical determination of clustering number,an improved K-means clustering based on gravitational search algorithm(GSA)was proposed.Using particle encoding strategy,cluster centers were regarded as population particles,and GSA was introduced to search the initial cluster centers with the best clustering quality.Mean square error was set as fitness function to guide the global search direction.Population maturity factor was set to avoid the algorithm falling into local optimum.Clustering quality evaluation index was added to obtain the best clustering number.The simulation tests on four UCI datasets show that the improved K-means clustering has higher accuracy and better stability.