Traffic Flow Prediction Through the Fusion of Spatial-Temporal Data and Points of Interest

Wanzhi Xiao,Li Kuang,Ying An
DOI: https://doi.org/10.1007/978-3-030-86472-9_29
2021-01-01
Abstract:Urban traffic flow prediction plays an important role in alleviating traffic congestion and helping manage public safety. However, the existing solutions to traffic flow prediction still fail to fuse various heterogeneous information effectively, such as historical traffic flow and the semantics of locations, and also fail to capture the traffic flow correlation in the spatial dimension at different levels. Therefore, in this paper, we propose a new traffic flow prediction method to predict the inflow and outflow of each region in the city. Specifically, we first fuse the historical traffic flow, POI, and timestamps data crosswise through a series of gating layers to learn the spatial-temporal interactions. Second, we propose a hierarchical adaptive graph convolution network, which generates multiple learnable adjacency matrices at different network layers to capture the flow correlations in the spatial dimension at different levels. Finally, we take the prediction of inflow, outflow, and overall flow as three tasks and design multi-task learning to improve the prediction performance. Experiments on a real traffic dataset show that our method outperforms the state-of-the-art solutions.
What problem does this paper attempt to address?