Daily Arxiv

This page organizes papers related to artificial intelligence published around the world.
This page is summarized using Google Gemini and is operated on a non-profit basis.
The copyright of the paper belongs to the author and the relevant institution. When sharing, simply cite the source.

CueGCL: Cluster-aware Personalized Self-Training for Unsupervised Graph Contrastive Learning

Created by
  • Haebom

Author

Yuecheng Li, Lele Fu, Sheng Huang, Chuan Chen, Lei Yang, Zibin Zheng

Outline

This paper proposes a Cluster-aware Graph Contrastive Learning Framework (CueGCL) that simultaneously performs cluster information learning and node representation learning to address the performance degradation of existing graph contrastive learning (GCL) algorithms in structure-related unsupervised learning tasks such as graph clustering. Specifically, we design a personalized self-learning (PeST) strategy for unsupervised learning scenarios to capture accurate cluster-level personalized information and mitigate class conflicts and unfairness. Furthermore, we demonstrate that aligned graph clustering (AGC) yields consistent node embeddings and, theoretically demonstrating its effectiveness, generates an embedding space with clearly distinct cluster structures. Experimental results on five benchmark datasets demonstrate that CueGCL achieves state-of-the-art performance.

Takeaways, Limitations

Takeaways:
We present a novel framework that improves the performance of GCL in structure-related unsupervised learning tasks such as graph clustering.
Mitigating class conflict and unfairness issues through personalized self-study (PeST) strategies.
Generating consistent node embeddings via aligned graph clustering (AGC).
Achieving state-of-the-art performance on datasets of varying sizes
Limitations:
Further analysis is needed on the computational complexity and scalability of the proposed method.
Need to evaluate generalization performance for various graph structures and data characteristics
In-depth research is needed on hyperparameter optimization of PeST and AGC strategies.
👍