[go: up one dir, main page]
More Web Proxy on the site http://driver.im/ skip to main content
10.1145/3603781.3603907acmotherconferencesArticle/Chapter ViewAbstractPublication PagescniotConference Proceedingsconference-collections
research-article

Transportation Flow Prediction Based on Graph Attention Echo State Network

Published: 27 July 2023 Publication History

Abstract

Abstract—Traffic flow prediction is of great importance in applications such as traffic management and urban planning. The complex spatial and temporal dependence of traffic flow between different roads poses a great challenge for accurate real-time traffic flow prediction. Traditional traffic flow prediction methods rely on the assumption of data smoothness, and the prediction accuracy decreases significantly in the face of complex, variable and large amount of traffic flow data. Spatio-temporal prediction models based on graph neural networks and recurrent neural networks can achieve better prediction accuracy, but there are still some problems, such as the need for a known static graph structure, inadequate spatial extraction and long training time of the model. To improve traffic flow prediction accuracy and real-time performance, this paper proposes a novel end-to-end deep learning framework called graph attention echo state network (GAESN), which uses attention mechanism and echo state network to extract spatio-temporal features. Experimental results on four real traffic flow datasets show that our proposed model achieves 17.35, 21.34, 24.12 and 17.31 in mean absolute error(MAE); 29.31, 32.67, 37.51and 26.84 in root mean square error(RMSE); 16.76%, 15.44%, 10.33% and 10.94% in mean absolute percentage error(MAPE), respectively. Compared with other existing models, this model reduces the number of parameters to be trained and the time required for model training, and also improves the accuracy of traffic flow prediction.

References

