| 제목 |
TRAFFIC FLOW FORECASTING OF GRAPH CONVOLUTIONAL NETWORK BASED ON SPATIO-TEMPORAL ATTENTION MECHANISM
|
| 분야 |
ADAS, AI, Autonomous Vehicles |
| 언어 |
English |
| 저자 |
Hong Zhang(Lanzhou University of Technology), Linlong Chen(Lanzhou University of Technology), Jie Cao(Lanzhou University of Technology), Xijun Zhang(Lanzhou University of Technology), Sunan Kan (Lanzhou University of Technology), Tianxin Zhao(Lanzhou University of Technology) |
| Key Words |
Traffic flow forecasting, Spatio-temporal attention mechanism, Graph convolutional network, Spatio- temporal correlation, Gated fusion mechanism |
| 초록 |
Accurate traffic flow forecasting is a prerequisite guarantee for the realization of intelligent transportation. Due to the complex time and space features of traffic flow, its forecasting has always been a research hotspot in this field. Aiming at the difficulty of capturing and modelling the temporal and spatial correlation and dynamic features of traffic flow, this paper proposes a novel graph convolutional network traffic flow forecasting model (STAGCN) based on the temporal and spatial attention mechanism. STAGCN model is mainly composed of three modules: Spatio-temporal Attention (STA-Block), Graph Convolutional Network (GCN) and Standard Convolutional Network (CN), model the periodicity, spatial correlation and time dependence of traffic flow respectively. STA-Block module models the spatio-temporal correlation between different time steps through the spatio-temporal attention mechanism and gating fusion mechanism, and uses GCN and CN to capture the spatial and temporal features of traffic flow respectively. Finally, the output of the three components is predicted through a gated fusion mechanism. A large number of experiments have been conducted on two data sets of PeMS. The experimental results demonstrate that compared with the baseline method, the STAGCN model proposed in this paper has better forecasting performance.
|
| 미리보기 |
다운로드
|
| 원문(PDF) |
다운로드
|
