Evaluating the Generalization Ability of Spatiotemporal Model in Urban Scenario
Not on the map. Why?
Abstract (arXiv)
Spatiotemporal neural networks have shown great promise in urban scenarios by effectively capturing temporal and spatial correlations. However, urban environments are constantly evolving, and current model evaluations are often limited to traffic scenarios and use data mainly collected only a few weeks after training period to evaluate model performance. The generalization ability of these models remains largely unexplored. To address this, we propose a Spatiotemporal Out-of-Distribution (ST-OOD) benchmark, which comprises six urban scenario: bike-sharing, 311 services, pedestrian counts, traffic speed, traffic flow, ride-hailing demand, and bike-sharing, each with in-distribution (same year) and out-of-distribution (next years) settings. We extensively evaluate state-of-the-art spatiotemporal models and find that their performance degrades significantly in out-of-distribution settings, with most models performing even worse than a simple Multi-Layer Perceptron (MLP). Our findings suggest that current leading methods tend to over-rely on parameters to overfit training data, which may lead to good performance on in-distribution data but often results in poor generalization. We also investigated whether dropout could mitigate the negative effects of overfitting. Our results showed that a slight dropout rate could significantly improve generalization performance on most datasets, with minimal impact on in-distribution performance. However, balancing in-distribution and out-of-distribution performance remains a challenging problem. We hope that the proposed benchmark will encourage further research on this critical issue.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 53 of its 94 references that are in the map. First posted on arXiv 7 Oct 2024.
Builds on
53 of 94 references.
References · 53
- 2017Attention Is All You Needcited by 59,932
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2016WaveNet: A Generative Model for Raw Audiocited by 1,710
- 2020Graph Contrastive Learning with Augmentationscited by 1,203
- 2017Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecastingcited by 1,125
- 2019Strategies for Pre-training Graph Neural Networkscited by 964
- 2019Graph WaveNet for Deep Spatial-Temporal Graph Modelingcited by 820
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
- 2020Adaptive Graph Convolutional Recurrent Network for Traffic Forecastingcited by 525
38 more
- 2022Non-stationary Transformers: Exploring the Stationarity in Time Series Forecastingcited by 385
- 2021Spectral Temporal Graph Neural Network for Multivariate Time-series Forecastingcited by 298
- 2023WeatherBench 2: A benchmark for the next generation of data-driven global weather modelscited by 227
- 2021Spatial-Temporal Graph ODE Networks for Traffic Flow Forecastingcited by 209
- 2020Data Augmentation for Graph Neural Networkscited by 200
- 2020Spatial-Temporal Transformer Networks for Traffic Flow Forecastingcited by 179
- 2021Dynamic Graph Convolutional Recurrent Network for Traffic Prediction: Benchmark and Solutioncited by 156
- 2022Spatial-Temporal Identity: A Simple yet Effective Baseline for Multivariate Time Series Forecastingcited by 149
- 2018Strike (with) a Pose: Neural Networks Are Easily Fooled by Strange Poses of Familiar Objectscited by 143
- 2021Graph Neural Controlled Differential Equations for Traffic Forecastingcited by 133
- 2018Dark Model Adaptation: Semantic Image Segmentation from Daytime to Nighttimecited by 122
- 2022Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecastingcited by 122
- 2021AdaRNN: Adaptive Learning and Forecasting of Time Seriescited by 120
- 2021PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Modelscited by 118
- 2023LargeST: A Benchmark Dataset for Large-Scale Traffic Forecastingcited by 116
- 2023Exploring Progress in Multivariate Time Series Forecasting: Comprehensive Benchmarking and Heterogeneity Analysiscited by 113
- 2022Out-Of-Distribution Generalization on Graphs: A Surveycited by 97
- 2022Wild-Time: A Benchmark of in-the-Wild Distribution Shift over Timecited by 85
- 2024UrbanGPT: Spatio-Temporal Large Language 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
- 2023OneNet: Enhancing Time Series Forecasting Models under Concept Drift by Online Ensemblingcited by 71
- 2021Generalizing Graph Neural Networks on Out-Of-Distribution Graphscited by 69
- 2022MaskTune: Mitigating Spurious Correlations by Forcing to Explorecited by 66
- 2023OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learningcited by 53
- 2023Deciphering Spatio-Temporal Graph Forecasting: A Causal Lens and Treatmentcited by 47
- 2018Bike Flow Prediction with Multi-Graph Convolutional Networkscited by 42
- 2019Predicting Citywide Crowd Flows in Irregular Regions Using Multi-View Graph Convolutional Networkscited by 40
- 2022Spatial-Temporal Sequential Hypergraph Network for Crime Prediction with Dynamic Multiplex Relation Learningcited by 32
- 2021Conditional Local Convolution for Spatio-temporal Meteorological Forecastingcited by 30
- 2022Out-of-Distribution Representation Learning for Time Series Classificationcited by 30
- 2024Time-Series Forecasting for Out-of-Distribution Generalization Using Invariant Learningcited by 26
- 2023DIVERSIFY: A General Framework for Time Series Out-of-distribution Detection and Generalizationcited by 24
- 2022ST-ExpertNet: A Deep Expert Framework for Traffic Predictioncited by 10
- 2023Spatio-temporal neural structural causal models for bike flow predictioncited by 9
- 2022Easy Begun is Half Done: Spatial-Temporal Graph Modeling with ST-Curriculum Dropoutcited by 7
- 2019Graph Message Passing with Cross-location Attentions for Long-term ILI Predictioncited by 6
- 2022COVID-19 is linked to changes in the time-space dimension of human mobilitycited by 5
50 most cited of 53 All 53 (JSON)
Built on by
Citing papers per year: 2024: 2; 2025: 2; 2026: 3.
7 papers.
- 2025Learning with Calibration: Exploring Test-Time Computing of Spatio-Temporal Forecastingcited by 9
- 2025Network-Wide Traffic Volume Estimation from Speed Profiles using a Spatio-Temporal Graph Neural Network with Directed Spatial Attentioncited by 2
- 2026Learning from Yesterday's Error: An Efficient Online Learning Method for Traffic Demand Predictioncited by 1
- 2024Incorporating Long-term Data in Training Short-term Traffic Prediction Modelcited by 0
- 2024Unveiling the Inflexibility of Adaptive Embedding in Traffic Forecastingcited by 0
- 2026OpFlow: Learning Opportunity-Conditioned Choice Potentials for Robust OD Flow Predictioncited by 0
- 2026Capacity-Aware Deep Learning for Generalizable Traffic Volume Estimation Across Links and Citiescited by 0
- 2026Capacity-Aware Deep Learning for Generalizable Traffic Volume Estimation Across Links and Cities27 Jul 2026
- 2026OpFlow: Learning Opportunity-Conditioned Choice Potentials for Robust OD Flow Prediction3 Jul 2026
- 2026Learning from Yesterday's Error: An Efficient Online Learning Method for Traffic Demand Prediction25 Feb 2026
- 2025Network-Wide Traffic Volume Estimation from Speed Profiles using a Spatio-Temporal Graph Neural Network with Directed Spatial Attention15 Dec 2025
- 2025Learning with Calibration: Exploring Test-Time Computing of Spatio-Temporal Forecasting31 May 2025
- 2024Unveiling the Inflexibility of Adaptive Embedding in Traffic Forecasting18 Nov 2024
- 2024Incorporating Long-term Data in Training Short-term Traffic Prediction Model16 Oct 2024