Traffic Flow Prediction Using Multi-View Graph Convolution and Masked Attention Mechanism

Lingqiang Chen,Pei Shi,Guanghui Li,Tao Qi
DOI: https://doi.org/10.1016/j.comcom.2022.08.008
IF: 5.047
2022-01-01
Computer Communications
Abstract:Traffic flow prediction is one of the essential technologies in the intelligent transportation system. Graph convolution is naturally suitable for graph structure data mining and is widely used to forecast traffic data. However, most graph convolution methods rely on the Euclidean distance adjacency matrix, which cannot mine richer semantic information. In addition, regardless of static or dynamic graph convolution methods, many graph convolution methods adopt the same adjacency matrix to mine graph information for each convolution layer. In reality, the proportion of abnormal traffic flows such as congestion is slight. Therefore, those networks tend to learn normal traffic patterns. This paper proposes a deep learning model including a dilated Temporal causal convolution module, multi-view diffusion Graph convolution module, and masked multi-head Attention module (TGANet) to address the above issues. The dilated temporal causal convolution is used to mine the temporal correlation of traffic data. Multi-view diffusion graph convolution operation leverages non-euclidean metrics like Jaccard distance and Pearson correlation to mine semantic information. In addition, the masked multi-head attention module is applied to mine fine-grained spatiotemporal dynamics. Experimental results on two public traffic data sets demonstrate that the proposed method outperforms other counterpart methods in most cases.
What problem does this paper attempt to address?