Scalable Deep Subspace Clustering Network
Nairouz Mrabah, Mohamed Bouguessa, Sihem Sami
TL;DR
This work tackles the cubic time bottleneck of deep subspace clustering by introducing SDSNet, a scalable framework that uses landmark-based low-rank factorization $\mathbf{C} \approx \mathbf{P}\mathbf{P}^\top$ and a convolutional auto-encoder to learn robust latent representations. Clustering is performed via a reduced eigenproblem derived from the anchor-based affinity, yielding overall complexity $O(n)$ with a fixed anchor count $m$. The method jointly optimizes reconstruction and subspace constraints through a block-coordinate scheme, leveraging Procrustes updates and closed-form anchor computations. Experiments on five real-world datasets show SDSNet achieves competitive accuracy while significantly reducing computation, demonstrating practical scalability for large-scale subspace clustering.
Abstract
Subspace clustering methods face inherent scalability limits due to the $O(n^3)$ cost (with $n$ denoting the number of data samples) of constructing full $n\times n$ affinities and performing spectral decomposition. While deep learning-based approaches improve feature extraction, they maintain this computational bottleneck through exhaustive pairwise similarity computations. We propose SDSNet (Scalable Deep Subspace Network), a deep subspace clustering framework that achieves $\mathcal{O}(n)$ complexity through (1) landmark-based approximation, avoiding full affinity matrices, (2) joint optimization of auto-encoder reconstruction with self-expression objectives, and (3) direct spectral clustering on factorized representations. The framework combines convolutional auto-encoders with subspace-preserving constraints. Experimental results demonstrate that SDSNet achieves comparable clustering quality to state-of-the-art methods with significantly improved computational efficiency.
