DyG-Mamba: Continuous State Space Modeling on Dynamic Graphs

Loading...
Thumbnail Image
File version

Version of Record (VoR)

Author(s)
Li, Dongyuan
Tan, Shiyin
Zhang, Ying
Jin, Ming
Pan, Shirui
Okumura, Manabu
Jiang, Renhe
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2025
Size
File type(s)
Location

San Diego, United States

Abstract

Dynamic graph modeling aims to uncover evolutionary patterns in real-world systems, enabling accurate social recommendation and early detection of cancer cells. Inspired by the success of recent state space models in efficiently capturing long-term dependencies, we propose DyG-Mamba by translating dynamic graph modeling into a long-term sequence modeling problem. Specifically, inspired by Ebbinghaus' forgetting curve, we treat the irregular timespans between events as control signals, allowing DyG-Mamba to dynamically adjust the forgetting of historical information. This mechanism ensures effective usage of irregular timespans, thereby improving both model effectiveness and inductive capability. In addition, inspired by Ebbinghaus' review cycle, we redefine core parameters to ensure that DyG-Mamba selectively reviews historical information and filters out noisy inputs, further enhancing the model’s robustness. Through exhaustive experiments on 12 datasets covering dynamic link prediction and node classification tasks, we show that DyG-Mamba achieves state-of-the-art performance on most datasets, while demonstrating significantly improved computational and memory efficiency. Our code is available at https://github.com/Clearloveyuan/DyG-Mamba.

Journal Title
Conference Title

39th Conference on Neural Information Processing Systems (NeurIPS 2025)

Book Title
Edition
Volume
Issue
Thesis Type
Degree Program
School
DOI
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement

This resource is distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (http://creativecommons.org/licenses/by/4.0/), which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.

Item Access Status
Note
Access the data
Related item(s)
Subject

Artificial intelligence

Machine learning

Persistent link to this record
Citation

Li, D; Tan, S; Zhang, Y; Jin, M; Pan, S; Okumura, M; Jiang, R, DyG-Mamba: Continuous State Space Modeling on Dynamic Graphs, 2025