arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.03389cs.LGcs.AI

对比层次聚类

Contrastive Hierarchical Clustering

Advanced Micro Devices, Inc.(超威半导体公司);Cracow University of Economics(克拉科夫经济大学);Jagiellonian University(雅盖隆大学);Adam Mickiewicz University(亚当·密茨凯维奇大学)

Michał Znaleźniak, Przemysław Rola, Patryk Kaszuba, Jacek Tabor, Marek Śmieja

AI总结:

本文提出基于深度神经网络的对比层次聚类模型CoHiClust,通过自监督学习将基础网络蒸馏为二叉树,无需标签数据,生成合理聚类结构并提升图像聚类准确性。

摘要:

深度聚类一直以扁平模型为主导,这些模型将数据集划分为预定数量的组。尽管近期方法在流行基准上实现了与真实标签极高的相似度,但扁平划分所包含的信息是有限的。在本文中,我们引入了CoHiClust,一种基于深度神经网络的对比层次聚类模型,可应用于典型的图像数据。通过采用自监督学习方法,CoHiClust在不访问任何标记数据的情况下,将基础网络蒸馏为一棵二叉树。层次聚类结构可用于分析聚类之间的关系,以及衡量数据点之间的相似性。实验表明,CoHiClust生成了合理的聚类结构,这与我们的直觉和图像语义一致。此外,与最先进的扁平聚类模型相比,它在大多数图像数据集上获得了优越的聚类准确性。

英文摘要:

Deep clustering has been dominated by flat models, which split a dataset into a predefined number of groups. Although recent methods achieve an extremely high similarity with the ground truth on popular benchmarks, the information contained in the flat partition is limited. In this paper, we introduce CoHiClust, a Contrastive Hierarchical Clustering model based on deep neural networks, which can be applied to typical image data. By employing a self-supervised learning approach, CoHiClust distills the base network into a binary tree without access to any labeled data. The hierarchical clustering structure can be used to analyze the relationship between clusters, as well as to measure the similarity between data points. Experiments demonstrate that CoHiClust generates a reasonable structure of clusters, which is consistent with our intuition and image semantics. Moreover, it obtains superior clustering accuracy on most of the image datasets compared to the state-of-the-art flat clustering models.

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

↑