State Space Models on Temporal Graphs: A First-Principles Study

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Li, Jintang, Wu, Ruofan, Jin, Xinzhou, Ma, Boqun, Chen, Liang, Zheng, Zibin
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916458764697600
author Li, Jintang
Wu, Ruofan
Jin, Xinzhou
Ma, Boqun
Chen, Liang
Zheng, Zibin
author_facet Li, Jintang
Wu, Ruofan
Jin, Xinzhou
Ma, Boqun
Chen, Liang
Zheng, Zibin
contents Over the past few years, research on deep graph learning has shifted from static graphs to temporal graphs in response to real-world complex systems that exhibit dynamic behaviors. In practice, temporal graphs are formalized as an ordered sequence of static graph snapshots observed at discrete time points. Sequence models such as RNNs or Transformers have long been the predominant backbone networks for modeling such temporal graphs. Yet, despite the promising results, RNNs struggle with long-range dependencies, while transformers are burdened by quadratic computational complexity. Recently, state space models (SSMs), which are framed as discretized representations of an underlying continuous-time linear dynamical system, have garnered substantial attention and achieved breakthrough advancements in independent sequence modeling. In this work, we undertake a principled investigation that extends SSM theory to temporal graphs by integrating structural information into the online approximation objective via the adoption of a Laplacian regularization term. The emergent continuous-time system introduces novel algorithmic challenges, thereby necessitating our development of GraphSSM, a graph state space model for modeling the dynamics of temporal graphs. Extensive experimental results demonstrate the effectiveness of our GraphSSM framework across various temporal graph benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2406_00943
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle State Space Models on Temporal Graphs: A First-Principles Study
Li, Jintang
Wu, Ruofan
Jin, Xinzhou
Ma, Boqun
Chen, Liang
Zheng, Zibin
Machine Learning
Artificial Intelligence
Over the past few years, research on deep graph learning has shifted from static graphs to temporal graphs in response to real-world complex systems that exhibit dynamic behaviors. In practice, temporal graphs are formalized as an ordered sequence of static graph snapshots observed at discrete time points. Sequence models such as RNNs or Transformers have long been the predominant backbone networks for modeling such temporal graphs. Yet, despite the promising results, RNNs struggle with long-range dependencies, while transformers are burdened by quadratic computational complexity. Recently, state space models (SSMs), which are framed as discretized representations of an underlying continuous-time linear dynamical system, have garnered substantial attention and achieved breakthrough advancements in independent sequence modeling. In this work, we undertake a principled investigation that extends SSM theory to temporal graphs by integrating structural information into the online approximation objective via the adoption of a Laplacian regularization term. The emergent continuous-time system introduces novel algorithmic challenges, thereby necessitating our development of GraphSSM, a graph state space model for modeling the dynamics of temporal graphs. Extensive experimental results demonstrate the effectiveness of our GraphSSM framework across various temporal graph benchmarks.
title State Space Models on Temporal Graphs: A First-Principles Study
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2406.00943