Attention based graph Bi-LSTM networks for traffic forecasting
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...
Saved in:
| Main Authors: | , , , , |
|---|---|
| 其他作者: | |
| 格式: | Conference or Workshop Item |
| 語言: | English |
| 出版: |
2021
|
| 主題: | |
| 在線閱讀: | https://hdl.handle.net/10356/146155 |
| 標簽: |
添加標簽
沒有標簽, 成為第一個標記此記錄!
|
| 機構: | Nanyang Technological University |
| 語言: | English |
| 總結: | 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. |
|---|