Skip to main content

Multivariate Traffic Demand Prediction via 2D Spectral Learning and Global Spatial Optimization

  • Conference paper
  • First Online:
Machine Learning and Knowledge Discovery in Databases. Research Track (ECML PKDD 2024)

Part of the book series: Lecture Notes in Computer Science ((LNAI,volume 14942))

  • 698 Accesses

Abstract

Traffic demand forecasting is of critical importance to the development of Intelligent Transportation System (ITS). Existing works employ various deep learning models to capture the complex spatial-temporal correlations in the time-evolving traffic data. However, it has been proven that deep neural networks tend to learn low-frequency variations of inputs, leaving the high-frequency information underexplored. To overcome this spectral bias issue, we design a novel embedded 2D spectral learning framework. Firstly, a well-devised spectral embedding function is employed to encapsulate both the low-frequency and high-frequency signals of the multivariate input. This function also induces a shift-invariant kernel to maintain good distance metrics. Secondly, we model the temporal variations and multivariate feature interactions as two effective dimensions in the frequency domain. The 2D Fourier transform is directly applied along both dimensions followed by Fourier domain representation learning to extract more intrinsic patterns for traffic forecasting. Moreover, existing works employ either local aggregation or pairwise matching mechanism when modeling spatial correlations, which failed to consider the global structure of the traffic network. We propose to formulate spatial feature learning as an optimal transport problem to globally optimize the citywide spatial correlations and promote the overall performance of traffic. Extensive experiments are conducted on two real-world benchmarks with state-of-the-art results, which corroborate the effectiveness of embedded 2D spectral learning and global spatial optimization.

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

Similar content being viewed by others

Notes

  1. 1.

    https://opendata.cityofnewyork.us/.

