Source-linked AI summary

ROLAND: Graph Learning Framework for Dynamic Graphs

Jiaxuan You, Tianyu Du, Jure Leskovec

arXiv:2208.07239v1cs.LGcs.AIcs.SI

TL;DR

Dynamic GNNs remain limited by gaps in model design, evaluation settings, and training strategies. ROLAND repurposes static GNNs for dynamic graphs and achieves a 62.7% average performance gain over the best baseline under the existing evaluation setting across three datasets.

  • Problem

    Dynamic graph learning remains limited by gaps in dynamic GNNs’ model design, evaluation settings, and training strategies, despite the success of GNNs on static graphs.

    Source: §1 · p. 1
  • Method

    ROLAND repurposes static GNNs for dynamic graphs and pairs this framework with live-update evaluation and meta-learning for sequential prediction tasks.

    Source: §1 · p. 2, §3.3 · p. 5
  • Results

    62.7% performance gain over the best baseline on average under the existing evaluation setting across three datasets.

    Source: §1 · p. 2
  • Takeaways & Limitations

    ROLAND provides a framework for building, training, and evaluating dynamic GNNs that scales to graphs with 56 million edges.

    Source: §1 · p. 2, §6 · p. 8

    For BSI-ZK, the experiments sampled only 100 negative edges because of memory constraints.

    Source: §4.1 · p. 5

Abstract

from arXiv · show

Graph Neural Networks (GNNs) have been successfully applied to many real-world static graphs. However, the success of static graphs has not fully translated to dynamic graphs due to the limitations in model design, evaluation settings, and training strategies. Concretely, existing dynamic GNNs do not incorporate state-of-the-art designs from static GNNs, which limits their performance. Current evaluation settings for dynamic GNNs do not fully reflect the evolving nature of dynamic graphs. Finally, commonly used training methods for dynamic GNNs are not scalable. Here we propose ROLAND, an effective graph representation learning framework for real-world dynamic graphs. At its core, the ROLAND framework can help researchers easily repurpose any static GNN to dynamic graphs. Our insight is to view the node embeddings at different GNN layers as hierarchical node states and then recurrently update them over time. We then introduce a live-update evaluation setting for dynamic graphs that mimics real-world use cases, where GNNs are making predictions and being updated on a rolling basis. Finally, we propose a scalable and efficient training approach for dynamic GNNs via incremental training and meta-learning. We conduct experiments over eight different dynamic graph datasets on future link prediction tasks. Models built using the ROLAND framework achieve on average 62.7% relative mean reciprocal rank (MRR) improvement over state-of-the-art baselines under the standard evaluation settings on three datasets. We find state-of-the-art baselines experience out-of-memory errors for larger datasets, while ROLAND can easily scale to dynamic graphs with 56 million edges. After re-implementing these baselines using the ROLAND training strategy, ROLAND models still achieve on average 15.5% relative MRR improvement over the baselines.

1 INTRODUCTION AND RELATED WORK

ROLAND adapts static GNNs to dynamic snapshot graphs by updating hierarchical node states, evaluating predictions in rolling live-update settings, and training incrementally with meta-learning. It targets limitations in dynamic GNN design, evaluation, and scalability.

ROLAND model

  • ROLAND views node embeddings across static GNN layers as hierarchical states, updating them from new nodes and edges with moving average, MLP, or GRU modules.This preserves the static GNN design while adapting it to dynamic graphs, unlike approaches that do not incorporate state-of-the-art static GNN designs.Source: §1 · pp. 1–2

ROLAND evaluation

  • The live-update protocol makes rolling predictions using only historical training data and permits fine-tuning on each new snapshot.Existing deterministic time splits evaluate only the final period and can overestimate performance under long-term pattern changes; a non-updated model can also become stale.Source: §1 · pp. 1–2

ROLAND training

  • Incremental truncated BPTT keeps only the incoming snapshot and historical node states in GPU memory, enabling training on a 56-million-edge transaction network.The network is about 13 times larger than existing benchmarks in edges per snapshot.Source: §1 · p. 2

