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. 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).
    2. 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).
    3. 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.
    4. 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.
    5. 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.
    6. 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.
    7. 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).
    8. 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.
    9. 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).
    10. Owais, Mahmoud & Moussa, Ghada S. & Hussain, Khaled F., 2019. "Sensor location model for O/D estimation: Multi-criteria meta-heuristics approach," Operations Research Perspectives, Elsevier, vol. 6(C).
    11. Xia, Dong & Zheng, Linjiang & Tang, Yi & Cai, Xiaolin & Chen, Li & Sun, Dihua, 2022. "Dynamic traffic prediction for urban road network with the interpretable model," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 605(C).
    12. Lu, Wenqi & Yi, Ziwei & Wu, Renfei & Rui, Yikang & Ran, Bin, 2022. "Traffic speed forecasting for urban roads: A deep ensemble neural network model," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 593(C).
    13. Cantelmo, Guido & Qurashi, Moeid & Prakash, A. Arun & Antoniou, Constantinos & Viti, Francesco, 2020. "Incorporating trip chaining within online demand estimation," Transportation Research Part B: Methodological, Elsevier, vol. 132(C), pages 171-187.
    14. Lederman, Roger & Wynter, Laura, 2011. "Real-time traffic estimation using data expansion," Transportation Research Part B: Methodological, Elsevier, vol. 45(7), pages 1062-1079, August.
    15. Zhou, Xuesong & Mahmassani, Hani S., 2007. "A structural state space model for real-time traffic origin-destination demand estimation and prediction in a day-to-day learning framework," Transportation Research Part B: Methodological, Elsevier, vol. 41(8), pages 823-840, October.
    16. Yan Zheng & Chunjiao Dong & Daiyue Dong & Shengyou Wang, 2021. "Traffic Volume Prediction: A Fusion Deep Learning Model Considering Spatial–Temporal Correlation," Sustainability, MDPI, vol. 13(19), pages 1-18, September.
    17. Yan, Jie & Nuertayi, Akejiang & Yan, Yamin & Liu, Shan & Liu, Yongqian, 2023. "Hybrid physical and data driven modeling for dynamic operation characteristic simulation of wind turbine," Renewable Energy, Elsevier, vol. 215(C).
    18. Wang, Wei & Zhang, Hanyu & Li, Tong & Guo, Jianhua & Huang, Wei & Wei, Yun & Cao, Jinde, 2020. "An interpretable model for short term traffic flow prediction," Mathematics and Computers in Simulation (MATCOM), Elsevier, vol. 171(C), pages 264-278.
    19. Min Li & Mengshan Li & Bilong Liu & Jiang Liu & Zhen Liu & Dijia Luo, 2022. "Spatio-Temporal Traffic Flow Prediction Based on Coordinated Attention," Sustainability, MDPI, vol. 14(12), pages 1-17, June.
    20. Alireza Ermagun & David Levinson, 2017. "Spatiotemporal Short-term Traffic Forecasting using the Network Weight Matrix and Systematic Detrending," Working Papers 000166, University of Minnesota: Nexus Research Group.

    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.