STARK: Enhancing Traffic Prediction Through Spatiotemporal Adaptive Refinement With Knowledge Distillation

Traffic flow prediction remains a complex task due to the intricate spatial and temporal correlations in real-world traffic data. Although existing graph neural network (GNN) approaches have shown promise in capturing these relationships, their high computational requirements limit their suitability for real-time deployment. To overcome these limitations, we propose spatiotemporal adaptive refinement with knowledge distillation (STARK), a novel and efficient framework that integrates graph fusion with adaptive knowledge distillation (AKD) in a spatiotemporal graph convolutional network (STGCN). Our method leverages graph fusion to capture both localized and global traffic dynamics, enhancing adaptability across diverse traffic conditions. It further employs two dedicated teacher models that independently emphasize spatial and temporal features, guiding a lightweight student model through a distillation process that dynamically adjusts based on prediction uncertainty. This adaptive learning mechanism enables the student model to prioritize and better learn from more difficult prediction instances. Evaluations on four benchmark traffic datasets [PEMS03, PEMS04, PEMSD7(M), and PEMS08] demonstrate that STARK achieves competitive predictive performance, measured by mean absolute error (MAE) and root mean square error (RMSE), while significantly reducing computational overhead. Our approach thus offers an effective and scalable solution for real-time traffic forecasting.