Clustering High-Dimensional Noisy Categorical Data

Zhiyi Tian,Jiaming Xu,Jen Tang
DOI: https://doi.org/10.1080/01621459.2023.2298028
IF: 4.369
2024-01-18
Journal of the American Statistical Association
Abstract:Clustering is a widely used unsupervised learning technique that groups data into homogeneous clusters. However, when dealing with real-world data that contain categorical values, existing algorithms can be computationally costly in high dimensions and can struggle with noisy data that has missing values. Furthermore, except for one algorithm, no others provide theoretical guarantees of clustering accuracy. In this article, we propose a general categorical data encoding method and a computationally efficient spectral-based algorithm to cluster high-dimensional noisy categorical data (nominal or ordinal). Under a statistical model for data on m attributes from n subjects in r clusters with missing probability ε , we show that our algorithm exactly recovers the true clusters with high probability when mn(1−ε)≥CMr2 log 3M , with M=max(n,m) and a fixed constant C . In addition, we show that mn(1−ε)2≥rδ/2 with 0<δ<1 is necessary for any algorithm to succeed with probability at least (1+δ)/2 . In cases where m = n and r are fixed, the sufficient condition matches with the necessary condition up to a polylog(n) factor. In numerical studies our algorithm outperforms several existing algorithms in both clustering accuracy and computational efficiency. Supplementary materials for this article are available online.
statistics & probability
What problem does this paper attempt to address?