End-to-end unsupervised clustering neural networks for image clustering.pdf (1.19 MB)
Download file

End-to-end unsupervised clustering neural networks for image clustering

Download (1.19 MB)
posted on 2023-02-27, 19:32 authored by HuaShun LiHuaShun Li, JingZhao Li, Meng Zhu

In this paper, we propose a new clustering module that can be trained jointly with existing neural network layers.  Specifically, we have designed a generic clustering module with a competitive update mechanism. The module consists of a Gaussian unit and a maximum pooling layer. The Gaussian unit forward propagation conforms to the joint Gaussian distribution and contains two sets of trainable parameters. It requires no tedious setup and has a plug-and-play feature. To improve the  representation capability of the network, we used an auto-encoder to extract the hidden semantics of the input features and combined the clustering module with the auto-encoder to construct an end-to-end unsupervised clustering neural network. We refer to this as HGL_CAE(High-dimensional Gaussian distribution layers combined with convolutional autoencoders).The network is highly adaptable to different input feature dimensions and can cope with situations where the number of clusters  cannot be determined in advance. We conducted experiments on the MNIST dataset and the Fashion_MNIST dataset with a clustering accuracy of 93.38% and 72.83% respectively. It is highly competitive with existing methods. 


Email Address of Submitting Author

ORCID of Submitting Author


Submitting Author's Institution

Anhui University of Science and Technology

Submitting Author's Country

  • China

Usage metrics