English

A High-Performance External Validity Index for Clustering with a Large Number of Clusters

Data Structures and Algorithms 2024-09-24 v1 Computer Science and Game Theory Machine Learning

Abstract

This paper introduces the Stable Matching Based Pairing (SMBP) algorithm, a high-performance external validity index for clustering evaluation in large-scale datasets with a large number of clusters. SMBP leverages the stable matching framework to pair clusters across different clustering methods, significantly reducing computational complexity to O(N2)O(N^2), compared to traditional Maximum Weighted Matching (MWM) with O(N3)O(N^3) complexity. Through comprehensive evaluations on real-world and synthetic datasets, SMBP demonstrates comparable accuracy to MWM and superior computational efficiency. It is particularly effective for balanced, unbalanced, and large-scale datasets with a large number of clusters, making it a scalable and practical solution for modern clustering tasks. Additionally, SMBP is easily implementable within machine learning frameworks like PyTorch and TensorFlow, offering a robust tool for big data applications. The algorithm is validated through extensive experiments, showcasing its potential as a powerful alternative to existing methods such as Maximum Match Measure (MMM) and Centroid Ratio (CR).

Keywords

Cite

@article{arxiv.2409.14455,
  title  = {A High-Performance External Validity Index for Clustering with a Large Number of Clusters},
  author = {Mohammad Yasin Karbasian and Ramin Javadi},
  journal= {arXiv preprint arXiv:2409.14455},
  year   = {2024}
}

Comments

16 pages, 14 tables