.png){kind=icon}
返回
Wasserstein Embedding Learning for Deep Clustering: A Generative Approach
.png){kind=icon}
PRE
AI
DOI：10.1109/TMM.2024.3369862.png)
.png){kind=icon}
.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Deep learning-based clustering methods, especially those incorporating deep generative models, have recently shown noticeable improvement on many multimedia benchmark datasets. However, existing generative models still suffer from unstable training, and the gradient vanishes, which results in the inability to learn desirable embedded features for clustering. In this paper, we aim to tackle this problem by exploring the capability of Wasserstein embedding in learning representative embedded features and introducing a new clustering module for jointly optimizing embedding learning and clustering. To this end, we propose Wasserstein embedding clustering (WEC), which integrates robust generative models with clustering. By directly minimizing the discrepancy between the prior and marginal distribution, we transform the optimization problem of Wasserstein distance from the original data space into embedding space, which differs from other generative approaches that optimize in the original data space. Consequently, it naturally allows us to construct a joint optimization framework with the designed clustering module in the embedding layer. Due to the substitutability of the penalty term in Wasserstein embedding, we further propose two types of deep clustering models by selecting different penalty terms. Comparative experiments conducted on nine publicly available multimedia datasets with several state-of-the-art methods demonstrate the effectiveness of our method.
Keyword:
Training
Data models
Generative adversarial networks
Clustering methods
Task analysis
Deep learning
Decoding
Unsupervised learning
clustering analysis
Wasserstein embedding
generative models
auto-encoder
期刊
IF:
9.7
论文数:
4.7K
被引数:
2.4W
机构
引用论文
分享收藏Unsupervised deep clustering via contractive feature representation and focal loss.png)
PATTERN RECOGNITION
PATTERN RECOGNITIONIF7.6
PREAI
分享收藏FPGAN: Face de-identification method with generative adversarial networks for social robotsFPGAN: 社交机器人的生成对抗网络人脸去识别方法NEURAL NETWORKS
NEURAL NETWORKSIF6.3
PREAI
分享收藏
查看更多内容