Abstract:Robust prediction of citywide traffic flows at different time periods plays a crucial role in intelligent transportation systems. While previous work has made great efforts to model spatio-temporal correlations, existing methods still suffer from two key limitations: i) Most models collectively predict all regions' flows without accounting for spatial heterogeneity, i.e., different regions may have skewed traffic flow distributions. ii) These models fail to capture the temporal heterogeneity induced by time-varying traffic patterns, as they typically model temporal correlations with a shared parameterized space for all time periods. To tackle these challenges, we propose a novel Spatio-Temporal Self-Supervised Learning (ST-SSL) traffic prediction framework which enhances the traffic pattern representations to be reflective of both spatial and temporal heterogeneity, with auxiliary self-supervised learning paradigms. Specifically, our ST-SSL is built over an integrated module with temporal and spatial convolutions for encoding the information across space and time. To achieve the adaptive spatio-temporal self-supervised learning, our ST-SSL first performs the adaptive augmentation over the traffic flow graph data at both attribute- and structure-levels. On top of the augmented traffic graph, two SSL auxiliary tasks are constructed to supplement the main traffic prediction task with spatial and temporal heterogeneity-aware augmentation. Experiments on four benchmark datasets demonstrate that ST-SSL consistently outperforms various state-of-the-art baselines. Since spatio-temporal heterogeneity widely exists in practical datasets, the proposed framework may also cast light on other spatial-temporal applications. Model implementation is available at <a class="link-external link-https" href="https://github.com/Echo-Ji/ST-SSL" rel="external noopener nofollow">this https URL</a>.

Data-unbalanced traffic accident prediction via adaptive graph and self-supervised learning

Learning Multiaspect Traffic Couplings by Multirelational Graph Attention Networks for Traffic Prediction

Self-Supervised State Space Model for Real-Time Traffic Accident Prediction Using eKAN Networks

Advancing urban traffic accident forecasting through sparse spatio-temporal dynamic learning

A Self-Adaptive Spatio-Temporal Graph Convolutional Network for Traffic Prediction

A Deep Learning Approach to the Prediction of Short-term Traffic Accident Risk

Uncertainty-Aware Probabilistic Graph Neural Networks for Road-Level Traffic Accident Prediction

Spatio-Temporal Self-Supervised Learning for Traffic Flow Prediction

Adaptive spatial-temporal graph attention networks for traffic flow forecasting

Urban Traffic Accident Risk Prediction Revisited: Regionality, Proximity, Similarity and Sparsity

Traffic Prediction with Self-Supervised Learning: A Heterogeneity-Aware Model for Urban Traffic Flow Prediction Based on Self-Supervised Learning

GraphSANet: A Graph Neural Network and Self Attention Based Approach for Spatial Temporal Prediction in Sensor Network

SMA-Hyper: Spatiotemporal Multi-View Fusion Hypergraph Learning for Traffic Accident Prediction

Latent Gaussian Processes based Graph Learning for Urban Traffic Prediction

Dynamic multi-scale spatial-temporal graph convolutional network for traffic flow prediction

Domain adversarial graph neural network with cross-city graph structure learning for traffic prediction

Gated Fusion Adaptive Graph Neural Network for Urban Road Traffic Flow Prediction

An analytic framework using deep learning for prediction of traffic accident injury severity based on contributing factors

Dynamic multiple-graph spatial-temporal synchronous aggregation framework for traffic prediction in intelligent transportation systems

Optimized Graph Convolution Recurrent Neural Network for Traffic Prediction

Self-Supervised Spatiotemporal Graph Neural Networks With Self-Distillation for Traffic Prediction