提出端到端对比学习框架,直接优化节点嵌入与聚类结果。
From Representation to Clusters: A Contrastive Learning Approach for Attributed Hypergraph Clustering
- 联合学习嵌入与聚类,避免无关信息干扰。
- 在8个数据集上优于现有方法,提升聚类精度。
- 适合需要高精度图聚类的科研与工业场景。
对比学习在属性超图聚类中表现优异。传统方法先学习节点嵌入,再用k-means等算法聚类,但缺乏直接聚类监督,可能导致学习到无关信息。为此,我们提出一种端到端的属性超图聚类对比学习方法CAHC,包含表示学习与聚类分配学习两步。前者采用新颖的对比学习策略,融合节点级与超边级目标生成嵌入;后者通过联合优化嵌入与聚类,利用聚类导向的指导精炼嵌入并同步获得聚类结果。大量实验表明,CAHC在8个数据集上均优于基线方法。
原文摘要 · Abstract (English)
Contrastive learning has demonstrated strong performance in attributed hypergraph clustering. Typically, existing methods based on contrastive learning first learn node embeddings and then apply clustering algorithms, such as k-means, to these embeddings to obtain the clustering results.However, these methods lack direct clustering supervision, risking the inclusion of clustering-irrelevant information in the learned graph.To this end, we propose a Contrastive learning approach for Attributed Hypergraph Clustering (CAHC), an end-to-end method that simultaneously learns node embeddings and obtains clustering results. CAHC consists of two main steps: representation learning and cluster assignment learning. The former employs a novel contrastive learning approach that incorporates both node-level and hyperedge-level objectives to generate node embeddings.The latter joint embedding and clustering optimization to refine these embeddings by clustering-oriented guidance and obtains clustering results simultaneously.Extensive experimental results demonstrate that CAHC outperforms baselines on eight datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。