Share your thoughts, 1 month free Claude Pro on usSee more
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--
192
Node ClassificationReddit (test)--
137
Link PredictionReddit (inductive)
AP98.84
81
Node ClassificationDBLP (test)--
70
Link PredictionEnron (inductive)
AP89.76
66
Inductive dynamic link predictionReddit (inductive)
AUC-ROC (%)98.71
65
Link PredictionEnron (transductive)
AP92.47
49
Dynamic Link PredictionCan. Parl. Inductive
AP87.74
48
Dynamic Link PredictionWikipedia (inductive)
AP98.59
44
Inductive dynamic link predictionWikipedia (inductive)
AUC-ROC0.9848
44
Showing 10 of 188 rows
...

Other info

Code

Follow for update