10 research outputs found
Spatial-temporal traffic modeling with a fusion graph reconstructed by tensor decomposition
Accurate spatial-temporal traffic flow forecasting is essential for helping
traffic managers to take control measures and drivers to choose the optimal
travel routes. Recently, graph convolutional networks (GCNs) have been widely
used in traffic flow prediction owing to their powerful ability to capture
spatial-temporal dependencies. The design of the spatial-temporal graph
adjacency matrix is a key to the success of GCNs, and it is still an open
question. This paper proposes reconstructing the binary adjacency matrix via
tensor decomposition, and a traffic flow forecasting method is proposed. First,
we reformulate the spatial-temporal fusion graph adjacency matrix into a
three-way adjacency tensor. Then, we reconstructed the adjacency tensor via
Tucker decomposition, wherein more informative and global spatial-temporal
dependencies are encoded. Finally, a Spatial-temporal Synchronous Graph
Convolutional module for localized spatial-temporal correlations learning and a
Dilated Convolution module for global correlations learning are assembled to
aggregate and learn the comprehensive spatial-temporal dependencies of the road
network. Experimental results on four open-access datasets demonstrate that the
proposed model outperforms state-of-the-art approaches in terms of the
prediction performance and computational cost.Comment: 11 pages, 8 figure