arrow
返回

A Decoder-Free Variational Deep Embedding for Unsupervised Clustering

delete2022-10-01
delete15
PRE
AI
Q
Qiang Ji
孙
孙艳丰 (Yanfeng Sun)
Junbin Gao 封面图
Junbin Gao (Junbin Gao)
Y
Yongli Hu
B
Baocai Yin *
DOI:10.1109/TNNLS.2021.3071275delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
In deep clustering frameworks, autoencoder (AE)- or variational AE-based clustering approaches are the most popular and competitive ones that encourage the model to obtain suitable representations and avoid the tendency for degenerate solutions simultaneously. However, for the clustering task, the decoder for reconstructing the original input is usually useless when the model is finished training. The encoder-decoder architecture limits the depth of the encoder so that the learning capacity is reduced severely. In this article, we propose a decoder-free variational deep embedding for unsupervised clustering (DFVC). It is well known that minimizing reconstruction error amounts to maximizing a lower bound on the mutual information (MI) between the input and its representation. That provides a theoretical guarantee for us to discard the bloated decoder. Inspired by contrastive self-supervised learning, we can directly calculate or estimate the MI of the continuous variables. Specifically, we investigate unsupervised representation learning by simultaneously considering the MI estimation of continuous representations and the MI computation of categorical representations. By introducing the data augmentation technique, we incorporate the original input, the augmented input, and their high-level representations into the MI estimation framework to learn more discriminative representations. Instead of matching to a simple standard normal distribution adversarially, we use end-to-end learning to constrain the latent space to be cluster-friendly by applying the Gaussian mixture distribution as the prior. Extensive experiments on challenging data sets show that our model achieves higher performance over a wide range of state-of-the-art clustering approaches.
Keyword:
Clustering algorithms
Data models
Neural networks
Image reconstruction
Decoding
Training
Estimation
Augmented mutual information (MI)
deep clustering
self-supervised learning (SSL)
variational embedding
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

U
University of Sydney
学者数:
6.5W
论文数: 6.2W
被引数: 90
B
Beijing University of Technology
学者数:
2.8W
论文数: 2.1W
被引数: 2.7W
引用论文

引用论文

err分享
err收藏
err分享
err收藏
Deep Clustering With Sample-Assignment Invariance Prior
err2020-11-01
err125
PREAI
errPeng, Xi; Zhu, Hongyuan; Feng, Jiashi; Shen, Chunhua; Zhang, Haixian; Zhou, Joey Tianyi
err分享
err收藏
学者 查看更多内容