References

  1. Bai, L., Yao, L., Kanhere, S., Wang, X., Sheng, Q., et al.: Stg2seq: spatial-temporal graph to sequence model for multi-step passenger demand forecasting. arXiv preprint arXiv:1905.10069 (2019)

  2. Bai, L., Yao, L., Li, C., Wang, X., Wang, C.: Adaptive graph convolutional recurrent network for traffic forecasting. In: Advances in Neural Information Processing Systems, vol. 33 (2020)

    Google Scholar 

  3. Basri, R., Galun, M., Geifman, A., Jacobs, D., Kasten, Y., Kritchman, S.: Frequency bias in neural networks for input of non-uniform density. In: International Conference on Machine Learning, pp. 685–694. PMLR (2020)

    Google Scholar 

  4. Cai, L., Janowicz, K., Mai, G., Yan, B., Zhu, R.: Traffic transformer: capturing the continuity and periodicity of time series for traffic forecasting. Trans. GIS 24(3), 736–755 (2020)

    Article  Google Scholar 

  5. Chi, L., Jiang, B., Mu, Y.: Fast Fourier convolution. In: Advances in Neural Information Processing Systems, vol. 33, pp. 4479–4488 (2020)

    Google Scholar 

  6. Distances, C.M.S.: Lightspeed computation of optimal transport. In: Advances in Neural Information Processing Systems, vol. 26, pp. 2292–2300 (2013)

    Google Scholar 

  7. Hochreiter, S., Schmidhuber, J.: Long short-term memory. Neural Comput. 9(8), 1735–1780 (1997)

    Article  Google Scholar 

  8. Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks. In: International Conference on Learning Representations (2017)

    Google Scholar 

  9. Krizhevsky, A., Sutskever, I., Hinton, G.E.: Imagenet classification with deep convolutional neural networks. In: Advances in Neural Information Processing Systems, vol. 25, pp. 1097–1105 (2012)

    Google Scholar 

  10. Lee-Thorp, J., Ainslie, J., Eckstein, I., Ontanon, S.: Fnet: mixing tokens with Fourier transforms. arXiv preprint arXiv:2105.03824 (2021)

  11. Li, Y., Yu, R., Shahabi, C., Liu, Y.: Diffusion convolutional recurrent neural network: data-driven traffic forecasting. In: International Conference on Learning Representations (2018)

    Google Scholar 

  12. Rahaman, N., et al.: On the spectral bias of neural networks. In: International Conference on Machine Learning, pp. 5301–5310. PMLR (2019)

    Google Scholar 

  13. Rahimi, A., Recht, B.: Random features for large-scale kernel machines. In: Advances in Neural Information Processing Systems, vol. 20 (2007)

    Google Scholar 

  14. Rao, Y., Zhao, W., Zhu, Z., Lu, J., Zhou, J.: Global filter networks for image classification. In: Advances in Neural Information Processing Systems, vol. 34, pp. 980–993 (2021)

    Google Scholar 

  15. Song, C., Lin, Y., Guo, S., Wan, H.: Spatial-temporal synchronous graph convolutional networks: a new framework for spatial-temporal network data forecasting. In: AAAI, vol. 34, pp. 914–921 (2020)

    Google Scholar 

  16. Vaswani, A., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems, pp. 5998–6008 (2017)

    Google Scholar 

  17. Villani, C., et al.: Optimal Transport: Old and New, vol. 338. Springer, Heidelberg (2009). https://doi.org/10.1007/978-3-540-71050-9

    Book  Google Scholar 

  18. Wang, B., Lin, Y., Guo, S., Wan, H.: Gsnet: learning spatial-temporal correlations from geographical and semantic aspects for traffic accident risk forecasting. In: AAAI (2021)

    Google Scholar 

  19. Woo, G., Liu, C., Sahoo, D., Kumar, A., Hoi, S.: Cost: contrastive learning of disentangled seasonal-trend representations for time series forecasting. arXiv preprint arXiv:2202.01575 (2022)

  20. Wu, Z., Pan, S., Long, G., Jiang, J., Chang, X., Zhang, C.: Connecting the dots: multivariate time series forecasting with graph neural networks. In: Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 753–763 (2020)

    Google Scholar 

  21. Wu, Z., Pan, S., Long, G., Jiang, J., Zhang, C.: Graph wavenet for deep spatial-temporal graph modeling. In: IJCAI, pp. 1907–1913 (2019)

    Google Scholar 

  22. Xu, M., et al.: Spatial-temporal transformer networks for traffic flow forecasting. arXiv preprint arXiv:2001.02908 (2020)

  23. Ye, J., et al.: Learning the evolutionary and multi-scale graph structure for multivariate time series forecasting. In: Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pp. 2296–2306 (2022)

    Google Scholar 

  24. Ye, J., Sun, L., Du, B., Fu, Y., Tong, X., Xiong, H.: Co-prediction of multiple transportation demands based on deep spatio-temporal neural network. In: Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 305–313 (2019)

    Google Scholar 

  25. Ye, J., Sun, L., Du, B., Fu, Y., Xiong, H.: Coupled layer-wise graph convolution for transportation demand prediction. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, pp. 4617–4625 (2021)

    Google Scholar 

  26. Yu, B., Yin, H., Zhu, Z.: Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting. In: Proceedings of the 27th International Joint Conference on Artificial Intelligence, pp. 3634–3640 (2018)

    Google Scholar 

  27. Zhang, J., Zheng, Y., Qi, D.: Deep spatio-temporal residual networks for citywide crowd flows prediction. In: AAAI (2017)

    Google Scholar 

  28. Zhang, X., Zhao, Z., Tsiligkaridis, T., Zitnik, M.: Self-supervised contrastive pre-training for time series via time-frequency consistency. arXiv preprint arXiv:2206.08496 (2022)

  29. Zheng, C., Fan, X., Wang, C., Qi, J.: Gman: a graph multi-attention network for traffic prediction. In: AAAI, vol. 34, pp. 1234–1241 (2020)

    Google Scholar 

  30. Zhou, T., Ma, Z., Wen, Q., Wang, X., Sun, L., Jin, R.: Fedformer: frequency enhanced decomposed transformer for long-term series forecasting. arXiv preprint arXiv:2201.12740 (2022)

Download references

Acknowledgement

This work is supported by the Australian Research Council under Grant DP210101347 and Australian Research Council under Grant DP240101349. Yanbin Liu is supported by the Google Cloud Research Credits program (GCP19980904), and the ECMS New Staff Research Grant from Auckland University of Technology.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Ling Chen .

Editor information

Editors and Affiliations

Ethics declarations

Ethical Statement

This paper investigates the traffic demand forecasting problem. The used datasets are all publicly available and do not contain any personal information. Besides, we adhere to academic standards and ethical requirements to avoid any behavior that may violate these requirements.

1 Electronic supplementary material

Below is the link to the electronic supplementary material.

Supplementary material 1 (pdf 516 KB)

Rights and permissions

Reprints and permissions

Copyright information

© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Chen, C., Liu, Y., Chen, L., Zhang, C. (2024). Multivariate Traffic Demand Prediction via 2D Spectral Learning and Global Spatial Optimization. In: Bifet, A., Davis, J., Krilavičius, T., Kull, M., Ntoutsi, E., Žliobaitė, I. (eds) Machine Learning and Knowledge Discovery in Databases. Research Track. ECML PKDD 2024. Lecture Notes in Computer Science(), vol 14942. Springer, Cham. https://doi.org/10.1007/978-3-031-70344-7_5

Download citation

  • DOI: https://doi.org/10.1007/978-3-031-70344-7_5

  • Published:

  • Publisher Name: Springer, Cham

  • Print ISBN: 978-3-031-70343-0

  • Online ISBN: 978-3-031-70344-7

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics