Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Adaptive Graph Pruning with Sudden-Events Evaluation for Traffic Prediction using Online Semi-Decentralized ST-GNNs

About

Spatio-Temporal Graph Neural Networks (ST-GNNs) are well-suited for processing high-frequency data streams from geographically distributed sensors in smart mobility systems. However, their deployment at the edge across distributed compute nodes (cloudlets) createssubstantial communication overhead due to repeated transmission of overlapping node features between neighbouring cloudlets. To address this, we propose an adaptive pruning algorithm that dynamically filters redundant neighbour features while preserving the most informative spatial context for prediction. The algorithm adjusts pruning rates based on recent model performance, allowing each cloudlet to focus on regions experiencing traffic changes without compromising accuracy. Additionally, we introduce the Sudden Event Prediction Accuracy (SEPA), a novel event-focused metric designed to measure responsiveness to traffic slowdowns and recoveries, which are often missed by standard error metrics. We evaluate our approach in an online semi-decentralized setting with traditional FL, server-free FL, and Gossip Learning on two large-scale traffic datasets, PeMS-BAY and PeMSD7-M, across short-, mid-, and long-term prediction horizons. Experiments show that, in contrast to standard metrics, SEPA exposes the true value of spatial connectivity in predicting dynamic and irregular traffic. Our adaptive pruning algorithm maintains prediction accuracy while significantly lowering communication cost in all online semi-decentralized settings, demonstrating that communication can be reduced without compromising responsiveness to critical traffic events.

Ivan Kralj, Lodovico Giaretta, Gordan Je\v{z}i\'c, Ivana Podnar \v{Z}arko, \v{S}ar\=unas Girdzijauskas• 2025

Related benchmarks

TaskDatasetResultRank
Traffic PredictionPeMS-BAY 15 min horizon (test)
MAE1.5
18
Traffic PredictionPeMS-BAY 30 min horizon (test)
MAE2.04
4
Traffic PredictionPeMSD7-M 15 min horizon (test)
MAE2.44
3
Traffic PredictionPeMS-BAY 60 min horizon (test)
MAE2.7
2
Traffic PredictionPeMSD7-M 30 min horizon (test)
MAE3.37
2
Traffic PredictionPeMSD7-M 60 min horizon (test)
MAE4.59
2
Showing 6 of 6 rows

Other info

Follow for update