Adaptive Spatiotemporal Transformer Graph Network for Traffic Flow Forecasting by IoT Loop Detectors

Extensive traffic flow data are received from the loop detector networks every second, which requires us to develop an effective and efficient algorithm to predict future traffic flow. However, dynamic traffic conditions on a road are not just influenced by sequential patterns in the temporal dimension, but also by other roadways in the spatial dimension. Although many successful models have been developed in previous studies to forecast future traffic flows, most of them have shortcomings in modeling spatial and temporal dependencies. In this article, we focus on spatial-temporal factors and propose a new adaptive spatial-temporal transformer graph network (ASTTGN) to improve the accuracy of traffic forecasting by jointly modeling the spatial-temporal information of road networks. Specifically, we propose an adaptive spatial-temporal transformer module, which contains two developed adaptive transformer modules for capturing dynamic spatial dependence and temporal dependence across multiple time steps, respectively. Finally, feature fusion is performed through a gated feature aggregation layer to simulate the effect of complex spatial-temporal factors on traffic conditions. In particular, the multihead attention mechanism employed by the transformer can effectively explore the potential spatial-temporal dependence patterns in different subspaces. Experimental results on two real-world traffic data sets demonstrate the superiority of the proposed model compared to existing techniques.