arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1910.08031cs.LGstat.ML
收藏

Deep clustering with concrete k-means

University of Edinburgh(爱丁堡大学);University of Surrey(萨里大学);Samsung AI Centre(三星人工智能中心)

Boyan Gao, Yongxin Yang, Henry Gouk, Timothy M. Hospedales

英文摘要:

We address the problem of simultaneously learning a k-means clustering and deep feature representation from unlabelled data, which is of interest due to the potential of deep k-means to outperform traditional two-step feature extraction and shallow-clustering strategies. We achieve this by developing a gradient-estimator for the non-differentiable k-means objective via the Gumbel-Softmax reparameterisation trick. In contrast to previous attempts at deep clustering, our concrete k-means model can be optimised with respect to the canonical k-means objective and is easily trained end-to-end without resorting to alternating optimisation. We demonstrate the efficacy of our method on standard clustering benchmarks.

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

↑