loading page

End-to-end unsupervised clustering neural networks for image clustering
  • HuaShun Li ,
  • JingZhao Li ,
  • Meng Zhu
HuaShun Li
Anhui University of Science and Technology, Anhui University of Science and Technology

Corresponding Author:[email protected]

Author Profile
JingZhao Li
Author Profile

Abstract

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.