Introduction
Bimonthly, started in 1957
Administrator
Shanxi Provincial Education Department
Sponsor
Taiyuan University of Technology
Publisher
Ed. Office of Journal of TYUT
Editor-in-Chief
SUN Hongbin
ISSN: 1007-9432
CN: 14-1220/N
Administrator
Shanxi Provincial Education Department
Sponsor
Taiyuan University of Technology
Publisher
Ed. Office of Journal of TYUT
Editor-in-Chief
SUN Hongbin
ISSN: 1007-9432
CN: 14-1220/N
location:
home> Online First

Dynamic Graph Convolution Network with Multi-head Attention for Traffic Flow Prediction
DOI:
10.16355/j.tyut.1007-9432.2023BD004
abstract:
【Purposes】Traffic flow prediction is crucial for the effective management and operation of urban transportation systems. The flows of different road sections or intersections in a traffic network not only change dynamically with time, but also the flows of spatially neighboring road sections or intersections affect each other. 【Methods】In order to better learn the spatial and temporal correlation of the traffic flow of different road sections or intersections from the traffic flow sequences, and to improve the performance of short-term prediction of traffic flow, this paper proposes a traffic flow prediction method based on Dynamic Graph Convolution Network with Multi-head Attention (DGCNMA). The DGCNMA model firstly introduces graph convolution networks into the Transformer framework to learn the spatial embedding of traffic flow sequences and incorporate them into the traffic flow sequences, and then adopts the mechanism of Multi-head Attention to capture the temporal and spatial correlation of the traffic flow sequences from multiple perspectives at the same time; secondly, the Interactive Dynamic Graph Convolution Network is introduced to simultaneously learn the local and global spatial-temporal correlations of traffic flow sequences through interactive learning of convolutional network and dynamic graph convolutional network, and interactive fusion of parity subsequence features. 【Findings】Experiments on highway traffic flow datasets (PEMS03, PEMS04, PEMS08) and subway crowd flow datasets (HZME inflow and HZME outflow) show that the proposed DGCNMA model has better traffic flow prediction performance than the baseline models.
Keywords:
traffic flow prediction; multi-head attention; interactive dynamic graph convolution network