Structure-Guided Spatiotemporal Attention Graph Neural Network for Traffic Flow Prediction
Not on the map. Why?
Abstract (arXiv)
Deep spatiotemporal models integrating graph convolutions and attention mechanisms have demonstrated excellent performance in network-level traffic flow prediction, owing to their exceptional ability to capture complex spatiotemporal dependencies. Despite their predictive success, deployment of such models in safety-critical urban systems remains constrained by their inherent lack of transparency. Existing post-hoc diagnostic methods often struggle with spurious correlations and fail to unveil the intrinsic decision-making mechanisms governing traffic dynamics, resulting in suboptimal interpretability and limited operational trustworthiness. To address these challenges, this paper proposes the Structure-Guided Spatiotemporal Attention Graph Neural Network (SGSAN). Departing from traditional architectures that rely on unconstrained adaptive graphs, SGSAN explicitly learns a static Directed Dependency Graph (DDG) to identify the invariant macroscopic propagation paths of traffic states. We further introduce an InfoNCE-based soft-coupling mechanism that anchors the model's dynamic spatiotemporal attention to this structural prior, offering a mechanistic account of the model's decision-making process while ensuring robust forecasting by aligning attention-based reasoning with identified macroscopic dependencies and preventing over-reliance on ephemeral local noise. Furthermore, a decoupled two-stage optimization framework is developed to resolve the fundamental conflict between structural discovery and predictive error minimization. Extensive experiments on multiple real-world datasets demonstrate that SGSAN achieves state-of-the-art predictive accuracy while providing built-in interpretability that organically aligns with the physical logic of traffic networks.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 49 references that are in the map. First posted on arXiv 14 Aug 2026.
Builds on
25 of 49 references.
References · 25
- 2017Graph Attention Networkscited by 8,923
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
- 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,070
- 2019Explainable Artificial Intelligence (XAI): Concepts, Taxonomies, Opportunities and Challenges toward Responsible AIcited by 1,552
- 2017Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecastingcited by 1,136
- 2018DAGs with NO TEARS: Continuous Optimization for Structure Learningcited by 752
- 2019Attention is not not Explanationcited by 636
- 2019GMAN: A Graph Multi-Attention Network for Traffic Predictioncited by 434
- 2018T-GCN: A Temporal Graph ConvolutionalNetwork for Traffic Predictioncited by 401
- 2020Graph Structure Learning for Robust Graph Neural Networkscited by 371
- 2019DAG-GNN: DAG Structure Learning with Graph Neural Networkscited by 331
13 more
- 2019Interpretable machine learning: definitions, methods, and applicationscited by 305
- 2020A Survey on Neural Network Interpretabilitycited by 291
- 2021Graph Neural Network for Traffic Forecasting: A Surveycited by 261
- 2019Learning Discrete Structures for Graph Neural Networkscited by 231
- 2018Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecastingcited by 146
- 2020A3T-GCN: Attention Temporal Graph Convolutional Network for Traffic Forecastingcited by 54
- 2021Traffic Flow Forecasting with Spatial-Temporal Graph Diffusion Networkcited by 52
- 2023Automated Spatio-Temporal Graph Contrastive Learningcited by 24
- 2019Causal Regularizationcited by 21
- 2023Causal conditional hidden Markov model for multimodal traffic predictioncited by 14
- 2022STGC-GNNs: A GNN-based traffic prediction framework with a spatial-temporal Granger causality graphcited by 12
- 2025Dynamic Trend Fusion Module for Traffic Flow Predictioncited by 5
- 2024Make Graph Neural Networks Great Again: A Generic Integration Paradigm of Topology-Free Patterns for Traffic Speed Predictioncited by 4
Built on by
Not cited yet.