Strongly augmented contrastive clustering

被引:36
|
作者
Deng, Xiaozhi [1 ]
Huang, Dong [1 ,2 ]
Chen, Ding-Hua [1 ]
Wang, Chang-Dong [3 ,4 ]
Lai, Jian-Huang [3 ,4 ]
机构
[1] South China Agr Univ, Coll Math & Informat, Guangzhou, Peoples R China
[2] Minist Agr & Rural Affairs, Key Lab Smart Agr Technol Trop South China, Beijing, Peoples R China
[3] Sun Yat sen Univ, Sch Comp Sci & Engn, Guangzhou, Peoples R China
[4] Guangdong Key Lab Informat Secur Technol, Guangdong, Peoples R China
关键词
Data clustering; Deep clustering; Image clustering; Contrastive learning; Deep neural network;
D O I
10.1016/j.patcog.2023.109470
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep clustering has attracted increasing attention in recent years due to its capability of joint representation learning and clustering via deep neural networks. In its latest developments, the contrastive learning has emerged as an effective technique to substantially enhance the deep clustering performance. However, the existing contrastive learning based deep clustering algorithms mostly focus on some carefullydesigned augmentations (often with limited transformations to preserve the structure), referred to as weak augmentations, but cannot go beyond the weak augmentations to explore the more opportunities in stronger augmentations (with more aggressive transformations or even severe distortions). In this paper, we present an end-to-end deep clustering approach termed S trongly A ugmented C ontrastive C lustering (SACC), which extends the conventional two-augmentation-view paradigm to multiple views and jointly leverages strong and weak augmentations for strengthened deep clustering. Particularly, we utilize a backbone network with triply-shared weights, where a strongly augmented view and two weakly augmented views are incorporated. Based on the representations produced by the backbone, the weak-weak view pair and the strong-weak view pairs are simultaneously exploited for the instance-level contrastive learning (via an instance projector) and the cluster-level contrastive learning (via a cluster projector), which, together with the backbone, can be jointly optimized in a purely unsupervised manner. Experimental results on five challenging image datasets have shown the superiority of our SACC approach over the state-of-the-art. The code is available at https://github.com/dengxiaozhi/SACC .(c) 2023 Elsevier Ltd. All rights reserved.
引用
收藏
页数:11
相关论文
共 50 条
  • [1] Contrastive Clustering
    Li, Yunfan
    Hu, Peng
    Liu, Zitao
    Peng, Dezhong
    Zhou, Joey Tianyi
    Peng, Xi
    THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 8547 - 8555
  • [2] Contrastive Hierarchical Clustering
    Znalezniak, Michal
    Rola, Przemyslaw
    Kaszuba, Patryk
    Tabor, Jacek
    Smieja, Marek
    MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES: RESEARCH TRACK, ECML PKDD 2023, PT I, 2023, 14169 : 627 - 643
  • [3] Contrastive Ensemble Clustering
    Chen, Man-Sheng
    Lin, Jia-Qi
    Wang, Chang-Dong
    Huang, Dong
    Lai, Jian-Huang
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2025,
  • [4] Graph Contrastive Clustering
    Zhong, Huasong
    Wu, Jianlong
    Chen, Chong
    Huang, Jianqiang
    Deng, Minghua
    Nie, Liqiang
    Lin, Zhouchen
    Hua, Xian-Sheng
    2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 9204 - 9213
  • [5] Safe Contrastive Clustering
    Tang, Pengwei
    Tang, Huayi
    Wang, Wei
    Liu, Yong
    MULTIMEDIA MODELING, MMM 2023, PT I, 2023, 13833 : 294 - 305
  • [6] Simple Contrastive Graph Clustering
    Liu, Yue
    Yang, Xihong
    Zhou, Sihang
    Liu, Xinwang
    Wang, Siwei
    Liang, Ke
    Tu, Wenxuan
    Li, Liang
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (10) : 13789 - 13800
  • [7] Pyramid contrastive learning for clustering
    Zhou, Zi-Feng
    Huang, Dong
    Wang, Chang-Dong
    NEURAL NETWORKS, 2025, 185
  • [8] Deep Temporal Contrastive Clustering
    Zhong, Ying
    Huang, Dong
    Wang, Chang-Dong
    NEURAL PROCESSING LETTERS, 2023, 55 (06) : 7869 - 7885
  • [9] Supporting Clustering with Contrastive Learning
    Zhang, Dejiao
    Nan, Feng
    Wei, Xiaokai
    Li, Shang-Wen
    Zhu, Henghui
    McKeown, Kathleen
    Nallapati, Ramesh
    Arnold, Andrew O.
    Xiang, Bing
    2021 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL-HLT 2021), 2021, : 5419 - 5430
  • [10] Federated Momentum Contrastive Clustering
    Miao, Runxuan
    Koyuncu, Erdem
    ACM TRANSACTIONS ON INTELLIGENT SYSTEMS AND TECHNOLOGY, 2024, 15 (04)