Research Area:  Machine Learning

Abstract:

We present a novel deep clustering algorithm that utilizes a variational autoencoder (VAE) framework with an entangled multi encoder-decoder neural architecture. Our model enforces a complementary structure that guides the learned latent representations towards a better space arrangement. It differs from previous VAE-based clustering algorithms by employing a new generative model that uses multiple encoder-decoders that are entangled to provide a joint clustering decision. The optimal clustering is found by optimizing a lower bound of the model likelihood function. Both the reconstruction component and the regularization component of the ELBO objective function are explicitly involved in the clustering procedure. We show that this modeling results in both better clustering capabilities and improved data generation. The proposed method is evaluated on standard datasets and is shown to significantly outperform state-of-the-art deep clustering methods.

Keywords:  

Author(s) Name:  Avi Caciularu , Jacob Goldberger

Publisher name:  ScienceDirect

DOI:  10.1016/j.neucom.2023.01.069

Volume Information:  Volume 529, Pages 182-189,(2023)

Paper Link:   https://www.sciencedirect.com/science/article/abs/pii/S0925231223001017