Multivariate correlation matrix-based deep learning model with enhanced heuristic optimization for short-term traffic forecasting

Shuai Zhang,Kun Zhu,Wenyu Zhang
DOI: https://doi.org/10.1109/tkde.2021.3118389
IF: 9.235
2021-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:Accurately capturing the spatial correlations of traffic network significantly benefits short-term traffic forecasting. Some existing works represent spatial correlations in a simple one-dimensional space, but they cannot represent the real spatial correlations among sensors comprehensively. The other existing works represent the spatial correlations through grid-based method, but the local correlation of constructed spatial map is too superficial to extract deep spatial features effectively. Therefore, a novel deep learning model is proposed, which aims to represent the spatial correlations more effectively through a new correlation matrix structure. In the proposed model, the correlations among sensors are calculated from multiple perspectives to construct the speed, volume, and occupancy correlation matrices respectively. Then, considering that highly correlated sensors are close in the spatial dimension, an enhanced heuristic optimization algorithm is proposed to evolve these three correlation matrices into optimal ones by reorganizing the highly correlated sensors into each other's neighborhood. Finally, the three optimal correlation matrices are combined to form a three-dimensional multivariate correlation matrix characterized by locally high correlation, which is beneficial to exploit the deep spatial features of traffic network. The experiments show that the proposed model has better accuracy and stability than other commonly used baseline models.
computer science, information systems, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?