mlsanityData to 14 Oct 2026

A Pretrained Probabilistic Transformer for City-Scale Traffic Volume Prediction

arXiv 2506.02654 ↗First posted 3 Jun 2025cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

City-scale traffic volume prediction plays a pivotal role in intelligent transportation systems, yet remains a challenge due to the inherent incompleteness and bias in observational data. Although deep learning-based methods have shown considerable promise, most existing approaches produce deterministic point estimates, thereby neglecting the uncertainty arising from unobserved traffic flows. Furthermore, current models are typically trained in a city-specific manner, which hinders their generalizability and limits scalability across diverse urban contexts. To overcome these limitations, we introduce TrafficPPT, a Pretrained Probabilistic Transformer designed to model traffic volume as a distributional aggregation of trajectories. Our framework fuses heterogeneous data sources-including real-time observations, historical trajectory data, and road network topology-enabling robust and uncertainty-aware traffic inference. TrafficPPT is initially pretrained on large-scale simulated data spanning multiple urban scenarios, and later fine-tuned on target cities to ensure effective domain adaptation. Experiments on real-world datasets show that TrafficPPT consistently surpasses state-of-the-art baselines, particularly under conditions of extreme data sparsity. Code will be open.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 40 references that are in the map. First posted on arXiv 3 Jun 2025.

Builds on

15 of 40 references.

References · 15

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2023GPT-4 Technical Reportcited by 18,396
  3. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  4. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
  5. 2017Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecastingcited by 1,136
  6. 2019Strategies for Pre-training Graph Neural Networkscited by 971
  7. 2018T-GCN: A Temporal Graph ConvolutionalNetwork for Traffic Predictioncited by 401
  8. 2016Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Predictioncited by 333
  9. 2020GPT-GNN: Generative Pre-Training of Graph Neural Networkscited by 303
  10. 2016OSMnx: New Methods for Acquiring, Constructing, Analyzing, and Visualizing Complex Street Networkscited by 110
  11. 2021View Vertically: A Hierarchical Network for Trajectory Prediction via Fourier Spectrumscited by 45
  12. 2020Spatio-Temporal Hybrid Graph Convolutional Network for Traffic Forecasting in Telecommunication Networkscited by 7
3 more
  1. 2022CBLab: Supporting the Training of Large-scale Traffic Control Policies with Scalable Traffic Simulationcited by 6
  2. 2023Explainable Trajectory Representation through Dictionary Learningcited by 6
  3. 2020Learning Probabilistic Intersection Traffic Models for Trajectory Predictioncited by 1

Built on by

Not cited yet.