From Representation to Clusters: A Contrastive Learning Approach for Attributed Hypergraph Clustering
Organizations: School of Computer Science and Technology, Anhui University, Hefei, China · School of Computer Science and Technology, Southwest University, Chongqing, China
Abstract
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.
Figures & tables
| Dataset | # Nodes | # Hyperedges | # Features | # Classes |
| Cora-C | 1,434 | 1,579 | 1,433 | 7 |
| Citeseer | 1,458 | 1,079 | 3,703 | 6 |
| Pubmed | 3,840 | 7,963 | 500 | 3 |
| Cora-A | 2,388 | 1,072 | 1,433 | 7 |
| DBLP | 41,302 | 22,363 | 1425 | 6 |
| 20NewsW100 | 16,242 | 100 | 100 | 4 |
| Dataset | Metric | Node2vec | DGI | RAGC | Hyper2vec | TriCL | SE-HSSL | CAHC |
| Cora-C | ACC | 47.4 5.4 | 68.4 3.4 | 68.6 1.6 | 59.7 5.1 | 70.0 1.1 | 72.0 3.5 | 71.4 1.7 |
| F1 | 44.7 7.1 | 62.0 3.0 | 64.8 1.6 | 58.5 0.4 | 65.1 1.8 | 67.0 4.3 | 69.1 2.8 | |
| NMI | 32.5 3.9 | 54.0 1.0 | 50.9 1.7 | 40.6 1.1 | 55.2 0.8 | 56.5 2.4 | 56.4 0.9 | |
| ARI | 20.2 4.2 | 47.1 2.0 | 43.5 2.2 | 33.6 0.9 | 49.2 0.2 | 49.3 3.5 | 50.8 1.6 | |
| Citeseer | ACC | 36.2 1.6 | 63.8 4.3 | 69.4 1.4 | 42.7 1.3 | 67.6 1.5 | 68.2 1.7 | 69.8 0.8 |
| F1 | 29.2 3.1 | 55.7 3.4 | 61.9 1.9 | 41.1 1.2 | 62.1 2.0 | 62.5 1.5 | 65.1 0.6 |
| Dataset | Metric | w/o re | w/o hy | w/o no | w/o cl | w/ mu | CAHC |
| Cora-C | ACC | 28.0 | 68.8 | 72.9 | 70.4 | 71.6 | 71.4 |
| F1 | 6.2 | 63.4 | 70.5 | 68.4 | 67.1 | 69.1 | |
| NMI | 0.0 | 51.4 | 54.4 | 55.2 | 54.5 | 56.4 | |
| ARI | 0.0 | 46.5 | 51.9 | 49.2 | 49.9 | 50.8 | |
| Citeseer | ACC | 21.7 | 69.1 | 65.5 | 68.8 | 67.0 | 69.8 |
| F1 | 5.9 | 64.9 | 60.6 | 64.3 | 62.4 | 65.1 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | D | LR | Ass. LR | WD | Pre. E | Ass. E | ||
| Cora-C | 0.4 | 0.4 | 512 | 5e-4 | 5e-4 | 5e-4 | 180 | 120 |
| Citeseer | 0.4 | 0.4 | 2048 | 1e-4 | 1e-5 | 5e-5 | 200 | 80 |
| Pubmed | 0.1 | 0.4 | 1024 | 5e-5 | 5e-5 | 5e-5 | 150 | 60 |
| Cora-A | 0.3 | 0.2 | 768 | 1e-4 | 1e-5 | 1e-5 | 100 | 140 |
| DBLP | 0.4 | 0.2 | 256 | 8e-4 | 8e-4 | 5e-5 | 260 | 100 |
| 20NewsW100 | 0.1 | 0.2 | 512 | 1e-3 | 1e-4 | 5e-4 | 80 | 30 |
Hypergraph Learning
Hypergraph learning models relationships among groups of entities to capture interactions that pairwise graphs cannot represent.
See all 54 papers