[1]
Augustin Soule, Kavé Salamatian, Antonio Nucci, and Nina Taft. 2005. Traffic matrix tracking using Kalman filters. SIGMETRICS Perform. Eval. Rev. 33, 3 (December 2005), 24–31. https://doi.org/10.1145/1111572.1111580
[2]
Vinay B. Gavirangaswamy, Gagan Gupta, Ajay Gupta, and Rajeev Agrawal. 2013. Assessment of ARIMA-based prediction techniques for road-traffic volume. In Proceedings of the Fifth International Conference on Management of Emergent Digital EcoSystems (MEDES '13). Association for Computing Machinery, New York, NY, USA, 246–251. https://doi.org/10.1145/2536146.2536176
[3]
Zikai Zou, Peter Gao, and Chang Yao. 2018. City-Level Traffic Flow Prediction via LSTM Networks. In Proceedings of the 2nd International Conference on Advances in Image Processing (ICAIP '18). Association for Computing Machinery, New York, NY, USA, 149–153. https://doi.org/10.1145/3239576.3239623
[4]
Jingyan Guo, Zijun Wang, and Huawei Chen. 2017. On-line Multi-step Prediction of Short Term Traffic Flow Based on GRU Neural Network. In Proceedings of the 2nd International Conference on Intelligent Information Processing (ICIIP '17). Association for Computing Machinery, New York, NY, USA, Article 11, 1–6. https://doi.org/10.1145/3144789.3144804
[5]
Yisheng Lv, Yanjie Duan, Wenwen Kang, Zhengxi Li, and Fei-Yue Wang. 2015. Traffic Flow Prediction With Big Data: A Deep Learning Approach. Trans. Intell. Transport. Syst. 16, 2 (April 2015), 865–873. https://doi.org/10.1109/TITS.2014.2345663
[6]
Junbo Zhang, Yu Zheng, and Dekang Qi. 2017. Deep spatio-temporal residual networks for citywide crowd flows prediction. In Proceedings of the Thirty-First AAAI Conference on Artificial Intelligence (AAAI'17). AAAI Press, 1655–1661.
[7]
Xiaolei Ma, Zhuang Dai, Zhengbing He, Jihui Na, Yong Wang, and Yunpeng Wang. 2017. Learning Traffic as Images: A Deep Convolutional Neural Network for Large-Scale Transportation Network Speed Prediction. arXiv:1701.04245. Retrieved from https://arxiv.org/abs/1701.04245.
[8]
Yisheng Lv, Yanjie Duan, Wenwen Kang, Zhengxi Li, and Fei-Yue Wang. 2015. Traffic Flow Prediction With Big Data: A Deep Learning Approach. Trans. Intell. Transport. Syst. 16, 2 (April 2015), 865–873. https://doi.org/10.1109/TITS.2014.2345663
[9]
Junbo Zhang, Yu Zheng, and Dekang Qi. 2017. Deep spatio-temporal residual networks for citywide crowd flows prediction. In Proceedings of the Thirty-First AAAI Conference on Artificial Intelligence (AAAI'17). AAAI Press, 1655–1661.
[10]
Xiaolei Ma, Zhuang Dai, Zhengbing He, Jihui Na, Yong Wang, and Yunpeng Wang. 2017. Learning Traffic as Images: A Deep Convolutional Neural Network for Large-Scale Transportation Network Speed Prediction. arXiv:1701.04245. Retrieved from https://arxiv.org/abs/1701.04245
[11]
Thomas N. Kipf and Max Welling. 2016. Semi-Supervised Classification with Graph Convolutional Networks. arXiv:1609. 02907. Retrieved from https://arxiv.org/abs/1609.02907
[12]
Xu Chen, Yuanxing Zhang, Lun Du, Zheng Fang, Yi Ren, Kaigui Bian and Kunqing Xie. 2020. TSSRGCN: Temporal Spectral Spatial Retrieval Graph Convolutional Network for Traffic Flow Forecasting. arXiv: 2011.14638. Retrieved from https://arxiv.org/abs/ 2011.14638
[13]
Zhao Ling, Song Yujiao, Zhang Chao, Liu Yu, Wang Pu, Lin Tao, Deng Min and Li Haifeng. 2020. T-GCN: A Temporal Graph Convolutional Network for Traffic Prediction. IEEE Transactions on Intelligent Transportation Systems. 21. 9 (Sept. 2020), 3848-3858. https://doi.org/10.1109/TITS.2019.2935152
[14]
Yaguang Li, Rose Yu, Cyrus Shahabi, and Yan Liu. 2017. Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting. arXiv:1707.01926. Retrieved from https://arxiv.org/abs/1707.01926
[15]
Zonghan Wu, Shirui Pan, Guodong Long, Jing Jiang, Xiaojun Chang, and Chengqi Zhang. 2020. Connecting the Dots: Multivariate Time Series Forecasting with Graph Neural Networks. In Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining (KDD '20). Association for Computing Machinery, New York, NY, USA, 753–763. https://doi.org/10.1145/3394486.3403118
[16]
Chao Song, Youfang Lin, Shengnan Guo and Huaiyu Wan. 2020. Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data Forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence. AAAI Press, Palo Alto, California USA, 4189-4196.

Index Terms

  1. Transportation Flow Prediction Based on Graph Attention Echo State Network
        Index terms have been assigned to the content through auto-classification.

        Recommendations

        Comments

        Please enable JavaScript to view thecomments powered by Disqus.

        Information & Contributors

        Information

        Published In

        cover image ACM Other conferences
        CNIOT '23: Proceedings of the 2023 4th International Conference on Computing, Networks and Internet of Things
        May 2023
        1025 pages
        ISBN:9798400700705
        DOI:10.1145/3603781
        Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

        Publisher

        Association for Computing Machinery

        New York, NY, United States

        Publication History

        Published: 27 July 2023

        Permissions

        Request permissions for this article.

        Check for updates

        Author Tags

        1. Echo state network
        2. Graph attention network
        3. Multi-headed attention mechanism
        4. Spatio-temporal dependence
        5. Traffic prediction

        Qualifiers

        • Research-article
        • Research
        • Refereed limited

        Conference

        CNIOT'23

        Acceptance Rates

        Overall Acceptance Rate 39 of 82 submissions, 48%

        Contributors

        Other Metrics

        Bibliometrics & Citations

        Bibliometrics

        Article Metrics

        • 0
          Total Citations
        • 32
          Total Downloads
        • Downloads (Last 12 months)24
        • Downloads (Last 6 weeks)1
        Reflects downloads up to 12 Dec 2024

        Other Metrics

        Citations

        View Options

        Login options

        View options

        PDF

        View or Download as a PDF file.

        PDF

        eReader

        View online with eReader.

        eReader

        HTML Format

        View this article in HTML Format.

        HTML Format

        Media

        Figures

        Other

        Tables

        Share

        Share

        Share this Publication link

        Share on social media