'Si'multaneous 'S'patial-'T'emporal Message Passing for Dynamic Graph Representation Learning

2026-05-25Machine Learning

Machine LearningArtificial Intelligence
AI summary

The authors point out that existing dynamic graph neural networks process time and space information in a strict sequence, which limits their ability to understand how a node's neighbors have evolved over time. To fix this, they created SiST-GNN, a model that combines spatial and temporal information simultaneously by treating a node's past state and current features as connected parts in a graph. This allows better communication of history during graph updates. Their experiments show that SiST-GNN outperforms previous methods by a large margin on several link prediction and node classification tasks across multiple datasets.

dynamic graph neural networkstemporal embeddingsspatial aggregationmessage passinggraph convolutionlink predictionnode classificationsnapshot sequencesrecurrent hidden statecontinuous-time dynamic graphs
Authors
Shubhajit Roy, Anirban Dasgupta
Abstract
Dynamic graph neural networks (DGNNs) that operate on snapshot sequences typically fall into one of two categories. \emph{Temporal-first} approaches build per-node temporal embeddings and only afterwards perform spatial aggregation, whereas \emph{Spatial-first} approaches invert this order, feeding the output of a graph convolution into a downstream temporal module. In either case, the rigid sequencing forces the second stage to consume an already-compressed summary produced by the first, ruling out joint reasoning over topology and evolution; concretely, the message-passing operator never gets to weight a neighbor's contribution by that neighbor's \emph{past} trajectory. This paper introduces \textbf{SiST-GNN} (\textbf{Si}multaneous \textbf{S}patial-\textbf{T}emporal \textbf{GNN}), which fuses the two signals inside a single message-passing operation rather than chaining them. Concretely, at each snapshot we maintain a recurrent hidden state per node that summarises its history, pair it with the node's current feature vector, and treat the pair as two nodes joined by a cross-time edge; running a standard graph convolution on this temporally augmented graph yields the updated representation. Our empirical study spans nine public baselines and fourteen model-dataset combinations, covering both fixed-split and live-update evaluation regimes. Across every public benchmark, SiST-GNN sets a new state of the art in link prediction task over the strongest prior method by $109$--$277\%$ in the fixed-split setting and by $68$--$194\%$ in the live-update setting. We additionally construct three dynamic node-classification tasks by discretising the underlying continuous-time event streams; here SiST-GNN beats the leading discrete-time (DTDG) baseline by $7$--$22\%$ and matches continuous-time (CTDG) methods that consume the raw events directly.