🤖 AI Summary
Determining the optimal number of clusters in high-dimensional, noisy data remains a fundamental challenge in unsupervised learning. To address this, we propose HD-BWDM, a robust nonparametric evaluation framework. HD-BWDM generalizes the Between-Within Distance Metric (BWDM) criterion to high-dimensional spaces by integrating random projection and PCA-based dimensionality reduction to mitigate the curse of dimensionality, and incorporates cluster trimming and center-point distance metrics to enhance robustness against outliers. Theoretically, it provides embedding consistency guarantees grounded in the Johnson–Lindenstrauss lemma. Extensive experiments on multiple high-dimensional and contaminated benchmark datasets demonstrate that HD-BWDM significantly outperforms conventional centroid-based indices—including Calinski–Harabasz and Silhouette—achieving superior stability, interpretability, and reliability. As a theoretically justified, efficient, and robust stopping criterion, HD-BWDM advances practical unsupervised clustering in challenging real-world scenarios.
📝 Abstract
Determining the appropriate number of clusters in unsupervised learning is a central problem in statistics and data science. Traditional validity indices such as Calinski-Harabasz, Silhouette, and Davies-Bouldin-depend on centroid-based distances and therefore degrade in high-dimensional or contaminated data. This paper proposes a new robust, nonparametric clustering validation framework, the High-Dimensional Between-Within Distance Median (HD-BWDM), which extends the recently introduced BWDM criterion to high-dimensional spaces. HD-BWDM integrates random projection and principal component analysis to mitigate the curse of dimensionality and applies trimmed clustering and medoid-based distances to ensure robustness against outliers. We derive theoretical results showing consistency and convergence under Johnson-Lindenstrauss embeddings. Extensive simulations demonstrate that HD-BWDM remains stable and interpretable under high-dimensional projections and contamination, providing a robust alternative to traditional centroid-based validation criteria. The proposed method provides a theoretically grounded, computationally efficient stopping rule for nonparametric clustering in modern high-dimensional applications.