Title | Deep clustering analysis via variational autoencoder with Gamma mixture latent embeddings |
Creator | |
Date Issued | 2025-03-01 |
Source Publication | Neural Networks
![]() |
ISSN | 0893-6080 |
Volume | 183 |
Abstract | This article proposes a novel deep clustering model based on the variational autoencoder (VAE), named GamMM-VAE, which can learn latent representations of training data for clustering in an unsupervised manner. Most existing VAE-based deep clustering methods use the Gaussian mixture model (GMM) as a prior on the latent space. We employ a more flexible asymmetric Gamma mixture model to achieve higher quality embeddings of the data latent space. Second, since the Gamma is defined for strictly positive variables, in order to exploit the reparameterization trick of VAE, we propose a transformation method from Gaussian distribution to Gamma distribution. This method can also be considered a Gamma distribution reparameterization trick, allows gradients to be backpropagated through the sampling process in the VAE. Finally, we derive the evidence lower bound (ELBO) based on the Gamma mixture model in an effective way for the stochastic gradient variational Bayesian (SGVB) estimator to optimize the proposed model. ELBO, a variational inference objective, ensures the maximization of the approximation of the posterior distribution, while SGVB is a method used to perform efficient inference and learning in VAEs. We validate the effectiveness of our model through quantitative comparisons with other state-of-the-art deep clustering models on six benchmark datasets. Moreover, due to the generative nature of VAEs, the proposed model can generate highly realistic samples of specific classes without supervised information. |
Keyword | Clustering Data augmentation Gamma mixture models VAE Variational inference |
DOI | 10.1016/j.neunet.2024.106979 |
URL | View source |
Language | 英语English |
Scopus ID | 2-s2.0-85211339866 |
Citation statistics | |
Document Type | Journal article |
Identifier | http://repository.uic.edu.cn/handle/39GCC9TT/12502 |
Collection | Beijing Normal-Hong Kong Baptist University |
Corresponding Author | Guo,Jiaxun |
Affiliation | 1.CIISE,Concordia University,Montreal,H3G 1T7,Canada 2.Guangdong Provincial Key Laboratory IRADS and Department of Computer Science,Beijing Normal University-Hong Kong Baptist University United International College,Zhuhai,Guangdong,China |
Recommended Citation GB/T 7714 | Guo,Jiaxun,Fan,Wentao,Amayri,Manaret al. Deep clustering analysis via variational autoencoder with Gamma mixture latent embeddings[J]. Neural Networks, 2025, 183. |
APA | Guo,Jiaxun, Fan,Wentao, Amayri,Manar, & Bouguila,Nizar. (2025). Deep clustering analysis via variational autoencoder with Gamma mixture latent embeddings. Neural Networks, 183. |
MLA | Guo,Jiaxun,et al."Deep clustering analysis via variational autoencoder with Gamma mixture latent embeddings". Neural Networks 183(2025). |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment