arXiv:2412.10912cs.LGcs.AI2024-12AAAI被引 9

解决时空图中节点无历史数据时的预测难题

ST-FiT: Inductive Spatial-Temporal Forecasting with Limited Training Data

  • 通过时间数据增强与空间拓扑学习,提升模型泛化能力
  • 在无训练数据节点上实现优于现有方法的预测精度
  • 适合城市疫情、交通等异步数据场景的预测应用

时空图广泛应用于各类现实场景。时空图神经网络(STGNN)已成为从此类数据中提取有效信息的强大工具。然而,在真实应用中,大多数节点在训练阶段可能缺乏可用的时间数据。例如,地理图上多数城市的疫情动态因爆发时间不同步而无法获取。这一现象与现有时空预测方法的训练需求相悖,严重影响其有效性,阻碍了广泛应用。本文提出一种新的有限训练数据下的归纳式预测问题。给定一个时空图,目标是学习一个可轻松推广至无任何时间训练数据节点的时空预测模型。为此,我们提出一个名为ST-FiT的原理性框架。ST-FiT包含两个关键学习组件:时间数据增强和空间图拓扑学习。该设计使其可无缝集成于任意现有STGNN之上,在无训练数据节点上实现更优性能。大量实验验证了ST-FiT在多个关键维度上的有效性。

原文摘要 · Abstract (English)

Spatial-temporal graphs are widely used in a variety of real-world applications. Spatial-Temporal Graph Neural Networks (STGNNs) have emerged as a powerful tool to extract meaningful insights from this data. However, in real-world applications, most nodes may not possess any available temporal data during training. For example, the pandemic dynamics of most cities on a geographical graph may not be available due to the asynchronous nature of outbreaks. Such a phenomenon disagrees with the training requirements of most existing spatial-temporal forecasting methods, which jeopardizes their effectiveness and thus blocks broader deployment. In this paper, we propose to formulate a novel problem of inductive forecasting with limited training data. In particular, given a spatial-temporal graph, we aim to learn a spatial-temporal forecasting model that can be easily generalized onto those nodes without any available temporal training data. To handle this problem, we propose a principled framework named ST-FiT. ST-FiT consists of two key learning components: temporal data augmentation and spatial graph topology learning. With such a design, ST-FiT can be used on top of any existing STGNNs to achieve superior performance on the nodes without training data. Extensive experiments verify the effectiveness of ST-FiT in multiple key perspectives.

时空预测图神经网络小样本学习

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