Towards Scalable and Structured Spatiotemporal Forecasting

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Hongyi, Li, Xiucheng, Chen, Xinyang, Li, Jing, Chen, Kehai, Nie, Liqiang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909800044953600
author Chen, Hongyi
Li, Xiucheng
Chen, Xinyang
Li, Jing
Chen, Kehai
Nie, Liqiang
author_facet Chen, Hongyi
Li, Xiucheng
Chen, Xinyang
Li, Jing
Chen, Kehai
Nie, Liqiang
contents In this paper, we propose a novel Spatial Balance Attention block for spatiotemporal forecasting. To strike a balance between obeying spatial proximity and capturing global correlation, we partition the spatial graph into a set of subgraphs and instantiate Intra-subgraph Attention to learn local spatial correlation within each subgraph; to capture the global spatial correlation, we further aggregate the nodes to produce subgraph representations and achieve message passing among the subgraphs via Inter-subgraph Attention. Building on the proposed Spatial Balance Attention block, we develop a multiscale spatiotemporal forecasting model by progressively increasing the subgraph scales. The resulting model is both scalable and able to produce structured spatial correlation, and meanwhile, it is easy to implement. We evaluate its efficacy and efficiency against the existing models on real-world spatiotemporal datasets from medium to large sizes. The experimental results show that it can achieve performance improvements up to 7.7% over the baseline methods at low running costs.
format Preprint
id arxiv_https___arxiv_org_abs_2509_18115
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Towards Scalable and Structured Spatiotemporal Forecasting
Chen, Hongyi
Li, Xiucheng
Chen, Xinyang
Li, Jing
Chen, Kehai
Nie, Liqiang
Machine Learning
In this paper, we propose a novel Spatial Balance Attention block for spatiotemporal forecasting. To strike a balance between obeying spatial proximity and capturing global correlation, we partition the spatial graph into a set of subgraphs and instantiate Intra-subgraph Attention to learn local spatial correlation within each subgraph; to capture the global spatial correlation, we further aggregate the nodes to produce subgraph representations and achieve message passing among the subgraphs via Inter-subgraph Attention. Building on the proposed Spatial Balance Attention block, we develop a multiscale spatiotemporal forecasting model by progressively increasing the subgraph scales. The resulting model is both scalable and able to produce structured spatial correlation, and meanwhile, it is easy to implement. We evaluate its efficacy and efficiency against the existing models on real-world spatiotemporal datasets from medium to large sizes. The experimental results show that it can achieve performance improvements up to 7.7% over the baseline methods at low running costs.
title Towards Scalable and Structured Spatiotemporal Forecasting
topic Machine Learning
url https://arxiv.org/abs/2509.18115