An improved dynamic Chebyshev graph convolution network for traffic flow prediction with spatial-temporal attention

Lyuchao Liao,Zhiyuan Hu,Yuxin Zheng,Shuoben Bi,Fumin Zou,Huai Qiu,Maolin Zhang
DOI: https://doi.org/10.1007/s10489-021-03022-w
IF: 5.3
2022-03-19
Applied Intelligence
Abstract:Accurate traffic flow prediction plays a significant role in urban traffic management, including traffic congestion control and public travel route planning. Recently, several approaches have been put forward to learn the patterns from historical traffic data. However, there exist some limitations resulting from the use of the static learning method to explore the dynamical characteristics of the road network. Besides, the dynamic global temporal and spatial properties are not considered in these models. These drawbacks lead to a low prediction performance and make applying to a more extensive road network challenging. To address these issues, from the inspiration of Chebyshev polynomial, we proposed an improved dynamic Chebyshev graph convolution neural network model called iDCGCN. In the proposed approach, a novel updating method for the Laplacian matrix, which approximately constructs features from different period data, is proposed based on the attention mechanism. In addition, a novel feature construction method is proposed to integrate long-short temporal and local-global spatial features for complex traffic flow representation. Experimental results have shown that iDCGCN outperforms the state-of-the-art GCN-based methods on four real-world highway traffic datasets.
computer science, artificial intelligence
What problem does this paper attempt to address?