arXiv:2501.04239cs.LG2025-01KDD被引 12

动态建模时空依赖,提升图神经网络效率与精度

Dynamic Localisation of Spatial-Temporal Graph Neural Network

  • 引入动态局部化机制,实时生成稀疏时空图
  • 在9个真实数据集上超越现有基准,分布式部署效率显著提升
  • 适合大规模分布式时空数据建模场景

时空数据在众多智能应用中至关重要,揭示了特定位置当前测量值与历史数据间潜在的因果关联。自适应时空图神经网络(ASTGNNs)通过数据驱动方式建模这些依赖关系,相比预定义图结构更具准确性,但计算开销更大。本文提出动态局部化视角,认为空间依赖应随时间动态演变。为此设计了DynAGS框架,集成动态局部化、时变空间图和个性化局部化,核心为轻量级动态图生成器,利用跨注意力机制以节点无关方式融合历史信息,提升当前节点特征表示,并据此生成无需昂贵数据交换的动态稀疏图,支持个性化局部化。在两种主流ASTGNN架构及九个不同应用的真实数据集上的实验表明,DynAGS显著优于现有基准,验证了动态建模空间依赖可大幅提升模型表达能力、灵活性与系统效率,尤其适用于分布式部署场景。

原文摘要 · Abstract (English)

Spatial-temporal data, fundamental to many intelligent applications, reveals dependencies indicating causal links between present measurements at specific locations and historical data at the same or other locations. Within this context, adaptive spatial-temporal graph neural networks (ASTGNNs) have emerged as valuable tools for modelling these dependencies, especially through a data-driven approach rather than pre-defined spatial graphs. While this approach offers higher accuracy, it presents increased computational demands. Addressing this challenge, this paper delves into the concept of localisation within ASTGNNs, introducing an innovative perspective that spatial dependencies should be dynamically evolving over time. We introduce \textit{DynAGS}, a localised ASTGNN framework aimed at maximising efficiency and accuracy in distributed deployment. This framework integrates dynamic localisation, time-evolving spatial graphs, and personalised localisation, all orchestrated around the Dynamic Graph Generator, a light-weighted central module leveraging cross attention. The central module can integrate historical information in a node-independent manner to enhance the feature representation of nodes at the current moment. This improved feature representation is then used to generate a dynamic sparse graph without the need for costly data exchanges, and it supports personalised localisation. Performance assessments across two core ASTGNN architectures and nine real-world datasets from various applications reveal that \textit{DynAGS} outshines current benchmarks, underscoring that the dynamic modelling of spatial dependencies can drastically improve model expressibility, flexibility, and system efficiency, especially in distributed settings.

图神经网络时空建模动态图分布式

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