Contrastive Hierarchical Clustering
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.
Code (1)
Tasks
ClusteringDeep ClusteringImage ClusteringSelf-Supervised LearningMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Contrastive Multi-view Hyperbolic Hierarchical Clustering
Hierarchical clustering recursively partitions data at an increasingly finer granularity. In real-world applications, multi-view data have become increasingly important. This raises a less investigated problem, i.e., mul…
ClusteringMHCCL: Masked Hierarchical Cluster-Wise Contrastive Learning for Multivariate Time Series
Learning semantic-rich representations from raw unlabeled time series data is critical for downstream tasks such as classification and forecasting. Contrastive learning has recently shown its promising representation lea…
ClusteringContrastive LearningRepresentation LearningTime Series+1HCL: Improving Graph Representation with Hierarchical Contrastive Learning
Contrastive learning has emerged as a powerful tool for graph representation learning. However, most contrastive learning methods learn features of graphs with fixed coarse-grained scale, which might underestimate either…
Contrastive LearningGraph ClassificationGraph Representation LearningNode Classification+2Improved Face Representation via Joint Label Classification and Supervised Contrastive Clustering
Face clustering tasks can learn hierarchical semantic information from large-scale data, which has the potential to help facilitate face recognition. However, there are few works on this problem. This paper explores it b…
ClusteringFace ClusteringFace RecognitionHGCL: Hierarchical Graph Contrastive Learning for User-Item Recommendation
Graph Contrastive Learning (GCL), which fuses graph neural networks with contrastive learning, has evolved as a pivotal tool in user-item recommendations. While promising, existing GCL methods often lack explicit modelin…
Contrastive Learning