Attention Based Graph Bi-LSTM Networks for Traffic Forecasting
Han Zhao, Huan Yang, Yu Wang, Danwei Wang, Rong Su
Abstract
Traffic forecasting is of great importance to vehicle routing, traffic signal control and urban planning. However, traffic forecasting task is challenging due to several factors, such as complex spatial topological structure and dynamic changing of traffic status. Most existing methods have limited ability to capture both spatial and temporal dependence of traffic data. In this paper, we propose a novel end-to-end deep learning model, Attention based Graph Bi-LSTM networks (AGBN) to perform the traffic forecasting task. It uses graph convolutional network (GCN) to extract spatial features and bidirectional long short-term memory networks (Bi-LSTM) to capture the temporal dependence. The attention mechanism is used to select relevant features at all time steps. Experiments show that our model could extract both spatial and temporal dependence well and outperforms other baselines on real-world traffic datasets.