English

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

Machine Learning 2024-12-18 v2 Artificial Intelligence Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2412.10912,
  title  = {ST-FiT: Inductive Spatial-Temporal Forecasting with Limited Training Data},
  author = {Zhenyu Lei and Yushun Dong and Jundong Li and Chen Chen},
  journal= {arXiv preprint arXiv:2412.10912},
  year   = {2024}
}
R2 v1 2026-06-28T20:35:24.074Z