arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.13489cs.LG

Learning Robust Representation for Clustering through Locality Preserving Variational Discriminative Network

Center for Data Science, Peking University(北京大学数据科学中心);MOE Key Laboratory of Computational Linguistic, School of EECS, Peking University(北京大学信息科学技术学院计算语言学教育部重点实验室);Mizuho Securities Co.,Ltd(瑞穗证券株式会社)

Ruixuan Luo, Wei Li, Zhiyuan Zhang, Ruihan Bao, Keiko Harimoto, Xu Sun

英文摘要:

Clustering is one of the fundamental problems in unsupervised learning. Recent deep learning based methods focus on learning clustering oriented representations. Among those methods, Variational Deep Embedding achieves great success in various clustering tasks by specifying a Gaussian Mixture prior to the latent space. However, VaDE suffers from two problems: 1) it is fragile to the input noise; 2) it ignores the locality information between the neighboring data points. In this paper, we propose a joint learning framework that improves VaDE with a robust embedding discriminator and a local structure constraint, which are both helpful to improve the robustness of our model. Experiment results on various vision and textual datasets demonstrate that our method outperforms the state-of-the-art baseline models in all metrics. Further detailed analysis shows that our proposed model is very robust to the adversarial inputs, which is a desirable property for practical applications.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