arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.05484cs.CV

Doubly Contrastive Deep Clustering

School of Electronic Engineering, Xidian University(西安电子科技大学电子工程学院);University of Pittsburgh(匹兹堡大学);JD Finance America Corporation(京东金融美国公司)

Zhiyuan Dang, Cheng Deng, Xu Yang, Heng Huang

英文摘要:

Deep clustering successfully provides more effective features than conventional ones and thus becomes an important technique in current unsupervised learning. However, most deep clustering methods ignore the vital positive and negative pairs introduced by data augmentation and further the significance of contrastive learning, which leads to suboptimal performance. In this paper, we present a novel Doubly Contrastive Deep Clustering (DCDC) framework, which constructs contrastive loss over both sample and class views to obtain more discriminative features and competitive results. Specifically, for the sample view, we set the class distribution of the original sample and its augmented version as positive sample pairs and set one of the other augmented samples as negative sample pairs. After that, we can adopt the sample-wise contrastive loss to pull positive sample pairs together and push negative sample pairs apart. Similarly, for the class view, we build the positive and negative pairs from the sample distribution of the class. In this way, two contrastive losses successfully constrain the clustering results of mini-batch samples in both sample and class level. Extensive experimental results on six benchmark datasets demonstrate the superiority of our proposed model against state-of-the-art methods. Particularly in the challenging dataset Tiny-ImageNet, our method leads 5.6\% against the latest comparison method. Our code will be available at \url{https://github.com/ZhiyuanDang/DCDC}.

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

↑