Deep Subspace Image Clustering Network with Self-Expression and Self-Supervision

Chen Chao,Lu Hu,Wei Hui,Geng Xia
DOI: https://doi.org/10.1007/s10489-022-03654-6
IF: 5.3
2022-01-01
Applied Intelligence
Abstract:The subspace clustering algorithms for image datasets apply a self-expression coefficient matrix to obtain the correlation between samples and then perform clustering. However, such algorithms proposed in recent years do not use the cluster labels in the subspace to guide the deep network and do not get an end-to-end feature extraction and trainable clustering framework. In this paper, we propose a self-supervised subspace clustering model with a deep end-to-end structure, which is called Deep Subspace Image Clustering Network with Self-expression and Self-supervision (DSCNSS). The model embeds the self-supervised module into the subspace clustering. In network model training, alternating iterative optimization is applied to realize the mutual promotion of the self-supervised module and the subspace clustering module. Additionally, we design a new self-supervised loss function to improve the overall performance of the model further. To verify the performance of the proposed method, we conducted experimental tests on standard image datasets such as Extended Yale B, COIL20, COIL100, and ORL. The experimental results show that the performance of the proposed method is better than the existing traditional subspace clustering algorithm and deep clustering algorithm.
What problem does this paper attempt to address?