Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Revisiting Self-Supervised Heterogeneous Graph Learning from Spectral Clustering Perspective

About

Self-supervised heterogeneous graph learning (SHGL) has shown promising potential in diverse scenarios. However, while existing SHGL methods share a similar essential with clustering approaches, they encounter two significant limitations: (i) noise in graph structures is often introduced during the message-passing process to weaken node representations, and (ii) cluster-level information may be inadequately captured and leveraged, diminishing the performance in downstream tasks. In this paper, we address these limitations by theoretically revisiting SHGL from the spectral clustering perspective and introducing a novel framework enhanced by rank and dual consistency constraints. Specifically, our framework incorporates a rank-constrained spectral clustering method that refines the affinity matrix to exclude noise effectively. Additionally, we integrate node-level and cluster-level consistency constraints that concurrently capture invariant and clustering information to facilitate learning in downstream tasks. We theoretically demonstrate that the learned representations are divided into distinct partitions based on the number of classes and exhibit enhanced generalization ability across tasks. Experimental results affirm the superiority of our method, showcasing remarkable improvements in several downstream tasks compared to existing methods.

Yujie Mo, Zhihe Lu, Runpeng Yu, Xiaofeng Zhu, Xinchao Wang• 2024

Related benchmarks

TaskDatasetResultRank
Node ClassificationACM
Macro F192.7
104
Node ClassificationDBLP
Micro-F194.7
94
Node ClusteringACM
ARI72.7
57
Node ClassificationAminer
Micro F186.8
46
Node ClusteringDBLP
NMI77.1
39
Node ClassificationYelp
Macro F1 Score93
15
Node ClassificationPhoto homogeneous (test)
Macro F1 Score91.9
12
Node ClassificationComputers homogeneous (test)
Macro F185.9
12
Node ClusteringYelp
NMI41.2
11
Node ClusteringAminer
NMI42.4
11
Showing 10 of 10 rows

Other info

Code

Follow for update