arXiv:2509.13735cs.LGcs.AI2025-09

首个面向有向图的状态空间模型,提升长程因果建模效率

State Space Models over Directed Graphs

  • 用k跳局部子图将有向图序列化,实现状态空间模型适配
  • 在3个有向图任务上达顶尖性能,训练速度提升1.5至2倍
  • 适合需要高效处理大规模有向图的科研与工业场景

有向图在多个领域中普遍存在,边的方向编码了关键因果关系。然而,现有针对有向图设计的图神经网络和图Transformer面临两大挑战:(1)有效捕捉由有向边带来的长程因果依赖;(2)在处理大规模图数据集时平衡精度与训练效率。近年来,状态空间模型(SSMs)在因果序列任务中取得显著进展,其图结构变体已在多种图学习基准上展现出顶尖精度与高效率。但现有图状态空间模型仅适用于无向图,限制了其在有向图学习中的表现。为此,我们提出创新方法DirEgo2Token,通过k跳邻域子图对有向图进行序列化。这是首个系统性将状态空间模型拓展至有向图学习的工作。基于此,我们构建了新的有向图神经网络架构DirGraphSSM,通过消息传递机制在有向图上实现状态空间模型。实验表明,DirGraphSSM在三个代表性有向图学习任务上达到当前最优性能,同时在另外两个任务上保持竞争力,并相较现有最先进模型实现1.5×至2×的训练加速。

原文摘要 · Abstract (English)

Directed graphs are ubiquitous across numerous domains, where the directionality of edges encodes critical causal dependencies. However, existing GNNs and graph Transformers tailored for directed graphs face two major challenges: (1) effectively capturing long-range causal dependencies derived from directed edges; (2) balancing accuracy and training efficiency when processing large-scale graph datasets. In recent years, state space models (SSMs) have achieved substantial progress in causal sequence tasks, and their variants designed for graphs have demonstrated state-of-the-art accuracy while maintaining high efficiency across various graph learning benchmarks. However, existing graph state space models are exclusively designed for undirected graphs, which limits their performance in directed graph learning. To this end, we propose an innovative approach DirEgo2Token which sequentializes directed graphs via k-hop ego graphs. This marks the first systematic extension of state space models to the field of directed graph learning. Building upon this, we develop DirGraphSSM, a novel directed graph neural network architecture that implements state space models on directed graphs via the message-passing mechanism. Experimental results demonstrate that DirGraphSSM achieves state-of-the-art performance on three representative directed graph learning tasks while attaining competitive performance on two additional tasks with 1.5$\times $ to 2$\times $ training speed improvements compared to existing state-of-the-art models.

有向图状态空间模型图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。