ST-TDCN: A two-channel tree-structure spatial–temporal convolutional network model for traffic velocity prediction

Abstract

In the development of intelligent transport systems which aims to provide safe, convenient, and comfortable transport and effectively addresses issues with traffic congestion and pollution, accurate collection of spatial–temporal data is of great significance. Based on the study of the topology and spatial characteristics of the transportation network, data with spatial–temporal attributed can be predicted by relevant models. Thus, the purpose of providing additional information for decision making and labor cost saving is achieved. This study proposed a more accurate traffic velocity prediction model, named Spatial-Temporal Tree-structure Dual-channel Convolution Network. The model designs a spatial tree convolution module for capturing the spatial features of nodes in the traffic network represented by the tree structure and applies a dual-channel temporal convolutional network module to capture the temporal information of velocity data more accurately. Comparative tests have demonstrated that the model proposed in this study has the advantages of higher accuracy, better convergence and more flexibility in responding to dynamic changes of traffic velocity.