Masked Subspace Clustering Methods
Jiebo Song, Huaming Ling
TL;DR
The paper tackles clustering in high-dimensional data by learning pairwise relations through masks and fuse two clustering objectives using a general Bilevel Clustering Optimization (BCO) framework. It introduces three masked subspace clustering variants—BMSC with a hard mask, GMSC with a soft mask, and RMSC with a learnable soft mask and recursive updates—validating them on standard datasets with significant performance gains. The approach leverages a bilevel structure where $M^s=\psi(y)$ guides the outer clustering objective, and the inner clustering informs mask construction, leading to refined affinity matrices. Collectively, the work demonstrates that learnable, recursively updated masks can substantially improve clustering accuracy and stability in high-dimensional settings with minimal supervision.
Abstract
To further utilize the unsupervised features and pairwise information, we propose a general Bilevel Clustering Optimization (BCO) framework to improve the performance of clustering. And then we introduce three special cases on subspace clustering with two different types of masks. At first, we reformulate the original subspace clustering as a Basic Masked Subspace Clustering (BMSC), which reformulate the diagonal constraints to a hard mask. Then, we provide a General Masked Subspace Clustering (GMSC) method to integrate different clustering via a soft mask. Furthermore, based on BCO and GMSC, we induce a learnable soft mask and design a Recursive Masked Subspace Clustering (RMSC) method that can alternately update the affinity matrix and the soft mask. Numerical experiments show that our models obtain significant improvement compared with the baselines on several commonly used datasets, such as MNIST, USPS, ORL, COIL20 and COIL100.