Key results

  • ROLAND gained 62.7% over the best baseline on average under existing evaluation on three datasets, and 15.5% over baselines re-implemented with ROLAND training.The experiments covered eight dynamic graph datasets; existing BPTT-based methods failed to scale to the larger datasets.Source: §1 · p. 2

ROLAND training

  • ROLAND formulates predictions across periods as sequential tasks and uses a meta-model initialization to adapt quickly to unseen future tasks.This addresses the limited research on generalizing and quickly adapting dynamic GNNs to new data.Source: §1 · p. 2

2 PRELIMINARIES

The section defines dynamic graphs as timestamped sequences of static snapshots and describes GNNs as iterative neighborhood-message aggregators. It also outlines how ROLAND extends static GNNs to dynamic graphs by updating hierarchical node states over time.

Graphs and dynamic graphs

  • A dynamic graph is represented as timestamped static graph snapshots, whose node sets may differ to accommodate node addition and deletion.Each snapshot contains nodes and edges timestamped at time t.Source: §2 · pp. 2–3

ROLAND model design principle

  • ROLAND extends static GNNs by inserting embedding-update modules that recurrently update hierarchical node states over time.The illustrated static architecture includes intra-layer designs such as BatchNorm and attention, and inter-layer skip connections.Source: Figure 2 · p. 3

Graph Neural Networks

  • A GNN iteratively aggregates messages from local neighborhoods into node embeddings, while static GNNs do not capture graph dynamics.GNNs may use different message-passing and aggregation functions.Source: §2 · p. 3

3 PROPOSED ROLAND FRAMEWORK

ROLAND generalizes static GNNs to dynamic graphs by treating embeddings at every layer as hierarchical node states that are updated over time. It pairs this framework with live-update evaluation and incremental, meta-learning-based training strategies.

3.1 From Static GNNs to Dynamic GNNs

  • Unlike sequence-on-GNN approaches that update only H(L) [47], ROLAND keeps and hierarchically updates all layer embeddings through a module insertable into any static GNN.The layer states encode information from neighbors at different hop distances.Source: §3.1 · p. 3
  • ROLAND’s GNN design incorporates edge features, bidirectional message information, sum, max, or mean aggregation, and skip connections, then uses an MLP to predict future edges.The paper notes these design choices boost performance in Section 4.4.Source: §3.1 · p. 4

3.2 Live-update Evaluation

  • Live-update evaluation fine-tunes on newly observed labels, evaluates predictions on new data using historical node state, and averages MRR across prediction steps without future information.The procedure is motivated by evolving data distributions that can make evaluation on only the final 10% of snapshots misleading.Source: §3.2 · p. 4, Figure 3 · p. 4

3.3 Training Strategies

  • Incremental training retains the model, current graph snapshot, and historical node states in GPU memory, making memory complexity independent of snapshot count and enabling training on 56 million edges and 733 snapshots.Historical states encode information through the previous time step, so training uses them with only the newly observed snapshot.Source: §3.3 · pp. 4–5
  • For fast adaptation, ROLAND treats prediction periods as separate tasks, initializes each model from a Reptile-trained meta-model, and updates that meta-model with a moving average of trained models.The meta-model provides an initialization for quickly deriving a specialized model for an unseen future task.Source: §3.3 · p. 5

4 EXPERIMENTS

ROLAND improves future-link prediction across fixed-split and live-update evaluations, while its incremental training scales beyond the default BPTT approach. Ablations support architectural choices, meta-learning, and retraining across snapshots.

4.1 Experimental Setup

  • Experiments cover eight dynamic graph datasets and future link prediction, comparing fixed-split evaluation on the last 10% of snapshots with live-update evaluation across available snapshots.MRR ranks each positive edge against sampled negatives; BSI-ZK uses 100 negatives because of memory constraints, while other datasets use 10001.Source: §4.1 · p. 5

