Skip to content
Open access

STGFormer: Spatio-Temporal Graph Transformer for Traffic Flow Prediction in Sparse-Sensing Scenarios

Aug 2026 · Italian National Conference on Sensors · Vol 26, pp. 5082 · 0 citations · 21 references
Medicine

TL;DR

An improved Transformer prediction model that integrates a graph convolutional network (GCN) and a self-attention mechanism is proposed for traffic flow prediction, combining temporal self-attention and learnable temporal encoding to capture both long-term traffic evolution patterns and sudden fluctuations.

Abstract

Highlights What are the main findings? An enhanced spatio-temporal graph Transformer (STGFormer) is proposed for traffic flow prediction, combining temporal self-attention and learnable temporal encoding to capture both long-term traffic evolution patterns and sudden fluctuations. Weather data are incorporated into the prediction process to achieve robust prediction under complex environmental conditions. What are the implications of the main findings? An effective solution is provided for traffic flow prediction in sparse-sensing scenarios. The applicability of traffic prediction models is extended to complex and heterogeneous real-world environments. Abstract With the continuous improvement in intelligent transportation and data perception levels, determining how to achieve high-precision and generalizable traffic flow prediction based on historical traffic data has become an important issue in intelligent highway management. The Transformer model, with its strong temporal modeling capabilities, has gradually become a research hotspot in time series prediction. However, its original structure has certain limitations in modeling spatial dependencies, making it difficult to fully exploit the topological relationships of the traffic network, and it has weak adaptability to external environmental changes. To address these issues, this paper proposes an improved Transformer prediction model that integrates a graph convolutional network (GCN) and a self-attention mechanism. The model captures spatial topological information through the GCN module, introduces temporal self-attention mechanism and temporal encoding to enhance the modeling of temporal features, and combines weather factors to achieve perception modeling of external disturbances. In the experimental design, considering the uneven distribution of perception resources in reality, the model input only uses the historical traffic data of some nodes, and different node coverage rates are set to test the performance of the model under sparse input conditions. The results show that the model can maintain good accuracy and stability under multiple coverage rates, verifying the effectiveness and application prospects of the structural improvement.

Read PDF

Similar papers

Open access 2026

Enhanced Traffic Prediction Using Spatial–Temporal Graph Neural Networks

The ability to forecast traffic conditions in urban environments is essential for intelligent transport systems because it provides proactive congestion management, traffic control and informed urban mobility planning. However, due to the extreme spatial and temporal volatility of traffic flow patterns, conventional statistical, machine learning, and sequence-based deep learning approaches either fail to account for the spatial relationships between highway segments or fail to sufficiently model long-range temporal dynamics. To address these gaps, the present study uses a Spatial-Temporal Graph Neural Network (STGNN) framework that can learn a combination of spatial and temporal relationships in road networks and changing time-varying patterns in traffic flow. The model has a graph-based architecture where graph-convolutional layers are combined with gated recurrent units and transformer-based attention units, thus creating a hybrid architecture that is capable of multi-scale spatio-temporal dependencies. The METR-LA dataset was used in experiments, and it was observed that STGNN had smaller Mean Absolute Percentage Error(MAPE), Mean Absolute Error(MAE), and Root Mean Squared Error(RMSE) at 15, 30, and 60 minute horizons of prediction than the baseline and the implemented models: Sequence to Sequence (Seq2Seq) and Temporal Graph Convolutional Network (T-GCN). The model achieved the best results for the 15-minute interval with an MAE of 2.71, an RMSE of 5.17, and a MAPE of 7.08. The results imply that adaptive spatial learning, together with the temporal sequence modeling, can produce much better forecasting stability, which underlines the potential STGNN-based traffic prediction systems have to contribute to real-time traffic control.

Akarsh Kumar, Aishwarya Sinha, Yadvendra Pratap Singh · 0 citations
Open access Jul 2026

Multi-Perspective Spatio-Temporal Feature Fusion Model for Urban Traffic Flow Prediction

Urban traffic flow is difficult to forecast accurately because its evolution is non-linear and governed by dependencies that operate over different spatial and temporal ranges. This paper introduces the Multi-Perspective Spatio-Temporal Feature Fusion Model (MPSTFFM) to describe these dependencies through complementary views. The temporal signal is separated into a slowly varying trend and a residual fluctuation, while the spatial structure is represented by four graphs: first-order adjacency, second-order in-degree, second-order out-degree, and a data-adaptive graph. These graphs respectively encode physical road connectivity, common inflow sources, common outflow destinations, and latent spatial associations. Whereas the first three are constructed from the known network topology, the adaptive graph is learned together with the prediction model and can therefore identify correlations not expressed by physical links. Within each spatio-temporal view, self-attention captures dependencies over long ranges, and convolutional operations extract local patterns. The features learned from all views are subsequently fused into a high-dimensional representation used to predict future flow. Experiments on real-world datasets compare MPSTFFM with twelve methods published during the preceding five years. On these benchmarks MPSTFFM outperforms every baseline, lowering the average MAE, RMSE, and MAPE across the four datasets by 13.04%, 5.28%, and 9.59%, respectively, relative to the best baseline on each one.

A. Marakhimov, Rustem Jalelov, J.K. Kudaybergenov et al. · 0 citations
Open access 2026

Robustness of Spatio-Temporal Graph Neural Networks for Traffic Forecasting Under Realistic Sensor Degradation

A robust focused comparative evaluation of seven traffic forecasting approaches suggests that traffic forecasting models should be assessed not only by clean-data accuracy but also by their robustness under degraded sensing conditions before deployment in real intelligent transportation systems.

Shreya N. Desai, Kasim Ishaque Ghanchi, Ali Mehdi Mirza et al. · 0 citations
Preprint Aug 2026

Spatiotemporal Graph Transformer for Traffic Intelligence in Edge Computing

Accurate traffic forecasting is essential for proactive resource management in edge computing, where service demand evolves dynamically across both space and time. In practical cellular edge systems, traffic exhibits strong spatial correlations among neighboring service regions and long-range temporal dependencies driven by user mobility and application behavior. Existing recurrent forecasting approaches can capture short-term dynamics but often struggle to model long-horizon traffic evolution under non-stationary conditions. To address this challenge, we propose a spatiotemporal graph Transformer framework that jointly models spatial interactions and temporal dependencies for traffic forecasting in edge computing. The framework employs graph neural networks to capture spatial correlations among service regions and leverages Transformer-based self-attention to learn long-range temporal patterns from historical traffic observations. By decoupling spatial representation learning from temporal reasoning, the proposed approach provides an effective mechanism for large-scale spatiotemporal traffic modeling. Extensive experiments on a real-world cellular network dataset demonstrate that the proposed graph Transformer consistently outperforms recurrent graph-based baselines, including GCN-RNN, GCN-LSTM, and GCN-GRU models, across multiple forecasting horizons. The resulting forecasts enable more effective proactive resource provisioning and reduce overload risk compared with reactive management strategies. These results highlight the potential of graph-enhanced attention mechanisms for building intelligent and adaptive edge computing systems.

Laha Ale, Letian Lin, Na Cao et al. · 0 citations