Deciphering Spatio-Temporal Graph Forecasting: A Causal Lens and Treatment
Not on the map. Why?
Abstract (arXiv)
Spatio-Temporal Graph (STG) forecasting is a fundamental task in many real-world applications. Spatio-Temporal Graph Neural Networks have emerged as the most popular method for STG forecasting, but they often struggle with temporal out-of-distribution (OoD) issues and dynamic spatial causation. In this paper, we propose a novel framework called CaST to tackle these two challenges via causal treatments. Concretely, leveraging a causal lens, we first build a structural causal model to decipher the data generation process of STGs. To handle the temporal OoD issue, we employ the back-door adjustment by a novel disentanglement block to separate invariant parts and temporal environments from input data. Moreover, we utilize the front-door adjustment and adopt the Hodge-Laplacian operator for edge-level convolution to model the ripple effect of causation. Experiments results on three real-world datasets demonstrate the effectiveness and practicality of CaST, which consistently outperforms existing methods with good interpretability.
Key facts. Cited by 47 papers in the map (data through 13 Oct 2026). Builds on 36 of its 68 references that are in the map. First posted on arXiv 23 Sep 2023.
Builds on
36 of 68 references.
References · 36
- 2014Adam: A Method for Stochastic Optimizationcited by 37,093
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,318
- 2017Graph Attention Networkscited by 8,888
- 2017Neural Discrete Representation Learningcited by 4,680
- 2018An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modelingcited by 1,460
- 2017Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecastingcited by 1,130
- 2019Graph WaveNet for Deep Spatial-Temporal Graph Modelingcited by 825
- 2018MINE: Mutual Information Neural Estimationcited by 682
- 2020Connecting the Dots: Multivariate Time Series Forecasting with Graph Neural Networkscited by 564
- 2013Generating Sequences With Recurrent Neural Networkscited by 553
- 2020Adaptive Graph Convolutional Recurrent Network for Traffic Forecastingcited by 527
- 2019GMAN: A Graph Multi-Attention Network for Traffic Predictioncited by 434
24 more
- 2016Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Predictioncited by 333
- 2020Causal Intervention for Weakly-Supervised Semantic Segmentationcited by 229
- 2021Spatial-Temporal Graph ODE Networks for Traffic Flow Forecastingcited by 210
- 2019Deep Learning for Spatio-Temporal Data Mining: A Surveycited by 172
- 2022CoST: Contrastive Learning of Disentangled Seasonal-Trend Representations for Time Series Forecastingcited by 144
- 2023Spatio-Temporal Graph Neural Networks for Predictive Learning in Urban Computing: A Surveycited by 140
- 2021Graph Neural Controlled Differential Equations for Traffic Forecastingcited by 133
- 2018Revisiting Spatial-Temporal Similarity: A Deep Learning Framework for Traffic Predictioncited by 120
- 2021AdaRNN: Adaptive Learning and Forecasting of Time Seriescited by 120
- 2021Causal Attention for Interpretable and Generalizable Graph Classificationcited by 108
- 2020Visual Commonsense R-CNNcited by 106
- 2018Random Walks on Simplicial Complexes and the normalized Hodge 1-Laplaciancited by 84
- 2023DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Modelscited by 81
- 2021OOD-GNN: Out-of-Distribution Generalized Graph Neural Networkcited by 78
- 2021DL-Traff: Survey and Benchmark of Deep Learning Models for Urban Traffic Predictioncited by 73
- 2022AirFormer: Predicting Nationwide Air Quality in China with Transformerscited by 73
- 2022STDEN: Towards Physics-Guided Neural Networks for Traffic Flow Predictioncited by 45
- 2022OOD Link Prediction Generalization Capabilities of Message-Passing GNNs in Larger Test Graphscited by 38
- 2018Flow Smoothing and Denoising: Graph Signal Processing in the Edge-Spacecited by 37
- 2021Edge Representation Learning with Hypergraphscited by 37
- 2023Do We Really Need Graph Neural Networks for Traffic Forecasting?cited by 20
- 2022Towards Out-of-Distribution Sequential Event Prediction: A Causal Treatmentcited by 10
- 2021Helmholtzian Eigenmap: Topological feature discovery & edge flow learning from point cloud datacited by 7
- 2023Heterogeneous Graph Convolutional Neural Network via Hodge-Laplacian for Brain Functional Datacited by 6
Built on by
Citing papers per year: 2023: 2; 2024: 22; 2025: 16; 2026: 7.
47 papers.
Sort:
By area:
- 2024Foundation Models for Time Series Analysis: A Tutorial and Surveycited by 237
- 2023Large Models for Time Series and Spatio-Temporal Data: A Survey and Outlookcited by 109
- 2024UniST: A Prompt-Empowered Universal Model for Urban Spatio-Temporal Predictioncited by 86
- 2024Deep Learning for Cross-Domain Data Fusion in Urban Computing: Taxonomy, Advances, and Outlookcited by 33
- 2024MolTC: Towards Molecular Relational Modeling In Language Modelscited by 22
- 2024Causal Evaluation of Language Modelscited by 18
- 2024Heterogeneity-Informed Meta-Parameter Learning for Spatiotemporal Time Series Forecastingcited by 18
- 2024Spatio-Temporal Few-Shot Learning via Diffusive Neural Network Generationcited by 15
- 2025Unified Spatial-Temporal Edge-Enhanced Graph Networks for Pedestrian Trajectory Predictioncited by 13
- 2025Learning with Calibration: Exploring Test-Time Computing of Spatio-Temporal Forecastingcited by 9
- 2023Exploring Causal Learning through Graph Neural Networks: An In-depth Reviewcited by 8
- 2024CaT-GNN: Enhancing Credit Card Fraud Detection via Causal Temporal Graph Neural Networkscited by 7
35 more
- 2024Evaluating the Generalization Ability of Spatiotemporal Model in Urban Scenariocited by 7
- 2025Causal Time Series Generation via Diffusion Modelscited by 7
- 2024Spatio-Temporal Fluid Dynamics Modeling via Physical-Awareness and Parameter Diffusion Guidancecited by 5
- 2025Invariant Link Selector for Spatial-Temporal Out-of-Distribution Problemcited by 5
- 2024Towards Robust Trajectory Representations: Isolating Environmental Confounders with Causal Learningcited by 4
- 2024DualCast: A Model to Disentangle Aperiodic Events from Traffic Seriescited by 4
- 2025FineCausal: A Causal-Based Framework for Interpretable Fine-Grained Action Quality Assessmentcited by 4
- 2025Causal Spatio-Temporal Prediction: An Effective and Efficient Multi-Modal Approachcited by 4
- 2025Reimagining Urban Science: Scaling Causal Inference with Large Language Modelscited by 3
- 2025Spatiotemporal Causal Decoupling Model for Air Quality Forecastingcited by 3
- 2025AquaSentinel: Next-Generation AI System Integrating Sensor Networks for Urban Underground Water Pipeline Anomaly Detection via Collaborative MoE-LLM Agent Architecturecited by 3
- 2026Policy4OOD: A Knowledge-Guided World Model for Policy Intervention Simulation against the Opioid Overdose Crisiscited by 3
- 2025AirRadar: Inferring Nationwide Air Quality in China with Deep Neural Networkscited by 2
- 2024Predicting Parking Availability in Singapore with Cross-Domain Data: A New Dataset and A Data-Driven Approachcited by 1
- 2024Robust Traffic Forecasting against Spatial Shift over Yearscited by 1
- 2024Causal Adjacency Learning for Spatiotemporal Prediction Over Graphscited by 1
- 2024Spatio-temporal Causal Learning for Streamflow Forecastingcited by 1
- 2024STDCformer: A Transformer-Based Model with a Spatial-Temporal Causal De-Confounding Strategy for Crowd Flow Predictioncited by 1
- 2024Memory-enhanced Invariant Prompt Learning for Urban Flow Prediction under Distribution Shiftscited by 1
- 2025SynEVO: A neuro-inspired spatiotemporal evolutional framework for cross-domain adaptationcited by 1
- 2025CaPulse: Detecting Anomalies by Tuning in to the Causal Rhythms of Time Seriescited by 1
- 2025AI-Driven Forecasting and Monitoring of Urban Water Systemcited by 1
- 2026A General ReLearner: Empowering Spatiotemporal Prediction by Re-learning Input-label Residualcited by 1
- 2026A Generalizable Physics-guided Causal Model for Trajectory Prediction in Autonomous Drivingcited by 1
- 2024Towards Causal Relationship in Indefinite Data: Baseline Model and New Datasetscited by 0
- 2024ComS2T: A complementary spatiotemporal learning system for data-adaptive model evolutioncited by 0
- 2024Incorporating Long-term Data in Training Short-term Traffic Prediction Modelcited by 0
- 2024Unveiling the Inflexibility of Adaptive Embedding in Traffic Forecastingcited by 0
- 2025Soft causal learning for generalized molecule property prediction: An environment perspectivecited by 0
- 2025Solar Forecasting with Causality: A Graph-Transformer Approach to Spatiotemporal Dependenciescited by 0
- 2025Towards Resilient Transportation: A Conditional Transformer for Accident-Informed Traffic Forecastingcited by 0
- 2026Efficient Prompt Learning for Traffic Forecastingcited by 0
- 2026Every Client Is an Environment: Federated De-confounding for Spatio-Temporal Forecastingcited by 0
- 2026When Correlations Mislead: Confounder-Aware Multi-View Urban Region Representation Learningcited by 0
- 2026AdaST: Adaptive Coupling for Spatial-Temporal Forecastingcited by 0
- 2026AdaST: Adaptive Coupling for Spatial-Temporal Forecasting28 Sep 2026
- 2026When Correlations Mislead: Confounder-Aware Multi-View Urban Region Representation Learning14 Sep 2026
- 2026Every Client Is an Environment: Federated De-confounding for Spatio-Temporal Forecasting27 Jul 2026
- 2026Efficient Prompt Learning for Traffic Forecasting8 May 2026
- 2026A Generalizable Physics-guided Causal Model for Trajectory Prediction in Autonomous Driving15 Feb 2026
- 2026Policy4OOD: A Knowledge-Guided World Model for Policy Intervention Simulation against the Opioid Overdose Crisis12 Feb 2026
- 2026A General ReLearner: Empowering Spatiotemporal Prediction by Re-learning Input-label Residual30 Jan 2026
- 2025Towards Resilient Transportation: A Conditional Transformer for Accident-Informed Traffic Forecasting10 Dec 2025
- 2025AquaSentinel: Next-Generation AI System Integrating Sensor Networks for Urban Underground Water Pipeline Anomaly Detection via Collaborative MoE-LLM Agent Architecture19 Nov 2025
- 2025AI-Driven Forecasting and Monitoring of Urban Water System8 Oct 2025
- 2025Causal Time Series Generation via Diffusion Models25 Sep 2025
- 2025Solar Forecasting with Causality: A Graph-Transformer Approach to Spatiotemporal Dependencies18 Sep 2025
35 more
- 2025CaPulse: Detecting Anomalies by Tuning in to the Causal Rhythms of Time Series6 Aug 2025
- 2025Learning with Calibration: Exploring Test-Time Computing of Spatio-Temporal Forecasting31 May 2025
- 2025Invariant Link Selector for Spatial-Temporal Out-of-Distribution Problem30 May 2025
- 2025Spatiotemporal Causal Decoupling Model for Air Quality Forecasting26 May 2025
- 2025Causal Spatio-Temporal Prediction: An Effective and Efficient Multi-Modal Approach23 May 2025
- 2025SynEVO: A neuro-inspired spatiotemporal evolutional framework for cross-domain adaptation21 May 2025
- 2025Soft causal learning for generalized molecule property prediction: An environment perspective7 May 2025
- 2025Reimagining Urban Science: Scaling Causal Inference with Large Language Models15 Apr 2025
- 2025FineCausal: A Causal-Based Framework for Interpretable Fine-Grained Action Quality Assessment31 Mar 2025
- 2025Unified Spatial-Temporal Edge-Enhanced Graph Networks for Pedestrian Trajectory Prediction4 Feb 2025
- 2025AirRadar: Inferring Nationwide Air Quality in China with Deep Neural Networks22 Jan 2025
- 2024Memory-enhanced Invariant Prompt Learning for Urban Flow Prediction under Distribution Shifts7 Dec 2024
- 2024STDCformer: A Transformer-Based Model with a Spatial-Temporal Causal De-Confounding Strategy for Crowd Flow Prediction4 Dec 2024
- 2024DualCast: A Model to Disentangle Aperiodic Events from Traffic Series27 Nov 2024
- 2024Spatio-temporal Causal Learning for Streamflow Forecasting26 Nov 2024
- 2024Causal Adjacency Learning for Spatiotemporal Prediction Over Graphs25 Nov 2024
- 2024Unveiling the Inflexibility of Adaptive Embedding in Traffic Forecasting18 Nov 2024
- 2024Incorporating Long-term Data in Training Short-term Traffic Prediction Model16 Oct 2024
- 2024Evaluating the Generalization Ability of Spatiotemporal Model in Urban Scenario7 Oct 2024
- 2024Robust Traffic Forecasting against Spatial Shift over Years1 Oct 2024
- 2024Predicting Parking Availability in Singapore with Cross-Domain Data: A New Dataset and A Data-Driven Approach29 May 2024
- 2024Heterogeneity-Informed Meta-Parameter Learning for Spatiotemporal Time Series Forecasting17 May 2024
- 2024Causal Evaluation of Language Models1 May 2024
- 2024Towards Robust Trajectory Representations: Isolating Environmental Confounders with Causal Learning22 Apr 2024
- 2024Foundation Models for Time Series Analysis: A Tutorial and Survey21 Mar 2024
- 2024Spatio-Temporal Fluid Dynamics Modeling via Physical-Awareness and Parameter Diffusion Guidance18 Mar 2024
- 2024ComS2T: A complementary spatiotemporal learning system for data-adaptive model evolution4 Mar 2024
- 2024Deep Learning for Cross-Domain Data Fusion in Urban Computing: Taxonomy, Advances, and Outlook29 Feb 2024
- 2024CaT-GNN: Enhancing Credit Card Fraud Detection via Causal Temporal Graph Neural Networks22 Feb 2024
- 2024Spatio-Temporal Few-Shot Learning via Diffusive Neural Network Generation19 Feb 2024
- 2024UniST: A Prompt-Empowered Universal Model for Urban Spatio-Temporal Prediction19 Feb 2024
- 2024MolTC: Towards Molecular Relational Modeling In Language Models6 Feb 2024
- 2024Towards Causal Relationship in Indefinite Data: Baseline Model and New Datasets16 Jan 2024
- 2023Exploring Causal Learning through Graph Neural Networks: An In-depth Review25 Nov 2023
- 2023Large Models for Time Series and Spatio-Temporal Data: A Survey and Outlook16 Oct 2023
- 2024Causal Evaluation of Language Modelscited by 18
- 2024Causal Evaluation of Language Models1 May 2024
- 2024Foundation Models for Time Series Analysis: A Tutorial and Surveycited by 237
- 2023Large Models for Time Series and Spatio-Temporal Data: A Survey and Outlookcited by 109
- 2024UniST: A Prompt-Empowered Universal Model for Urban Spatio-Temporal Predictioncited by 86
- 2024Deep Learning for Cross-Domain Data Fusion in Urban Computing: Taxonomy, Advances, and Outlookcited by 33
- 2024MolTC: Towards Molecular Relational Modeling In Language Modelscited by 22
- 2024Heterogeneity-Informed Meta-Parameter Learning for Spatiotemporal Time Series Forecastingcited by 18
- 2025Unified Spatial-Temporal Edge-Enhanced Graph Networks for Pedestrian Trajectory Predictioncited by 13
- 2025Learning with Calibration: Exploring Test-Time Computing of Spatio-Temporal Forecastingcited by 9
- 2023Exploring Causal Learning through Graph Neural Networks: An In-depth Reviewcited by 8
- 2024CaT-GNN: Enhancing Credit Card Fraud Detection via Causal Temporal Graph Neural Networkscited by 7
- 2024Evaluating the Generalization Ability of Spatiotemporal Model in Urban Scenariocited by 7
- 2024Spatio-Temporal Fluid Dynamics Modeling via Physical-Awareness and Parameter Diffusion Guidancecited by 5
12 most cited of 44 All 44 (JSON)
- 2026AdaST: Adaptive Coupling for Spatial-Temporal Forecasting28 Sep 2026
- 2026When Correlations Mislead: Confounder-Aware Multi-View Urban Region Representation Learning14 Sep 2026
- 2026Every Client Is an Environment: Federated De-confounding for Spatio-Temporal Forecasting27 Jul 2026
- 2026Efficient Prompt Learning for Traffic Forecasting8 May 2026
- 2026A Generalizable Physics-guided Causal Model for Trajectory Prediction in Autonomous Driving15 Feb 2026
- 2026Policy4OOD: A Knowledge-Guided World Model for Policy Intervention Simulation against the Opioid Overdose Crisis12 Feb 2026
- 2026A General ReLearner: Empowering Spatiotemporal Prediction by Re-learning Input-label Residual30 Jan 2026
- 2025Towards Resilient Transportation: A Conditional Transformer for Accident-Informed Traffic Forecasting10 Dec 2025
- 2025AquaSentinel: Next-Generation AI System Integrating Sensor Networks for Urban Underground Water Pipeline Anomaly Detection via Collaborative MoE-LLM Agent Architecture19 Nov 2025
- 2025AI-Driven Forecasting and Monitoring of Urban Water System8 Oct 2025
- 2025Solar Forecasting with Causality: A Graph-Transformer Approach to Spatiotemporal Dependencies18 Sep 2025
- 2025CaPulse: Detecting Anomalies by Tuning in to the Causal Rhythms of Time Series6 Aug 2025
12 newest of 44 All 44 (JSON)