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

Variational Graph Recurrent Neural Networks

About

Representation learning over graph structured data has been mostly studied in static graph settings while efforts for modeling dynamic graphs are still scant. In this paper, we develop a novel hierarchical variational model that introduces additional latent random variables to jointly model the hidden states of a graph recurrent neural network (GRNN) to capture both topology and node attribute changes in dynamic graphs. We argue that the use of high-level latent random variables in this variational GRNN (VGRNN) can better capture potential variability observed in dynamic graphs as well as the uncertainty of node latent representation. With semi-implicit variational inference developed for this new VGRNN architecture (SI-VGRNN), we show that flexible non-Gaussian latent representations can further help dynamic graph analytic tasks. Our experiments with multiple real-world dynamic graph datasets demonstrate that SI-VGRNN and VGRNN consistently outperform the existing baseline and state-of-the-art methods by a significant margin in dynamic link prediction.

Ehsan Hajiramezanali, Arman Hasanzadeh, Nick Duffield, Krishna R Narayanan, Mingyuan Zhou, Xiaoning Qian• 2019

Related benchmarks

TaskDatasetResultRank
Dynamic Link DetectionENRON
AP96.31
44
Dynamic new link predictionSocial Evo.
AP0.774
37
Dynamic new link predictionCOLAB
AUC0.8595
10
Dynamic new link predictionFacebook
AUC90.94
10
Dynamic Link DetectionCOLAB (10 runs on random splits)
AUC89.15
7
Dynamic Link DetectionFacebook 10 runs on random splits
AUC0.8812
7
Dynamic Link DetectionSocial Evo. (10 runs on random splits)
AUC83.36
7
Dynamic Link DetectionHEP-TH last 10 snapshots (test)
AUC91.12
7
Dynamic Link DetectionCora Includes node attributes (10 runs on random splits)
AUC94.07
7
Link PredictionEnron (last three snapshots)
AUC92.68
6
Showing 10 of 12 rows

Other info

Code

Follow for update