| Home | KSAE | E-Submission | Sitemap | Contact Us |  
International Journal of Automotive Technology > Volume 25(3); 2024 > Article
International Journal of Automotive Technology 2024;25(3): 455-468.
doi: https://doi.org/10.1007/s12239-024-00036-4
Traffic Flow Forecasting Based on Transformer with Diffusion Graph Attention Network
Hong Zhang , Hongyan Wang , Linlong Chen , Tianxin Zhao , Sunan Kan
Lanzhou University of Technology
PDF Links Corresponding Author.  Hong Zhang  , Email. zhanghong@lut.edu.cn
Because of the high nonlinearity and complexity, it is still a challenge to forecast traffic flow accurately. Most of the existing methods, which lack the ability to model the dynamic spatio-temporal correlations of traffic flows, rely on a priori knowledge of the graph structure to capture static spatial correlations. However, spatial correlation can be dynamic and the physical structure does not reflect the true relationship between roads. To better capture complex spatio-temporal dependencies and predict traffic conditions on road networks, a new forecasting model based on Transformer and Diffusion Graph Attention Network (T-DGAN) is proposed. This model adopts the Transformer encoder-decoder architecture, where the encoder and decoder contain multiple Spatio-Temporal Convolutional Network Blocks (ST-CB) and Diffusion Graph Attention Blocks (DGA-B), which are utilized to capture the spatio-temporal dependency and the dynamic spatial correlation of traffic flow separately. In addition, an information auxiliary module is added to the decoder to summarize the traffic flow information between the encoder and the decoder. Experiments on different datasets verify that the T-DGAN model has achieved higher forecasting performance than the baseline methods. In PEMS03, the Mean Absolute Error (MAE) and Root Mean Square Error (RMSE) of T-DGAN to 60 min are improved by 4.28% and 3.66%, compared with the state-of-the-art model, respectively.
Key Words: Traffi c forecasting · Transformer · Spatio-temporal correlation · Deep learning
Preview  Preview
Full text via DOI  Full text via DOI
Download Citation  Download Citation
Related article
Editorial Office
21 Teheran-ro 52-gil, Gangnam-gu, Seoul 06212, Korea
TEL: +82-2-564-3971   FAX: +82-2-564-3973   E-mail: manage@ksae.org
About |  Browse Articles |  Current Issue |  For Authors and Reviewers
Copyright © The Korean Society of Automotive Engineers.                 Developed in M2PI
Close layer
prev next