Abstract
Predicting traffic flow is crucial for transportation management and resource allocation, which has attracted more and more attention from researchers. The traffic flow in a city generally changes over time periods but always exhibits certain periodicity. Previous works focused on modeling spatial and temporal correlations using convolutional and recurrent neural networks respectively. Typically, a method that can effectively absorb more time-interval inputs and integrate more periodic information will achieve better performance. In this paper, we propose a Frequency-aware Spatio-temporal Network (FASTNet) for traffic flow prediction. In addition to modeling the spatio-temporal correlations, we dynamically filter the inputs to explicitly incorporate frequency information for traffic prediction. By applying Discrete Fourier Transform (DFT) on traffic flow, we obtain the spectrum of traffic flow sequence which reflects certain travel patterns of passengers. We then adopt a frequency-based filtering mechanism to filter the traffic flow series based on the explored spectrum information. To utilize the filtered tensor, a 3D convolutional network is designed to extract the spatio-temporal features automatically. Inspired by the frequency spectrum of traffic flows, this spatio-temporal convolutional network has various kernels with different sizes on temporal dimension, which models the temporal correlations with multi-scale frequencies. The final prediction layer summarizes the spatio-temporal features extracted by the spatio-temporal convolutional network. Our model outperforms the state-of-the-art methods through extensive experiments on three real datasets for citywide traffic flow prediction.
Keywords
This is a preview of subscription content, log in via an institution.
Buying options
Tax calculation will be finalised at checkout
Purchases are for personal use only
Learn about institutional subscriptionsReferences
Ahn, J., Ko, E., Kim, E.Y.: Predicting spatiotemporal traffic flow based on support vector regression and Bayesian classifier. In: Fifth IEEE International Conference on Big Data and Cloud Computing, BDCloud 2015, Dalian, China, 26–28 August 2015, pp. 125–130 (2015). https://doi.org/10.1109/BDCloud.2015.64
Bahdanau, D., Cho, K., Bengio, Y.: Neural machine translation by jointly learning to align and translate. arXiv preprint arXiv:1409.0473 (2014)
Chung, J., Gulcehre, C., Cho, K., Bengio, Y.: Empirical evaluation of gated recurrent neural networks on sequence modeling. arXiv preprint arXiv:1412.3555 (2014)
Deng, D., Shahabi, C., Demiryurek, U., Zhu, L., Yu, R., Liu, Y.: Latent space model for road networks to predict time-varying traffic. In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 1525–1534. ACM (2016)
El-Yaniv, R., Faynburd, A.: Autoregressive short-term prediction of turning points using support vector regression. CoRR abs/1209.0127 http://arxiv.org/abs/1209.0127 (2012)
Girija, S.S.: Tensorflow: large-scale machine learning on heterogeneous distributed systems (2016)
Girshick, R.: Fast R-CNN. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 1440–1448 (2015)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Computer Vision and Pattern Recognition, pp. 770–778 (2016)
Hochreiter, S., Schmidhuber, J.: Long short-term memory. Neural Comput. 9(8), 1735–1780 (1997)
Kay, S.M., Marple, S.L.: Spectrum analysis—a modern perspective. Proc. IEEE 69(11), 1380–1419 (1981)
Krizhevsky, A., Sutskever, I., Hinton, G.E.: Imagenet classification with deep convolutional neural networks, pp. 1097–1105 (2012)
Li, S., Li, W., Cook, C., Zhu, C., Gao, Y.: Independently recurrent neural network (IndRNN): building a longer and deeper RNN. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 5457–5466 (2018)
Lu, Z., Zhou, C., Wu, J., Jiang, H., Cui, S.: Integrating granger causality and vector auto-regression for traffic prediction of large-scale wlans. TIIS 10(1), 136–151 (2016). https://doi.org/10.3837/tiis.2016.01.008
Shekhar, S., Williams, B.: Adaptive seasonal time series models for forecasting short-term traffic flow. Transp. Res. Rec. J. Transp. Res. Board 2024, 116–125 (2008)
Sun, H., Liu, H., Xiao, H., He, R., Ran, B.: Use of local linear regression model for short-term traffic forecasting. Transp. Res. Rec. J. Transp. Res. Board 1836, 143–150 (2003)
Szegedy, C., Ioffe, S., Vanhoucke, V., Alemi, A.A.: Inception-v4, inception-resnet and the impact of residual connections on learning. In: AAAI, vol. 4, p. 12 (2017)
Tieleman, T., Hinton, G.: RMSprop gradient optimization (2014). http://www.cs.toronto.edu/tijmen/csc321/slides/lecture_slides_lec6.pdf
Wu, Y., Chen, F., Lu, C., Yang, S.: Urban traffic flow prediction using a spatio-temporal random effects model. J. Intellig. Transp. Syst. 20(3), 282–293 (2016). https://doi.org/10.1080/15472450.2015.1072050
Xingjian, S., 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, pp. 802–810 (2015)
Xu, J., Deng, D., Demiryurek, U., Shahabi, C., Van Der Schaar, M.: Context-aware online spatiotemporal traffic prediction. In: 2014 IEEE International Conference on Data Mining Workshop (ICDMW), pp. 43–46. IEEE (2014)
Yao, H., et al.: Deep multi-view spatial-temporal network for taxi demand prediction. arXiv preprint arXiv:1802.08714 (2018)
Zhang, J., Zheng, Y., Qi, D.: Deep spatio-temporal residual networks for citywide crowd flows prediction. In: AAAI, pp. 1655–1661 (2017)
Zhang, J., Zheng, Y., Qi, D., Li, R., Yi, X.: DNN-based prediction model for spatio-temporal data. In: Proceedings of the 24th ACM SIGSPATIAL International Conference on Advances in Geographic Information Systems, p. 92. ACM (2016)
Zhou, X., Shen, Y., Zhu, Y., Huang, L.: Predicting multi-step citywide passenger demands using attention-based neural networks, pp. 736–744 (2018)
Acknowledgements
This work is supported by the National Key Research and Development Program of China (No. 2018YFC0831604). Yanyan Shen is in part supported by NSFC (No. 61602297). Yanmin Zhu is in part supported by NSFC (No. 61772341, 61472254) and STSCM (No. 18511103002). Yuting Chen is in part supported by NSFC (No. 61572312) and Shanghai Municipal Commission of Economy and Informatization (No. 201701052).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Switzerland AG
About this paper
Cite this paper
Peng, S., Shen, Y., Zhu, Y., Chen, Y. (2019). A Frequency-Aware Spatio-Temporal Network for Traffic Flow Prediction. In: Li, G., Yang, J., Gama, J., Natwichai, J., Tong, Y. (eds) Database Systems for Advanced Applications. DASFAA 2019. Lecture Notes in Computer Science(), vol 11447. Springer, Cham. https://doi.org/10.1007/978-3-030-18579-4_41
Download citation
DOI: https://doi.org/10.1007/978-3-030-18579-4_41
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-18578-7
Online ISBN: 978-3-030-18579-4
eBook Packages: Computer ScienceComputer Science (R0)