Abstract
Traffic forecasting has remained a challenging topic in the field of transportation, due to the time-varying traffic patterns and complicated spatial dependencies on road networks. To address such challenges, we propose an adaptive graph co-attention network (AGCAN) to predict traffic conditions on a road network graph. In our model, an adaptive graph modelling method is adopted to learn a dynamic relational graph in which the links can capture the dynamic spatial correlations of traffic patterns among nodes, even though the adjacent nodes may not be physically connected. Besides, we propose a novel co-attention network targeting long- and short-term traffic patterns. The long-term graph attention module is used to derive periodic patterns from historical data, while the short-term graph attention module is employed to respond to sudden traffic changes, like car accidents and special events. To minimize the loss generated during the learning process, we adopt an encoder-decoder architecture, where both the encoder and decoder consist of novel hierarchical spatio-temporal attention blocks to model the impact of potential factors on traffic conditions. Overall, the experimental results on two real-world traffic prediction tasks demonstrate the superiority of AGCAN.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
An, S.H., Lee, B.H., Shin, D.R.: A survey of intelligent transportation systems. In: 2011 Third International Conference on Computational Intelligence, Communication Systems and Networks, pp. 332–337. IEEE (2011)
Bolshinsky, E., Friedman, R.: Traffic flow forecast survey. Technical report, Computer Science Department, Technion (2012)
Cascetta, E.: Transportation Systems Engineering: Theory and Methods, vol. 49. Springer, Cham (2013)
Ermagun, A., Levinson, D.: Spatiotemporal traffic forecasting: review and proposed directions. Transp. Rev. 38(6), 786–814 (2018)
Grover, A., Leskovec, J.: node2vec: Scalable feature learning for networks. In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 855–864 (2016)
Jaderberg, M., Simonyan, K., Zisserman, A., et al.: Spatial transformer networks. In: Advances in Neural Information Processing Systems, pp. 2017–2025 (2015)
Lana, I., Del Ser, J., Velez, M., Vlahogianni, E.I.: Road traffic forecasting: recent advances and new challenges. IEEE Intell. Transp. Syst. Mag. 10(2), 93–109 (2018)
Li, Y., Yu, R., Shahabi, C., Liu, Y.: Diffusion convolutional recurrent neural network: data-driven traffic forecasting. arXiv preprint arXiv:1707.01926 (2017)
Lu, H., Huang, D., Song, Y., Jiang, D., Zhou, T., Qin, J.: St-trafficnet: a spatial-temporal deep learning network for traffic forecasting. Electronics 9(9), 1474 (2020)
Lv, Z., Xu, J., Zheng, K., Yin, H., Zhao, P., Zhou, X.: LC-RNN: a deep learning model for traffic speed prediction. In: IJCAI, pp. 3470–3476 (2018)
Ma, X., Dai, Z., He, Z., Ma, J., Wang, Y., Wang, Y.: Learning traffic as images: a deep convolutional neural network for large-scale transportation network speed prediction. Sensors 17(4), 818 (2017)
Maas, A., Le, Q.V., O’neil, T.M., Vinyals, O., Nguyen, P., Ng, A.Y.: Recurrent neural networks for noise reduction in robust ASR (2012)
Makridakis, S., Hibon, M.: ARMA models and the box-Jenkins methodology. J. Forecast. 16(3), 147–163 (1997)
Oord, A.V.D., et al.: Wavenet: A generative model for raw audio. arXiv preprint arXiv:1609.03499 (2016)
Park, C., Lee, C., Bahng, H., Kim, K., Jin, S., Ko, S., Choo, J., et al.: Stgrat: A spatio-temporal graph attention network for traffic forecasting. arXiv preprint arXiv:1911.13181 (2019)
Qi, L.: Research on intelligent transportation system technologies and applications. In: 2008 Workshop on Power Electronics and Intelligent Transportation System, pp. 529–531. IEEE (2008)
Shi, X., Chen, Z., Wang, H., Yeung, D.Y., Wong, W.K., Woo, W.C.: Convolutional LSTM network: a machine learning approach for precipitation nowcasting. In: Advances in Neural Information Processing Systems, vol. 28, pp. 802–810 (2015)
Shuman, D.I., Narang, S.K., Frossard, P., Ortega, A., Vandergheynst, P.: The emerging field of signal processing on graphs: Extending high-dimensional data analysis to networks and other irregular domains. IEEE Signal Process. Mag. 30(3), 83–98 (2013)
Sutskever, I., Vinyals, O., Le, Q.V.: Sequence to sequence learning with neural networks. In: Advances in Neural Information Processing Systems, pp. 3104–3112 (2014)
Thekumparampil, K.K., Wang, C., Oh, S., Li, L.J.: Attention-based graph neural network for semi-supervised learning. arXiv preprint arXiv:1803.03735 (2018)
Vaswani, A., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems, pp. 5998–6008 (2017)
Wei, L., et al.: Dual graph for traffic forecasting. IEEE Access (2019)
Wu, Y., Tan, H.: Short-term traffic flow forecasting with spatial-temporal correlation in a hybrid deep learning framework. arXiv preprint arXiv:1612.01022 (2016)
Wu, Z., Pan, S., Long, G., Jiang, J., Zhang, C.: Graph wavenet for deep spatial-temporal graph modeling. arXiv preprint arXiv:1906.00121 (2019)
Yu, B., Yin, H., Zhu, Z.: Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting. arXiv preprint arXiv:1709.04875 (2017)
Zambrano-Martinez, J.L., Calafate, C.T., Soler, D., Cano, J.C., Manzoni, P.: Modeling and characterization of traffic flows in urban environments. Sensors 18(7), 2020 (2018)
Zheng, C., Fan, X., Wang, C., Qi, J.: GMAN: a graph multi-attention network for traffic prediction. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, pp. 1234–1241 (2020)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 Springer Nature Switzerland AG
About this paper
Cite this paper
Li, B., Guo, T., Wang, Y., Gandomi, A.H., Chen, F. (2021). Adaptive Graph Co-Attention Networks for Traffic Forecasting. In: Karlapalem, K., et al. Advances in Knowledge Discovery and Data Mining. PAKDD 2021. Lecture Notes in Computer Science(), vol 12712. Springer, Cham. https://doi.org/10.1007/978-3-030-75762-5_22
Download citation
DOI: https://doi.org/10.1007/978-3-030-75762-5_22
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-75761-8
Online ISBN: 978-3-030-75762-5
eBook Packages: Computer ScienceComputer Science (R0)