A Benchmarking Evaluation of Graph Neural Networks on Traffic Speed Prediction

Loading...
Thumbnail Image
File version

Accepted Manuscript (AM)

Author(s)
Quach, KND
Yang, C
Vu, VH
Nguyen, TT
Nguyen, QVH
Jo, J
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2022
Size
File type(s)
Location

Brisbane, Australia

License
Abstract

Traffic speed prediction is the task of forecasting the average moving speed on certain roads and highways. Since transportation is an essential part of our daily life, traffic prediction helps reduce wasting time caused by traffic congestion and the negative environmental impacts of idling vehicles during traffic jams. Recently, graph neural network has achieved a certain level of success in forecasting traffic, thanks to the ability to capture both spatio-temporal features of traffic data. Although many publications propose and benchmark different traffic prediction methods, those models’ performance and other profiles are hardly mentioned and evaluated, such as the running time and memories taken. This research aims to evaluate the performance of five mainstream GNN models in traffic prediction: DCRNN, Graph Wavenet, MTGNN, STGCN, and T-GCN. Training time, inference time, and other profiles of mentioned models are also further investigated and reported. During the experiments, we record various behaviors/factors from models, such as the memory allocation. Different models are believed to be better in specific scenarios in traffic forecasting based on their profiles. The finding shows that some prediction methods can only perform well if the training time is significant, such as T-GCN. The performances on single/multiple step forecasting vary with different models, where MTGNN outperforms other models. Additionally, the memory usage and inferencing time of GNN models are different, potentially impacting the selection of models for some limited memory machines and other scenarios.

Journal Title
Conference Title

ADMA 2022: Advanced Data Mining and Applications

Book Title
Edition
Volume

13725

Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)

ARC

Grant identifier(s)

DE200101465

Rights Statement
Rights Statement

© 2022 Springer, Cham. This is the author-manuscript version of this paper. Reproduced in accordance with the copyright policy of the publisher.The original publication is available at www.springerlink.com

Item Access Status
Note
Access the data
Related item(s)
Subject

Neural networks

Data mining and knowledge discovery

Transport planning

Information and computing sciences

Persistent link to this record
Citation

Quach, KND; Yang, C; Vu, VH; Nguyen, TT; Nguyen, QVH; Jo, J, A Benchmarking Evaluation of Graph Neural Networks on Traffic Speed Prediction, ADMA 2022: Advanced Data Mining and Applications, 2022, 13725, pp. 472-488