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

Towards Better Dynamic Graph Learning: New Architecture and Unified Library

About

We propose DyGFormer, a new Transformer-based architecture for dynamic graph learning. DyGFormer is conceptually simple and only needs to learn from nodes' historical first-hop interactions by: (1) a neighbor co-occurrence encoding scheme that explores the correlations of the source node and destination node based on their historical sequences; (2) a patching technique that divides each sequence into multiple patches and feeds them to Transformer, allowing the model to effectively and efficiently benefit from longer histories. We also introduce DyGLib, a unified library with standard training pipelines, extensible coding interfaces, and comprehensive evaluating protocols to promote reproducible, scalable, and credible dynamic graph learning research. By performing exhaustive experiments on thirteen datasets for dynamic link prediction and dynamic node classification tasks, we find that DyGFormer achieves state-of-the-art performance on most of the datasets, demonstrating its effectiveness in capturing nodes' correlations and long-term temporal dependencies. Moreover, some results of baselines are inconsistent with previous reports, which may be caused by their diverse but less rigorous implementations, showing the importance of DyGLib. All the used resources are publicly available at https://github.com/yule-BUAA/DyGLib.

Le Yu, Leilei Sun, Bowen Du, Weifeng Lv• 2023

Related benchmarks

TaskDatasetResultRank
Node ClassificationReddit (test)--
134
Node ClassificationDBLP (test)--
70
Link PredictionReddit (inductive)
AP98.84
52
Dynamic Link DetectionENRON
AP92.47
44
Dynamic Graph Anomaly DetectionWikipedia S2
AUROC79.2
42
Dynamic Graph Anomaly DetectionMOOC S2
AUROC65.92
42
Dynamic new link predictionSocial Evo.
AP0.9768
37
Link PredictionEnron (inductive)
AP89.76
37
Link PredictionReddit (transductive)
AP99.11
30
Link PredictionLastFM (transductive)
AP92.07
28
Showing 10 of 82 rows
...

Other info

Code

Follow for update