Deep Clustering Via Weighted $K$-Subspace Network

Weitian Huang,Ming Yin,Jianzhong Li,Shengli Xie
DOI: https://doi.org/10.1109/lsp.2019.2941368
2019-01-01
IEEE Signal Processing Letters
Abstract:Subspace clustering aims to separate the data into clusters under the hypothesis that the samples within the same cluster will lie in the same low-dimensional subspace. Due to the tough pairwise constraints, $k$-subspace clustering is sensitive to outliers and initialization. In this letter, we present a novel deep architecture for $k$-subspace clustering to address this issue, called as Deep Weighted $k$-Subspace Clustering (DWSC). Specifically, our framework consists of autoencoder and weighted $k$-subsapce network. We first use the autoencoder to non-linearly compress the samples into the low-dimensional latent space. In the weighted $k$-subspace network, we feed the latent representation into the assignment network to output soft assignments which represent the probability of data belonging to the according subspace. Subsequently, the optimal $k$ subspaces are identified by minimizing the projection residuals of the latent representations to all subspaces, using the learned soft assignments as a weighting vector. Finally, we jointly optimize the representation learning and clustering in a unified framework. Experimental results show that our approach outperforms the state-of-the-art subspace clustering methods on two benchmark datasets.
What problem does this paper attempt to address?