arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1806.01547cs.LGcs.CVstat.ML

Semi-Supervised Clustering with Neural Networks

IIIT-Delhi(德里英德拉普拉斯塔信息技术学院)

Ankita Shukla, Gullal Singh Cheema, Saket Anand

英文摘要:

Clustering using neural networks has recently demonstrated promising performance in machine learning and computer vision applications. However, the performance of current approaches is limited either by unsupervised learning or their dependence on large set of labeled data samples. In this paper, we propose ClusterNet that uses pairwise semantic constraints from very few labeled data samples (<5% of total data) and exploits the abundant unlabeled data to drive the clustering approach. We define a new loss function that uses pairwise semantic similarity between objects combined with constrained k-means clustering to efficiently utilize both labeled and unlabeled data in the same framework. The proposed network uses convolution autoencoder to learn a latent representation that groups data into k specified clusters, while also learning the cluster centers simultaneously. We evaluate and compare the performance of ClusterNet on several datasets and state of the art deep clustering approaches.

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

↑