Abstract:Accurate traffic forecasting is crucial for effective urban planning and transportation management, enabling efficient resource allocation and enhanced travel experiences. However, existing models often face limitations in generalization, struggling with zero-shot prediction on unseen regions and cities, as well as diminished long-term accuracy. This is primarily due to the inherent challenges in handling the spatial and temporal heterogeneity of traffic data, coupled with the significant distribution shift across time and space. In this work, we aim to unlock new possibilities for building versatile, resilient and adaptive spatio-temporal foundation models for traffic prediction. To achieve this goal, we introduce a novel foundation model, named OpenCity, that can effectively capture and normalize the underlying spatio-temporal patterns from diverse data characteristics, facilitating zero-shot generalization across diverse urban environments. OpenCity integrates the Transformer architecture with graph neural networks to model the complex spatio-temporal dependencies in traffic data. By pre-training OpenCity on large-scale, heterogeneous traffic datasets, we enable the model to learn rich, generalizable representations that can be seamlessly applied to a wide range of traffic forecasting scenarios. Experimental results demonstrate that OpenCity exhibits exceptional zero-shot predictive performance. Moreover, OpenCity showcases promising scaling laws, suggesting the potential for developing a truly one-for-all traffic prediction solution that can adapt to new urban contexts with minimal overhead. We made our proposed OpenCity model open-source and it is available at the following link: <a class="link-external link-https" href="https://github.com/HKUDS/OpenCity" rel="external noopener nofollow">this https URL</a>.

CityTraffic: Modeling Citywide Traffic via Neural Memorization and Generalization Approach

GPS-based Citywide Traffic Congestion Forecasting Using CNN-RNN and C3D Hybrid Model

Joint Modeling of Dense and Incomplete Trajectories for Citywide Traffic Volume Inference

A Memory-augmented Conditional Neural Process model for traffic prediction

TrafficPro: A Road Speed Prediction Framework for Urban Signal-Controlled Road Networks

The Deep 3D Convolutional Multi-Branching Spatial-Temporal-Based Unit Predicting Citywide Traffic Flow

City-Wide Traffic Volume Inference With Loop Detector Data And Taxi Trajectories

Citywide Traffic Volume Estimation Using Trajectory Data.

One Size Fits All: A Unified Traffic Predictor for Capturing the Essential Spatial–Temporal Dependency

Traffic Modeling Based on Data-Driven Method for Simulation Test of Autonomous Driving

A Knowledge-Driven Memory System for Traffic Flow Prediction.

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

Turning traffic volume imputation for persistent missing patterns with GNNs

A Neural Network Model for Urban Traffic Volumes Compression

Advancing Traffic Simulation Precision and Scalability: A Data-Driven Approach Utilizing Deep Neural Networks

High-resolution large-scale urban traffic speed estimation with multi-source crowd sensing data

City Traffic Forecasting Using Taxi GPS Data: A Coarse-Grained Cellular Automata Model.

A Multiscale-Grid-Based Stacked Bidirectional GRU Neural Network Model for Predicting Traffic Speeds of Urban Expressways

Real-Time Traffic Speed Estimation for Smart Cities with Spatial Temporal Data: A Gated Graph Attention Network Approach

OpenCity: Open Spatio-Temporal Foundation Models for Traffic Prediction

Modeling multi-regional temporal correlation with gated recurrent unit and multiple linear regression for urban traffic flow prediction