Graph Pyramid Autoformer for Long- Term Traffic Forecasting

Weiheng Zhong, Tanwi Mallick, Jane MacFarlane, Hadi Meidani, Prasanna Balaprakash

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Accurate traffic forecasting is vital to an intelligent transportation system. Although many deep learning models have achieved state-of-art performance for short-term traffic forecasting of up to 1 hour, long-term traffic forecasting that spans multiple hours remains a major challenge. To that end, we develop Graph Pyramid Autoformer (GPA), an attention-based spatial-temporal graph neural network that uses a novel pyramid autocorrelation attention mechanism. It enables learning from long temporal sequences on graphs and improves long-term traffic forecasting accuracy. We demonstrate the efficacy of the GPA using two benchmark traffic datasets: Los Angeles' METR-LA and the Bay Area's PEMS-BAY. Notably, our model has outperformed a range of existing state-of-the-art methods, delivering up to a 25 % improvement in the accuracy of long-term traffic forecasts. Our code is available at: https://github.com/WeiheneZlExplainable-Graph-Autoformer.

Original languageEnglish
Title of host publicationProceedings - 22nd IEEE International Conference on Machine Learning and Applications, ICMLA 2023
EditorsM. Arif Wani, Mihai Boicu, Moamar Sayed-Mouchaweh, Pedro Henriques Abreu, Joao Gama
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages384-391
Number of pages8
ISBN (Electronic)9798350345346
DOIs
StatePublished - 2023
Event22nd IEEE International Conference on Machine Learning and Applications, ICMLA 2023 - Jacksonville, United States
Duration: Dec 15 2023Dec 17 2023

Publication series

NameProceedings - 22nd IEEE International Conference on Machine Learning and Applications, ICMLA 2023

Conference

Conference22nd IEEE International Conference on Machine Learning and Applications, ICMLA 2023
Country/TerritoryUnited States
CityJacksonville
Period12/15/2312/17/23

Keywords

  • Graph Neural Network
  • Long-term forecasting
  • Pyramid Autocorrelation attention

Fingerprint

Dive into the research topics of 'Graph Pyramid Autoformer for Long- Term Traffic Forecasting'. Together they form a unique fingerprint.

Cite this