Subspace Clustering Based on a Multichannel Attention Mechanism
Yuxi Zhao,Longge Wang,Junyang Yu,Fang Zuo,Tingyu Wang,Zhicheng Wang,Han Li
DOI: https://doi.org/10.1007/s13042-023-01975-7
2024-01-01
International Journal of Machine Learning and Cybernetics
Abstract:Existing self-representation models based on multilayer perceptrons (MLPs) have gained widespread attention for their outstanding clustering performance in subspace clustering. However, when images contain rich spatial information, the use of fully connected neural networks that only accept vector inputs results in a significant loss of spatial information, thereby greatly reducing the clustering performance of the models. To address the clustering problem of data with rich spatial information, this paper proposes a multichannel subspace clustering method based on a self-representation network (CGSNet). This method incorporates modules capable of mining spatial features into the self-representation network to highlight different data characteristics and supplement the spatial information lost by the MLP. CGSNet successfully uncovers the latent features within the input data samples and extracts the spatial relationships among different image features by employing channel and spatial attention modules. Additionally, static parameterized channel mapping and spatial mapping are used to refine and filter the obtained spatial information, further enhancing the quality of self-representation. Finally, by leveraging the self-representation network, the clustering task is completed by learning the affinity matrix. The experimental results demonstrate that CGSNet outperforms the self-expressive network (SENet), achieving improvements of 1.5%, 3.3%, 0.9%, and 4.4% in terms of the clustering accuracy with the MNIST, FashionMNIST, CIFAR-10, and EMNIST datasets, respectively. CGSNet achieves the highest accuracy among competitive clustering methods, including EnSC, SENet, and 14 others.