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>.

Self-Supervised Traffic Classification: Flow Embedding and Few-Shot Solutions

Short-Term Forecasting of Traffic Flow and Speed: A Deep Learning Approach

Towards a Profiling View for Unsupervised Traffic Classification by Exploring the Statistic Features and Link Patterns

Learning By Analogy: Reliable Supervision From Transformations For Unsupervised Optical Flow Estimation

Multitask Learning for Network Traffic Classification

Fast and Robust Online Traffic Classification Supporting Unseen Applications

A Novel Self-Supervised Framework Based on Masked Autoencoder for Traffic Classification

TTTFlow: Unsupervised Test-Time Training with Normalizing Flow

Deep Learning for Short-Term Traffic Flow Prediction

Traffic Classification - Towards Accurate Real Time Network Applications

EV-FlowNet: Self-Supervised Optical Flow Estimation for Event-based Cameras

Energy-Guided Data Sampling for Traffic Prediction with Mini Training Datasets

Flow-MAE: Leveraging Masked AutoEncoder for Accurate, Efficient and Robust Malicious Traffic Classification

Spatio-Temporal Self-Supervised Learning for Traffic Flow Prediction

TrafficCAM: A Versatile Dataset for Traffic Flow Segmentation

Flow-based Encrypted Network Traffic Classification with Graph Neural Networks

Network Traffic Classification based on Single Flow Time Series Analysis

Self-attentive deep learning method for online traffic classification and its interpretability

Short-term traffic flow forecasting with spatial-temporal correlation in a hybrid deep learning framework

DeepFlow: Abnormal Traffic Flow Detection Using Siamese Networks

Bidirectional Flow-Based Image Representation Method for Detecting Network Traffic Service Categories