arrow
Return

Adaptive Initialization Method for K-Means Algorithm

delete2021-11-25
delete8
delete
OA
AI
J
Jie Yang
Y
Yu–Kai Wang
X
Xin Yao
C
Chin‐Teng Lin *
DOI:10.3389/frai.2021.740817delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
The K-means algorithm is a widely used clustering algorithm that offers simplicity and efficiency. However, the traditional K-means algorithm uses a random method to determine the initial cluster centers, which make clustering results prone to local optima and then result in worse clustering performance. In this research, we propose an adaptive initialization method for the K-means algorithm (AIMK) which can adapt to the various characteristics in different datasets and obtain better clustering performance with stable results. For larger or higher-dimensional datasets, we even leverage random sampling in AIMK (name as AIMK-RS) to reduce the time complexity. 22 real-world datasets were applied for performance comparisons. The experimental results show AIMK and AIMK-RS outperform the current initialization methods and several well-known clustering algorithms. Specifically, AIMK-RS can significantly reduce the time complexity to O (n). Moreover, we exploit AIMK to initialize K-medoids and spectral clustering, and better performance is also explored. The above results demonstrate superior performance and good scalability by AIMK or AIMK-RS. In the future, we would like to apply AIMK to more partition-based clustering algorithms to solve real-life practical problems.
Keywords:
k-means
adaptive
initialization method
initial cluster centers
clustering
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

F
Frontiers in Artificial Intelligence
IF:
4.7
Papers:
2.3K
Citations:
4.4K

Organization

U
university of technology sydney
Scholars:
1.6W
Papers: 2.0W
Citations: 25