GDFormer: A Graph Diffusing Attention based approach for Traffic Flow Prediction

Jie Su

doi:10.1016/j.patrec.2022.03.005

GDFormer: A Graph Diffusing Attention based approach for Traffic Flow Prediction

2022-04-01

11

PRE

AI

DOI：10.1016/j.patrec.2022.03.005

原文链接

原文求助

摘要

En 中文

In this paper, we propose a novel traffic flow prediction approach, called as Graph Diffusing trans-Former (GDFormer). GDFormer is in architecture of transformer, which is composed by the encoder sequence and decoder sequence. both of the encoder sequence and decoder sequence in GDFormer are constituted by the novel designed Graph Diffusing Attention (GDA) module and the auxiliaries. The GDA module utilizes the query-key-value attention to learn the diffusion parameters for each diffusion step, and dynamically updates the adjacency transition, which reflects the dynamically changing traffic flow between the traffic monitors. To verify the efficiency of our approach, we conduct a lot of experiments on two real-world data sets. With a comparison between our approach and the benchmarks, we find that our approach has achieved state of the art performance. Ablation experiments are conducted to illustrate the effectiveness of the key components in the model. For ease of reproducibility, the code, the processed real-world data sets and the evaluation results are available at https://github.com/dublinsky/GDFormer . (c) 2022 Elsevier B.V. All rights reserved.

Keyword:

Graph neural network

Diffusion process

Attention mechanism

Traffic flow prediction

GDFormer: A Graph Diffusing Attention based approach for Traffic Flow Prediction

摘要

期刊

机构

引用论文