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

GraphCroc: Cross-Correlation Autoencoder for Graph Structural Reconstruction

About

Graph-structured data is integral to many applications, prompting the development of various graph representation methods. Graph autoencoders (GAEs), in particular, reconstruct graph structures from node embeddings. Current GAE models primarily utilize self-correlation to represent graph structures and focus on node-level tasks, often overlooking multi-graph scenarios. Our theoretical analysis indicates that self-correlation generally falls short in accurately representing specific graph features such as islands, symmetrical structures, and directional edges, particularly in smaller or multiple graph contexts. To address these limitations, we introduce a cross-correlation mechanism that significantly enhances the GAE representational capabilities. Additionally, we propose GraphCroc, a new GAE that supports flexible encoder architectures tailored for various downstream tasks and ensures robust structural reconstruction, through a mirrored encoding-decoding process. This model also tackles the challenge of representation bias during optimization by implementing a loss-balancing strategy. Both theoretical analysis and numerical evaluations demonstrate that our methodology significantly outperforms existing self-correlation-based GAEs in graph structure reconstruction.

Shijin Duan, Ruyi Ding, Jiaxing He, Aidong Adam Ding, Yunsi Fei, Xiaolin Xu• 2024

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy79.09
742
Graph ClassificationCOLLAB
Accuracy82.4
329
Graph ClassificationIMDB-B
Accuracy78.75
322
Adjacency Matrix ReconstructionPROTEINS (test)
ROC AUC99.58
6
Adjacency Matrix ReconstructionIMDB-B (test)
ROC AUC0.9992
6
Adjacency Matrix ReconstructionCOLLAB (test)
ROC AUC0.9989
6
Adjacency Matrix ReconstructionQM9 (test)
ROC-AUC0.9987
6
Adjacency Matrix ReconstructionPPI (test)
ROC AUC0.9831
5
Directed graph reconstructionCora_ML N=41 (test)
AUC0.9946
2
Directed graph reconstructionCora_ML N=77 (test)
AUC99.96
2
Showing 10 of 11 rows

Other info

Code

Follow for update