Domain Adversarial Spatial-Temporal Network: A Transferable Framework for Short-term Traffic Forecasting across Cities

by   Yihong Tang, et al.
Beijing University of Posts and Telecommunications
City University of Hong Kong
Vanderbilt University
The University of Hong Kong
The Hong Kong Polytechnic University

Accurate real-time traffic forecast is critical for intelligent transportation systems (ITS) and it serves as the cornerstone of various smart mobility applications. Though this research area is dominated by deep learning, recent studies indicate that the accuracy improvement by developing new model structures is becoming marginal. Instead, we envision that the improvement can be achieved by transferring the "forecasting-related knowledge" across cities with different data distributions and network topologies. To this end, this paper aims to propose a novel transferable traffic forecasting framework: Domain Adversarial Spatial-Temporal Network (DASTNet). DASTNet is pre-trained on multiple source networks and fine-tuned with the target network's traffic data. Specifically, we leverage the graph representation learning and adversarial domain adaptation techniques to learn the domain-invariant node embeddings, which are further incorporated to model the temporal traffic data. To the best of our knowledge, we are the first to employ adversarial multi-domain adaptation for network-wide traffic forecasting problems. DASTNet consistently outperforms all state-of-the-art baseline methods on three benchmark datasets. The trained DASTNet is applied to Hong Kong's new traffic detectors, and accurate traffic predictions can be delivered immediately (within one day) when the detector is available. Overall, this study suggests an alternative to enhance the traffic forecasting methods and provides practical implications for cities lacking historical traffic data.


page 1

page 8


Explainable Graph Pyramid Autoformer for Long-Term Traffic Forecasting

Accurate traffic forecasting is vital to an intelligent transportation s...

Towards Good Practices of U-Net for Traffic Forecasting

This technical report presents a solution for the 2020 Traffic4Cast Chal...

Few-Shot Traffic Prediction with Graph Networks using Locale as Relational Inductive Biases

Accurate short-term traffic prediction plays a pivotal role in various s...

PSTN: Periodic Spatial-temporal Deep Neural Network for Traffic Condition Prediction

Accurate forecasting of traffic conditions is critical for improving saf...

Graph Markov Network for Traffic Forecasting with Missing Data

Traffic forecasting is a classical task for traffic management and it pl...

Traffic Accident Risk Forecasting using Contextual Vision Transformers

Recently, the problem of traffic accident risk forecasting has been gett...

ADL-ID: Adversarial Disentanglement Learning for Wireless Device Fingerprinting Temporal Domain Adaptation

As the journey of 5G standardization is coming to an end, academia and i...

Please sign up or login with your details

Forgot password? Click here to reset