CavMerge: Merging K-means Based on Local Log-Concavity
Organizations: Department of Statistics, Iowa State University
Abstract
K-means clustering, a classic and widely-used clustering technique, is known to exhibit suboptimal performance when applied to non-linearly separable data. Numerous adjustments and modifications have been proposed to address this issue, including methods that merge K-means results from a relatively large K to obtain a final cluster assignment. However, existing methods of this nature often encounter computational inefficiencies and suffer from hyperparameter tuning. Here we present CavMerge, a novel K-means merging algorithm that is intuitive, free of parameter tuning, and computationally efficient. Operating under minimal local distributional assumptions, our algorithm demonstrates strong consistency and rapid convergence guarantees. Empirical studies on various simulated and real datasets demonstrate that our method yields more reliable clusters in comparison to current state-of-the-art algorithms.
Figures & tables
| Data | CavMerge | SK | DBSCAN | k-mH | KNOB | HC |
|---|---|---|---|---|---|---|
| Aggregation | 0.990 | |||||
| (0.013) | 0.841 | |||||
| (0.098) | 0.809 | |||||
| (0) | 0.884 | |||||
| (0.102) | 0.812 | |||||
| (0.042) | 0.804 |
| Data | CavMerge | SK | DBSCAN | k-mH | KNOB | HC |
| Digits | 0.720 | |||||
| (0.046) | 0.551 | |||||
| (0.082) | 0.048 | |||||
| (0) | - | |||||
| (-) | 0.128 | |||||
| (0.098) | 0 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Data | CavMerge | SK | DBSCAN | k-mH | KNOB | HC |
|---|---|---|---|---|---|---|
| Aggregation | 0.209 | 0.195 | 0.015 | 30.4 | 3.39 | 0.017 |
| Banana-Clump | 0.048 | 0.026 | 0.001 | 5.07 | 1.70 | 0.002 |
| Bananas-Arcs | 0.427 | 0.349 | 0.019 | 31.2 | 14.4 | 0.627 |
| Bullseye | 0.074 | 0.033 | 0.003 | 15.2 | 2.22 | 0.006 |
| Bullseye-Cigarette | 0.368 | 0.214 | 0.015 | 16.8 | 11.1 | 0.350 |
| Compound | 0.065 | 0.047 | 0.002 | 14.4 | 2.41 | 0.005 |
Clustering
Clustering groups data points by shared structure, developing methods to discover, validate, and interpret groups across diverse data and constraints.
See all 342 papers