IAE-ClusterGAN: A new Inverse autoencoder for Generative Adversarial Attention Clustering network

Clustering is a challenging and crucial task in unsupervised learning. Recently, though many clustering algorithms combined with deep learning have been proposed, we observe that the existing deep clustering algorithms do not considerably preserve the clustering structure and information of raw data...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:Neurocomputing (Amsterdam) Ročník 465; s. 406 - 416
Hlavní autoři: Ling, Chao, Cao, Guitao, Cao, Wenming, Wang, Hong, Ren, He
Médium: Journal Article
Jazyk:angličtina
Vydáno: Elsevier B.V 20.11.2021
Témata:
Attention mechanism
Clustering
Inverse autoencoder
Latent code
Unsupervised learning
Attention mechanism
Latent code
Inverse autoencoder
Clustering
Unsupervised learning
ISSN:0925-2312, 1872-8286
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí:Clustering is a challenging and crucial task in unsupervised learning. Recently, though many clustering algorithms combined with deep learning have been proposed, we observe that the existing deep clustering algorithms do not considerably preserve the clustering structure and information of raw data in the learned latent space. To address this issue, we propose a Generative Adversarial Attention Clustering network Based on Inverse autoencoder (IAE-ClusterGAN), which can control the distribution type of the learned latent code without additional constraints so that unsupervised clustering tasks can be done efficiently. Meanwhile, we integrate the attention mechanism into the network to make the latent code contain more useful clustering information. Moreover, we utilize hyperspherical mapping in the discriminator to improve the stability of model training and reduce the training parameters. Experimental results demonstrate that IAE-ClusterGAN achieves competitive results compared to the state-of-the-art models on five benchmark datasets.
ISSN:0925-2312
1872-8286
DOI:10.1016/j.neucom.2021.08.128