ALPCAHUS: Subspace Clustering for Heteroscedastic Data
Javier Salazar Cavazos, Jeffrey A Fessler, Laura Balzano
TL;DR
ALPCAHUS addresses subspace clustering under heteroscedastic noise by jointly learning per-sample noise variances and subspace bases for each cluster. It extends LR-ALPCAH to the union-of-subspaces setting and incorporates an ensemble strategy to stabilize clustering via co-association affinities, with convergence guarantees and adaptive rank estimation. Empirical results on synthetic and real data (quasars and Indian Pines hyperspectral imagery) show that ALPCAHUS often outperforms conventional methods, approaching the performance of a noisy oracle, especially when data quality varies across samples. The method demonstrates practical robustness to heteroscedasticity and highlights future directions for manifold extensions and feature-space heteroscedastic models.
Abstract
Principal component analysis (PCA) is a key tool in the field of data dimensionality reduction. Various methods have been proposed to extend PCA to the union of subspace (UoS) setting for clustering data that comes from multiple subspaces like K-Subspaces (KSS). However, some applications involve heterogeneous data that vary in quality due to noise characteristics associated with each data sample. Heteroscedastic methods aim to deal with such mixed data quality. This paper develops a heteroscedastic-based subspace clustering method, named ALPCAHUS, that can estimate the sample-wise noise variances and use this information to improve the estimate of the subspace bases associated with the low-rank structure of the data. This clustering algorithm builds on K-Subspaces (KSS) principles by extending the recently proposed heteroscedastic PCA method, named LR-ALPCAH, for clusters with heteroscedastic noise in the UoS setting. Simulations and real-data experiments show the effectiveness of accounting for data heteroscedasticity compared to existing clustering algorithms. Code available at https://github.com/javiersc1/ALPCAHUS.
