Spatio-Temporal Parallel Transformer Based Model for Traffic Prediction

• Tipo: Artigo em Revista Científica Internacional
• Data: 2024

Autores

Kumar, R (Autor) Outra; João Mendes-Moreira (Autor) FEUP; Chandra, J (Autor) Outra

Revista

Título: ACM Transactions on Knowledge Discovery from Data

Vol. 18

ISSN: 1556-4681

Editora: ACM

Indexação

ISI Web of Knowledge - 8 Citações

Scopus - 7 Citações

Outras Informações

ID Authenticus: P-011-34W

DOI: 10.1145/3679017

Abstract (EN): Traffic forecasting problems involve jointly modeling the non-linear spatio-temporal dependencies at different scales. While graph neural network models have been effectively used to capture the non-linear spatial dependencies, capturing the dynamic spatial dependencies between the locations remains a major challenge. The errors in capturing such dependencies propagate in modeling the temporal dependencies between the locations, thereby severely affecting the performance of long-term predictions. While transformer-based mechanisms have been recently proposed for capturing the dynamic spatial dependencies, these methods are susceptible to fluctuations in data brought on by unforeseen events like traffic congestion and accidents. To mitigate these issues we propose an improvised spatio-temporal parallel transformer (STPT) based model for traffic prediction that uses multiple adjacency graphs passed through a pair of coupled graph transformer- convolution network units, operating in parallel, to generate more noise-resilient embeddings. We conduct extensive experiments on 4 real-world traffic datasets and compare the performance of STPT with several state-of-the-art baselines, in terms of measures like RMSE, MAE, and MAPE. We find that using STPT improves the performance by around 10 - 34% as compared to the baselines. We also investigate the applicability of the model on other spatio-temporal data in other domains. We use a Covid-19 dataset to predict the number of future occurrences in different regions from a given set of historical occurrences. The results demonstrate the superiority of our model for such datasets.

Idioma: Inglês

Tipo (Avaliação Docente): Científica

Nº de páginas: 25