Virtual Nodes Improve Long-term Traffic Prediction

TL;DR

This paper introduces a novel graph neural network framework with virtual nodes that significantly improves long-term traffic prediction accuracy by enhancing information flow and interpretability in urban traffic systems.

Contribution

The study proposes a semi-adaptive adjacency matrix with virtual nodes to address over-squashing and improve long-term traffic forecasting in spatio-temporal graph neural networks.

Findings

Virtual nodes improve long-term prediction accuracy.

Enhanced layer-wise sensitivity reduces over-squashing.

Visualization shows focus on key traffic areas.

Abstract

Effective traffic prediction is a cornerstone of intelligent transportation systems, enabling precise forecasts of traffic flow, speed, and congestion. While traditional spatio-temporal graph neural networks (ST-GNNs) have achieved notable success in short-term traffic forecasting, their performance in long-term predictions remains limited. This challenge arises from over-squashing problem, where bottlenecks and limited receptive fields restrict information flow and hinder the modeling of global dependencies. To address these challenges, this study introduces a novel framework that incorporates virtual nodes, which are additional nodes added to the graph and connected to existing nodes, in order to aggregate information across the entire graph within a single GNN layer. Our proposed model incorporates virtual nodes by constructing a semi-adaptive adjacency matrix. This matrix integrates distance-based and adaptive adjacency matrices, allowing the model to leverage geographical information while also learning task-specific features from data. Experimental results demonstrate that the inclusion of virtual nodes significantly enhances long-term prediction accuracy while also improving layer-wise sensitivity to mitigate the over-squashing problem. Virtual nodes also offer enhanced explainability by focusing on key intersections and high-traffic areas, as shown by the visualization of their adjacency matrix weights on road network heat maps. Our advanced approach enhances the understanding and management of urban traffic systems, making it particularly well-suited for real-world applications.