HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Variational Graph Recurrent Neural Networks

Ehsan Hajiramezanali; Arman Hasanzadeh; Nick Duffield; Krishna R Narayanan; Mingyuan Zhou; Xiaoning Qian

Variational Graph Recurrent Neural Networks

Abstract

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.

Code Repositories

marlin-codes/HTGN
pytorch
Mentioned in GitHub
VGraphRNN/VGRNN
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
dynamic-link-prediction-on-dblp-temporalVGRNN
AP: 87.77
AUC: 85.95
dynamic-link-prediction-on-dblp-temporalSI-VGRNN
AP: 88.36
AUC: 85.45
dynamic-link-prediction-on-enron-emailSI-VGRNN
AP: 93.93
AUC: 94.44
dynamic-link-prediction-on-enron-emailVGRNN
AP: 93.10
AUC: 93.29

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp