IDEAS home Printed from https://ideas.repec.org/a/eee/phsmap/v608y2022ip1s0378437122008329.html
   My bibliography  Save this article

Urban road traffic flow prediction: A graph convolutional network embedded with wavelet decomposition and attention mechanism

Author

Listed:
  • Zheng, Yan
  • Wang, Shengyou
  • Dong, Chunjiao
  • Li, Wenquan
  • Zheng, Wen
  • Yu, Jingcai

Abstract

Urban road traffic flow prediction is the key basis for the development of Intelligent Transportation System. The complex urban structure leads to irregular shape and layout of the road network, which poses a challenge to capture the spatio-temporal correlation of traffic flow at different nodes in the region. In this study, a graph convolutional network model framework embedded with wavelet decomposition and attention mechanism (WDA-GCN) is proposed to predict the traffic flow of each traffic monitor at the regional level by exploring the spatio-temporal correlation among traffic monitors. Specifically, the spatial correlation between different monitors is encoded into two graphs by Graph Convolutional Network (GCN): geographical neighbor graph and functional similarity graph. The Gated Recurrent Unit (GRU) is used to learn the spatial features extracted by GCN, and the attention mechanism is added to improve the prediction accuracy. Finally, the time series data and spatio-temporal correlation of traffic flow are input into the encoder–decoder based on GRU to realize regional traffic flow prediction. The model is validated and compared with the real traffic monitor data in Daxing District of Beijing, China, and the results show that the prediction accuracy of WDA-GCN model can reach 81.03% after embedding wavelet decomposition and attention mechanism, which is better than the traditional time series prediction methods and deep learning methods.

