VQGG: Generating Adaptive Graphs for Traffic Forecasting Via a Vector-Quantized Graph Generator

Songyu Ke,Chenyu Wu,Jinjin Guo,Junbo Zhang,Yu Zheng
DOI: https://doi.org/10.1109/ijcnn60899.2024.10649950
2024-01-01
Abstract:Traffic forecasting is crucial in the realm of intelligent urban planning, playing an essential role in route optimization, arrival time estimation, and the prevention of congestion-related incidents. Deep learning has catalyzed the development of advanced spatio-temporal graph neural networks (STGNNs) for traffic predictions. A key advantage of STGNNs is their capability to capture complex spatial correlations, thereby improving prediction accuracy. Nevertheless, the existing methodologies, including dynamical graph-based networks and attention-based architectures, tend to overlook regular traffic patterns, leading to overly complex neural networks with less interpretability and more computational costs, which could be a bottleneck for practical applications.To address the challenge, we propose a lightweight and user-friendly dynamic adaptive graph generator, termed the Vector-Quantized Graph Generator (VQGG), which can autonomously identify prevalent spatial patterns and construct the corresponding adaptive graphs efficiently for enhanced traffic forecasting. We have extensively tested the VQGG with two benchmark datasets by integrating it as a substitute for the adaptive graphs in baseline models with minor modifications. Extensive experimental results demonstrate that VQGG can reduce forecasting errors by approximately 2.59%, indicating a significant improvement in predictive performance.
What problem does this paper attempt to address?