arrow
Return

Multiple Information Spatial-Temporal Attention based Graph Convolution Network for traffic prediction

delete2023-03-01
delete16
PRE
AI
H
Huyin Zhang *
F
Fei Yang
Y
Yonghao Wu
C
Cong Li
DOI:10.1016/j.asoc.2023.110052delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Traffic prediction (forecasting) is a key problem in intelligent transportation. It helps engineers to obtain traffic trends in advance so that they can make favorable decisions quickly and accurately, or even improve the working mode of existing systems. However, it is very challenging to design a model for such problem that fully utilize the factors related to traffic. This paper investigates machine learning in traffic prediction and proposes Multiple Information Spatial-Temporal Attention based Graph Convolution Networks (MISTAGCN). The model consists of two parts. The first part utilizes combinations of different inputs and graph structures to compute the corresponding latent variables. In the second part of the model, the latent variables are comprehensively integrated and deeply mined. In particular, as the basic component of the model, a STBlock integrates mechanisms such as temporal attention, spatial attention, graph convolution, ordinary convolution and residual connections to fully explore the potential information contained in the data. Experiments on Haikou online car-hailing dataset and New York yellow taxi trip dataset illustrate that the proposed model outperforms state-of-art baseline models.(c) 2023 Elsevier B.V. All rights reserved.
Keywords:
Spatial-temporal attention
Graph structure
Multidimensional time series

Journal

Applied Soft Computing cover
Applied Soft Computing
IF:
6.6
Papers:
1.4W
Citations:
4.8W

Organization

W
wuhan university
Scholars:
8.0W
Papers: 5.8W
Citations: 70