Clustering High-Dimensional Noisy Categorical Data

• 成果类型: Article
• 署名作者: Tian, Zhiyi; Xu, Jiaming; Tang, Jen
• 署名单位: IQVIA; Duke University; Purdue University System; Purdue University
• 刊物名称: JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION
• ISSN/ISSBN: 0162-1459
• DOI: 10.1080/01621459.2023.2298028
• 发表日期: 2024
• 页码: 3008-3019

摘要

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 E, we show that our algorithm exactly recovers the true clusters with high probability when mn(1-epsilon) >= CMr(2) log(3) M, with M = max(n,m) and a fixed constant C. In addition, we show that mn(1-epsilon)(2) >= r delta/2 with 0 < delta < 1 is necessary for any algorithm to succeed with probability at least (1 + delta)/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.