Unsupervised Contrastive Clustering Via Density Cluster Representative Combination

Jianyun Lu,Junming Shao
DOI: https://doi.org/10.1016/j.eswa.2024.125967
IF: 8.5
2024-01-01
Expert Systems with Applications
Abstract:The recent development of contrastive clustering for deep image clustering has shown promising results by combining representation learning and clustering prediction into a unified framework. In this paper, we introduce an unsupervised contrastive clustering method inspired by contrastive learning to address the issue of cluster center selection in density peaks clustering. We propose a new type of information entropy peaks based on dynamic reverse nearest neighbor sequence to measure local density for each object, and select a small number of top information entropy peaks as cluster representative candidates to cover the right cluster centers as comprehensively as possible. To mitigate the risk of wrongly determining cluster centers, we generate a cluster centers set by computing various cluster representative candidates combinations. In the contrastive clustering stage, we introduce transitive nearest neighbor distance to compute cluster-level loss for the core data set. The prototype clustering is obtained with the minimum cluster-level loss, and the non-core data set is grouped into prototype clustering by nearest neighbor distance. Our comprehensive experiments on both artificial and real-world data sets demonstrate that the proposed unsupervised contrastive clustering method is effective and outperforms the compared state-of-the-art clustering algorithms.
What problem does this paper attempt to address?