Variational Mode-Driven Graph Convolutional Network for Spatiotemporal Traffic Forecasting

TL;DR

This paper introduces VMGCN, a hybrid graph neural network that decomposes spatiotemporal traffic data into interpretable modes using variational mode decomposition, enhancing prediction accuracy and interpretability.

Contribution

It proposes a novel two-stage framework combining variational mode decomposition with an attention-augmented GCN for improved traffic forecasting.

Findings

Significant accuracy improvements over existing methods.

Enhanced interpretability through frequency-level mode analysis.

Effective multi-horizon traffic flow prediction demonstrated.

Abstract

This paper focuses on spatiotemporal (ST) traffic prediction using graph neural networks (GNNs). Given that ST data comprises non-stationary and complex temporal patterns, interpreting and predicting such trends is inherently challenging. Representing ST data in decomposed modes helps infer underlying behavior and assess the impact of noise on predictive performance. We propose a framework that decomposes ST data into interpretable modes using variational mode decomposition (VMD) and processes them through a neural network for future state forecasting. Unlike existing graph-based traffic forecasters that operate directly on raw or aggregated time series, the proposed hybrid approach, termed the Variational Mode Graph Convolutional Network (VMGCN), first decomposes non-stationary signals into interpretable variational modes by determining the optimal mode count via reconstruction-loss minimization and then learns both intramode and cross-mode spatiotemporal dependencies through a novel attention-augmented GCN. Additionally, we analyze the significance of each mode and the effect of bandwidth constraints on multi-horizon traffic flow predictions. The proposed two-stage design yields significant accuracy gains while providing frequency-level interpretability with demonstrated superior performance on the LargeST dataset for both short-term and long-term forecasting tasks. The implementation is publicly available on https://github.com/OsamaAhmad369/VMGCN.