Learning Long Range Spatio-Temporal Representations over Continuous Time Dynamic Graphs with State Space Models

· Source: cs.AI updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics · Depth: Expert, extended

Summary

The paper introduces CTDG-SSM, a parameter-efficient state-space modeling framework for continuous-time dynamic graphs (CTDGs). This model addresses the challenge of capturing both long-range temporal (LRT) and long-range spatial (LRS) dependencies in evolving relational data, a limitation of existing event-driven and sequence-based models. CTDG-SSM derives from a novel continuous-time Topology-Aware higher-order polynomial projection operator (CTT-HiPPO), which jointly encodes temporal dynamics and graph structure by projecting classical HiPPO solutions through a polynomial of the Laplacian matrix. A computationally efficient discrete formulation is achieved using the zero-order hold approach. Across benchmarks including dynamic link prediction, dynamic node classification, and sequence classification, CTDG-SSM achieves state-of-the-art performance, showing large gains on datasets requiring LRT and LRS reasoning (e.g., LastFM, MOOC, Enron). It also demonstrates high parameter efficiency, using about one-tenth fewer parameters than competing methods, and robustness to structural perturbations.

Key takeaway

For AI Architects designing systems for evolving relational data, CTDG-SSM offers a robust solution for simultaneously addressing long-range temporal and spatial dependencies. You should consider integrating this parameter-efficient state-space framework, especially for applications like financial fraud detection or social network analysis where fine-grained, multi-hop interactions are critical. Its proven state-of-the-art performance and computational efficiency on benchmarks like LastFM and Enron suggest it can improve accuracy while reducing resource overhead.

Key insights

CTDG-SSM unifies state-space modeling with graph topology to capture long-range spatio-temporal dependencies in continuous-time dynamic graphs.

Principles

Method

CTDG-SSM derives CTT-HiPPO by minimizing discrepancy between observed node features and graph-filtered polynomial approximations. This continuous formulation is then discretized using zero-order hold for efficient implementation.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Architect

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.