Dynamic Correlation Adjacency-Matrix-Based Graph Neural Networks for Traffic Flow Prediction

Sensors (Basel). 2023 Mar 7;23(6):2897. doi: 10.3390/s23062897.

Abstract

Modeling complex spatial and temporal dependencies in multivariate time series data is crucial for traffic forecasting. Graph convolutional networks have proved to be effective in predicting multivariate time series. Although a predefined graph structure can help the model converge to good results quickly, it also limits the further improvement of the model due to its stationary state. In addition, current methods may not converge on some datasets due to the graph structure of these datasets being difficult to learn. Motivated by this, we propose a novel model named Dynamic Correlation Graph Convolutional Network (DCGCN) in this paper. The model can construct adjacency matrices from input data using a correlation coefficient; thus, dynamic correlation graph convolution is used for capturing spatial dependencies. Meanwhile, gated temporal convolution is used for modeling temporal dependencies. Finally, we performed extensive experiments to evaluate the performance of our proposed method against ten existing well-recognized baseline methods using two original and four public datasets.

Keywords: dynamic adjacency matrix; graph neural networks; multivariate time series; traffic prediction.

Grants and funding

This research received no external funding.