4.2 Results in the Standard Evaluation Settings

  • 62.69%: ROLAND with GRU achieves an average performance gain over state-of-the-art baselines under the standard fixed-split setting.The comparison follows the EvolveGCN paper’s snapshot frequency, test snapshots, and MRR computation.Source: §4.2 · p. 6, Table 2 · p. 6

4.3 Results in the Live-update Settings

  • Default BPTT fails to scale to large datasets and models, while even its best models on smaller datasets underperform ROLAND-trained baselines by almost 10%.BPTT requires storing historical node embeddings in GPU memory; it fails on BSI-ZK, AS-733, and large models on BSI-SVT.Source: §4.3 · p. 6

4.4 Ablation Study

  • Batch normalization, skip-connections, and max aggregation improve the ROLAND architecture on BSI-SVT in live-update evaluation.Adding skip-connections yields more than 20% performance gain.Source: §4.4 · p. 6, Figure 4 · p. 7

4.3 Results in the Live-update Settings

  • 15.48%: ROLAND with GRU outperforms the best baseline on average after baselines are reimplemented with ROLAND training, except on AS-733.All baselines were successfully trained with the incremental pipeline; gains range from 2.40% on BSI-ZK to 44.22% on Reddit-Body.Source: §4.3 · p. 6, Table 3 · p. 7

4.4 Ablation Study

  • Meta-learning improves MRR by 2.84% to 13.19% on average across embedding update designs and generally stabilizes performance.The study varies α from 0.1 to 1.0; α = 1 uses the previous model directly to initialize training.Source: §4.4 · p. 7
  • Meta-training enables ROLAND to retrain periodically for a few epochs, and retraining across snapshots performs significantly better than stopping after the first 25%.The experiments show transaction counts varying over time as ROLAND retrains to fit the data.Source: §4.4 · p. 7

5 ADDITIONAL RELATED WORK

ROLAND differs from recurrent-backbone dynamic GNNs by updating hierarchical node embeddings and builds on established static GNN designs. Its related-work discussion also contrasts its scalable training and snapshot-based framework with alternative approaches to dynamic graphs.

Dynamic GNNs

  • ROLAND recurrently updates hierarchical node embeddings, whereas EvolveGCN recurrently updates GNN weights and loses explicit historical information.Other recurrent-backbone approaches either replace a feature encoder with a GNN, ignoring lower-level embedding evolution, or replace RNN-cell linear layers with GNN layers rather than building on an established static GNN architecture. The text also reports that EvolveGCN performs undesirably when the number of snapshots is large.Source: §5 · pp. 7–8

Scalable training

  • Incremental training retains historical information in node hidden states while backpropagating only within the current snapshot, unlike snapshot-limiting heuristics.Some Transformer-based sequence models require the entire historical graph in GPU memory to compute attention over time.Source: §5 · p. 8

Non-snapshot dynamic graph representation

  • Timestamped-node-and-edge dynamic graphs can be converted to snapshots by grouping nodes and edges within a period, then applying ROLAND.Source: §5 · p. 8

Dynamic GNNs

  • ROLAND incorporates mature static-GNN designs, including edge features, batch normalization, and skip-connections, and the paper reports that it can significantly outperform existing dynamic GNNs.Source: §5 · p. 8

Other learning methods for dynamic graphs

  • Dynamic-graph methods also include matrix factorization, random walks, and point processes; this paper focuses on GNNs for their performance and inductive learning capabilities.Source: §5 · p. 8

6 CONCLUSION

ROLAND enables researchers to adapt static GNNs for dynamic graphs and evaluate them with live updates. The framework scales to large datasets and its dynamic GNNs outperform existing state-of-the-art models.

  • ROLAND helps researchers repurpose static GNNs for dynamic graphs and includes a live-update evaluation pipeline that mimics real-world use.Source: §6 · p. 8
  • ROLAND-built dynamic GNNs scale to large datasets and outperform existing state-of-the-art models.Source: §6 · pp. 8–9
Loading 2208.07239v1…