@inproceedings{
yang2024graphlearner,
title={GraphLearner: Graph Node Clustering with Fully Learnable Augmentation},
author={Xihong Yang and Erxue Min and KE LIANG and Yue Liu and Siwei Wang and sihang zhou and Huijun Wu and Xinwang Liu and En Zhu},
booktitle={ACM Multimedia 2024},
year={2024},
url={https://openreview.net/forum?id=MvkLwdeW5q}
}
Contrastive deep graph clustering (CDGC) leverages the power of contrastive learning to group nodes into different clusters.  The quality of contrastive samples is crucial for achieving better performance, making augmentation techniques a key factor in the process. However, the augmentation samples in existing methods are always predefined by human experiences, and agnostic from the downstream task clustering, thus leading to high human resource costs and poor performance. To overcome these limitations, we propose a Graph Node Clustering with Fully Learnable Augmentation, termed GraphLearner. It introduces learnable augmentors to generate high-quality and task-specific augmented samples for CDGC. GraphLearner incorporates two learnable augmentors specifically designed for capturing attribute and structural information. Moreover, we introduce two refinement matrices, including the high-confidence pseudo-label matrix and the cross-view sample similarity matrix, to enhance the reliability of the learned affinity matrix. During the training procedure, we notice the distinct optimization goals for training learnable augmentors and contrastive learning networks. In other words, we should both guarantee the consistency of the embeddings as well as the diversity of the augmented samples. To address this challenge, we propose an adversarial learning mechanism within our method. Besides, we leverage a two-stage training strategy to refine the high-confidence matrices. Extensive experimental results on six benchmark datasets validate the effectiveness of GraphLearner.
GraphLearner: Graph Node Clustering with Fully Learnable Augmentation
Xihong Yang Erxue Min KE LIANG Yue Liu Siwei Wang sihang zhou Huijun Wu Xinwang Liu En Zhu