PatchSTG: Scalable Spatiotemporal Graph Transformers for Traffic Forecasting on Irregular Sensor Networks

arXiv cs.LG Papers
traffic-forecasting spatiotemporal graph-transformers patch-based sensor-networks scalable

Summary

PatchSTG introduces a patch-based spatiotemporal graph Transformer for traffic forecasting on irregular sensor networks, achieving near-linear complexity while maintaining competitive performance.

arXiv:2606.09872v1 Announce Type: new Abstract: Traffic forecasting is a fundamental component of intelligent transportation systems, yet remains challenging in real-world settings due to irregular sensor distributions and the high computational cost of modeling large-scale spatiotemporal dependencies. In practical traffic networks, sensors are unevenly distributed across regions, leading to non-uniform spatial structures that limit the effectiveness and scalability of existing graph-based and attention-based models. To address these challenges, we propose PatchSTG, a patch-based spatiotemporal graph Transformer designed for efficient forecasting on irregular sensor networks. The key idea is to introduce a hierarchical spatial representation that partitions sensors into balanced, locality-preserving patches based on geographic information. On top of this structure, a dual attention encoder alternates between intra-patch attention for capturing local interactions and inter-patch attention for modeling global dependencies, reducing computational complexity from quadratic to near-linear scaling. We evaluate PatchSTG on real-world traffic data from Rhode Island and additional large-scale datasets. Experimental results demonstrate that the proposed model achieves stable and competitive forecasting performance across multiple horizons, while significantly improving computational efficiency. Ablation studies further validate the effectiveness of spatial partitioning and dual attention in capturing both local and long-range traffic dynamics. These results suggest that patch-based spatiotemporal modeling provides a scalable and effective framework for traffic forecasting under irregular spatial settings.
Original Article
View Cached Full Text

Cached at: 06/10/26, 06:14 AM 

# PatchSTG: Scalable Spatiotemporal Graph Transformers for Traffic Forecasting on Irregular Sensor Networks
Source: [https://arxiv.org/abs/2606.09872](https://arxiv.org/abs/2606.09872)
[View PDF](https://arxiv.org/pdf/2606.09872)

> Abstract:Traffic forecasting is a fundamental component of intelligent transportation systems, yet remains challenging in real\-world settings due to irregular sensor distributions and the high computational cost of modeling large\-scale spatiotemporal dependencies\. In practical traffic networks, sensors are unevenly distributed across regions, leading to non\-uniform spatial structures that limit the effectiveness and scalability of existing graph\-based and attention\-based models\. To address these challenges, we propose PatchSTG, a patch\-based spatiotemporal graph Transformer designed for efficient forecasting on irregular sensor networks\. The key idea is to introduce a hierarchical spatial representation that partitions sensors into balanced, locality\-preserving patches based on geographic information\. On top of this structure, a dual attention encoder alternates between intra\-patch attention for capturing local interactions and inter\-patch attention for modeling global dependencies, reducing computational complexity from quadratic to near\-linear scaling\. We evaluate PatchSTG on real\-world traffic data from Rhode Island and additional large\-scale datasets\. Experimental results demonstrate that the proposed model achieves stable and competitive forecasting performance across multiple horizons, while significantly improving computational efficiency\. Ablation studies further validate the effectiveness of spatial partitioning and dual attention in capturing both local and long\-range traffic dynamics\. These results suggest that patch\-based spatiotemporal modeling provides a scalable and effective framework for traffic forecasting under irregular spatial settings\.

## Submission history

From: Xuanming Shi \[[view email](https://arxiv.org/show-email/ce016d85/2606.09872)\] **\[v1\]**Tue, 2 Jun 2026 09:53:28 UTC \(2,081 KB\)

Similar Articles

Spatiotemporal Graph Transformer for Traffic Intelligence in Edge Computing

arXiv cs.LG

This paper proposes a spatiotemporal graph Transformer framework for traffic forecasting in edge computing, combining graph neural networks for spatial correlations and Transformer self-attention for long-range temporal dependencies. Experiments on real-world cellular data show it outperforms recurrent graph-based baselines like GCN-LSTM and GCN-GRU.

STKAN: Kolmogorov-Arnold Networks for Spatio-Temporal Forecasting

arXiv cs.LG

This paper introduces STKAN, a spatio-temporal forecasting architecture that integrates Taylor-polynomial Kolmogorov-Arnold Network modules for spatial and temporal token mixing. Experiments on five traffic benchmarks show competitive performance, suggesting nonlinear function approximators can complement architectural design.