arXiv · 2006.08177
Dissimilarity Mixture Autoencoder for Deep Clustering
Abstract
The dissimilarity mixture autoencoder (DMAE) is a neural network model for feature-based clustering that incorporates a flexible dissimilarity function and can be integrated into any kind of deep learning architecture. It internally represents a dissimilarity mixture model (DMM) that extends classical methods like K-Means, Gaussian mixture models, or Bregman clustering to any convex and differentiable dissimilarity function through the reinterpretation of probabilities as neural network representations. DMAE can be integrated with deep learning architectures into end-to-end models, allowing the simultaneous estimation of the clustering and neural network's parameters. Experimental evaluation was performed on image and text clustering benchmark datasets showing that DMAE is competitive in terms of unsupervised classification accuracy and normalized mutual information. The source code with the implementation of DMAE is publicly available at: https://github.com/juselara1/dmae
Explore related subjects
Keep this discovery
Juan S. Lara, Fabio A. González. 2020-06-15. Dissimilarity Mixture Autoencoder for Deep Clustering. https://arxiv.org/abs/2006.08177
Cite the original work for its findings. Save a collection to share your selection of sources.