通过渐进式社区检测提升深度聚类性能
Deep Clustering via Gradual Community Detection
- 从大量伪社区开始,逐步合并形成聚类
- 在多个图像数据集上达到当前最佳表现
- 适合需要高质量自监督聚类的场景
深度聚类是现代人工智能中的关键任务,旨在将数据样本划分为给定数量的同质组(即聚类)。尽管近期研究提出了更先进的深度神经网络和训练策略,显著提升了性能,但因监督信号不足,深度聚类仍具挑战性。本文基于现有表征学习主干网络,提出一种新型渐进式社区检测聚类策略:先将样本划分为大量伪社区,再通过社区合并逐步扩展聚类。该方法引入图论中社区检测的新视角,利用全局结构特征提升聚类伪标签纯度,对自监督性能至关重要。我们在主流骨干网络上实现该方法,并在基准图像数据集上验证其有效性。大量实验表明,该策略显著优于现有方法。消融实验证明,新视角有效提升了伪标签纯度,从而改善自监督效果。
原文摘要 · Abstract (English)
Deep clustering is an essential task in modern artificial intelligence, aiming to partition a set of data samples into a given number of homogeneous groups (i.e., clusters). Recent studies have proposed increasingly advanced deep neural networks and training strategies for deep clustering, effectively improving performance. However, deep clustering generally remains challenging due to the inadequacy of supervision signals. Building upon the existing representation learning backbones, this paper proposes a novel clustering strategy of gradual community detection. It initializes clustering by partitioning samples into many pseudo-communities and then gradually expands clusters by community merging. Compared with the existing clustering strategies, community detection factors in the new perspective of cluster network analysis in the clustering process. The new perspective can effectively leverage global structural characteristics to enhance cluster pseudo-label purity, which is critical to the performance of self-supervision. We have implemented the proposed approach based on the popular backbones and evaluated its efficacy on benchmark image datasets. Our extensive experiments have shown that the proposed clustering strategy can effectively improve the SOTA performance. Our ablation study also demonstrates that the new network perspective can effectively improve community pseudo-label purity, resulting in improved self-supervision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。