Suggested Citation

  • Zheng, Yan & Wang, Shengyou & Dong, Chunjiao & Li, Wenquan & Zheng, Wen & Yu, Jingcai, 2022. "Urban road traffic flow prediction: A graph convolutional network embedded with wavelet decomposition and attention mechanism," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 608(P1).
  • Handle: RePEc:eee:phsmap:v:608:y:2022:i:p1:s0378437122008329
    DOI: 10.1016/j.physa.2022.128274
    as

    Download full text from publisher

    File URL: http://www.sciencedirect.com/science/article/pii/S0378437122008329
    Download Restriction: Full text for ScienceDirect subscribers only. Journal offers the option of making the article available online on Science direct for a fee of $3,000

    File URL: https://libkey.io/10.1016/j.physa.2022.128274?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    References listed on IDEAS

    as
    1. Okutani, Iwao & Stephanedes, Yorgos J., 1984. "Dynamic prediction of traffic volume through Kalman filtering theory," Transportation Research Part B: Methodological, Elsevier, vol. 18(1), pages 1-11, February.
    2. Xiaoquan Wang & Chunfu Shao & Chaoying Yin & Chengxiang Zhuge & Wenjun Li, 2018. "Application of Bayesian Multilevel Models Using Small and Medium Size City in China: The Case of Changchun," Sustainability, MDPI, vol. 10(2), pages 1-15, February.
    3. Wang, Ke & Ma, Changxi & Qiao, Yihuan & Lu, Xijin & Hao, Weining & Dong, Sheng, 2021. "A hybrid deep learning model with 1DCNN-LSTM-Attention networks for short-term traffic flow prediction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 583(C).
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Sun, Xiaoyong & Chen, Fenghao & Wang, Yuchen & Lin, Xuefen & Ma, Weifeng, 2023. "Short-term traffic flow prediction model based on a shared weight gate recurrent unit neural network," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 618(C).
    2. Zhang, Weibin & Zha, Huazhu & Zhang, Shuai & Ma, Lei, 2023. "Road section traffic flow prediction method based on the traffic factor state network," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 618(C).
    3. Ma, Changxi & Zhao, Mingxi, 2023. "Spatio-temporal multi-graph convolutional network based on wavelet analysis for vehicle speed prediction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 630(C).

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Hu, Guojing & Whalin, Robert W. & Kwembe, Tor A. & Lu, Weike, 2023. "Short-term traffic flow prediction based on secondary hybrid decomposition and deep echo state networks," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 632(P1).
    2. Zhao, Jiandong & Yu, Zhixin & Yang, Xin & Gao, Ziyou & Liu, Wenhui, 2022. "Short term traffic flow prediction of expressway service area based on STL-OMS," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 595(C).
    3. Wang, Yaguan & Qin, Yong & Guo, Jianyuan & Cao, Zhiwei & Jia, Limin, 2022. "Multi-point short-term prediction of station passenger flow based on temporal multi-graph convolutional network," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 604(C).
    4. Xing, Tao & Zhou, Xuesong & Taylor, Jeffrey, 2013. "Designing heterogeneous sensor networks for estimating and predicting path travel time dynamics: An information-theoretic modeling approach," Transportation Research Part B: Methodological, Elsevier, vol. 57(C), pages 66-90.
    5. Wang, Xiaoquan & Yin, Chaoying & Zhang, Junyi & Shao, Chunfu & Wang, Shengyou, 2021. "Nonlinear effects of residential and workplace built environment on car dependence," Journal of Transport Geography, Elsevier, vol. 96(C).
    6. M. Bierlaire & F. Crittin, 2004. "An Efficient Algorithm for Real-Time Estimation and Prediction of Dynamic OD Tables," Operations Research, INFORMS, vol. 52(1), pages 116-127, February.
    7. David Watling & Giulio Cantarella, 2015. "Model Representation & Decision-Making in an Ever-Changing World: The Role of Stochastic Process Models of Transportation Systems," Networks and Spatial Economics, Springer, vol. 15(3), pages 843-882, September.
    8. Cai, Lingru & Zhang, Zhanchang & Yang, Junjie & Yu, Yidan & Zhou, Teng & Qin, Jing, 2019. "A noise-immune Kalman filter for short-term traffic flow forecasting," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 536(C).
    9. Huayuan Chen & Zhizhe Lin & Yamin Yao & Hai Xie & Youyi Song & Teng Zhou, 2024. "Hybrid Extreme Learning for Reliable Short-Term Traffic Flow Forecasting," Mathematics, MDPI, vol. 12(20), pages 1-15, October.
    10. Jiasong Zhu & Anthony Gar-On Yeh, 2012. "A Self-Learning Short-Term Traffic Forecasting System," Environment and Planning B, , vol. 39(3), pages 471-485, June.
    11. Ma, Changxi & Zhang, Bowen & Li, Shukai & Lu, Youpeng, 2024. "Urban rail transit passenger flow prediction with ResCNN-GRU based on self-attention mechanism," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 638(C).
    12. Safikhani, Abolfazl & Kamga, Camille & Mudigonda, Sandeep & Faghih, Sabiheh Sadat & Moghimi, Bahman, 2020. "Spatio-temporal modeling of yellow taxi demands in New York City using generalized STAR models," International Journal of Forecasting, Elsevier, vol. 36(3), pages 1138-1148.
    13. Zhang, Jie & Song, Chunyue & Cao, Shan & Zhang, Chun, 2023. "FDST-GCN: A Fundamental Diagram based Spatiotemporal Graph Convolutional Network for expressway traffic forecasting," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 630(C).
    14. Balaji Ganesh Rajagopal & Manish Kumar & Pijush Samui & Mosbeh R. Kaloop & Usama Elrawy Shahdah, 2022. "A Hybrid DNN Model for Travel Time Estimation from Spatio-Temporal Features," Sustainability, MDPI, vol. 14(21), pages 1-20, October.
    15. Hongxia Ge & Siteng Li & Rongjun Cheng & Zhenlei Chen, 2022. "Self-Attention ConvLSTM for Spatiotemporal Forecasting of Short-Term Online Car-Hailing Demand," Sustainability, MDPI, vol. 14(12), pages 1-16, June.
    16. Lu, Xijin & Ma, Changxi & Qiao, Yihuan, 2021. "Short-term demand forecasting for online car-hailing using ConvLSTM networks," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 570(C).
    17. He, Yuxin & Zhao, Yang & Luo, Qin & Tsui, Kwok-Leung, 2022. "Forecasting nationwide passenger flows at city-level via a spatiotemporal deep learning approach," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 589(C).
    18. Shenghan Zhou & Chaofan Wei & Chaofei Song & Yu Fu & Rui Luo & Wenbing Chang & Linchao Yang, 2022. "A Hybrid Deep Learning Model for Short-Term Traffic Flow Pre-Diction Considering Spatiotemporal Features," Sustainability, MDPI, vol. 14(16), pages 1-14, August.
    19. Zhai, Linbo & Yang, Yong & Song, Shudian & Ma, Shuyue & Zhu, Xiumin & Yang, Feng, 2021. "Self-supervision Spatiotemporal Part-Whole Convolutional Neural Network for Traffic Prediction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 579(C).
    20. Tao, Zihan & Zhang, Chu & Xiong, Jinlin & Hu, Haowen & Ji, Jie & Peng, Tian & Nazir, Muhammad Shahzad, 2023. "Evolutionary gate recurrent unit coupling convolutional neural network and improved manta ray foraging optimization algorithm for performance degradation prediction of PEMFC," Applied Energy, Elsevier, vol. 336(C).

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:eee:phsmap:v:608:y:2022:i:p1:s0378437122008329. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Catherine Liu (email available below). General contact details of provider: http://www.journals.elsevier.com/physica-a-statistical-mechpplications/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.