Ultra-fast Traffic Nowcasting and Control via Differentiable Agent-based Simulation
Not on the map. Why?
Abstract (arXiv)
Traffic digital twins, which inform policymakers of effective interventions based on large-scale, high-fidelity computational models calibrated to real-world traffic, hold promise for addressing societal challenges in our rapidly urbanizing world. However, conventional fine-grained traffic simulations are non-differentiable and typically rely on inefficient gradient-free optimization, making calibration for real-world applications computationally infeasible. Here we present a differentiable agent-based traffic simulator that enables ultra-fast model calibration, traffic nowcasting, and control on large-scale networks. We develop several differentiable computing techniques for simulating individual vehicle movements, including stochastic decision-making and inter-agent interactions, while ensuring that entire simulation trajectories remain end-to-end differentiable for efficient gradient-based optimization. On the large-scale Chicago road network, with over 10,000 calibration parameters, our model simulates more than one million vehicles at 173 times real-time speed. This ultra-fast simulation, together with efficient gradient-based optimization, enables us to complete model calibration using the previous 30 minutes of traffic data in 455 s, provide a one-hour-ahead traffic nowcast in 21 s, and solve the resulting traffic control problem in 728 s. This yields a full calibration--nowcast--control loop in under 20 minutes, leaving about 40 minutes of lead time for implementing interventions. Our work thus provides a practical computational basis for realizing traffic digital twins.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 16 of its 51 references that are in the map. First posted on arXiv 26 Mar 2026.
Builds on
16 of 51 references.
References · 16
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
- 2016The Concrete Distribution: A Continuous Relaxation of Discrete Random Variablescited by 1,135
- 2015Automatic differentiation in machine learning: a surveycited by 846
- 2019DiffTaichi: Differentiable Programming for Physical Simulationcited by 248
- 2016A Differentiable Physics Engine for Deep Learning in Roboticscited by 107
- 2022JAX-FLUIDS: A fully-differentiable high-order computational fluid dynamics solver for compressible two-phase flowscited by 36
- 2024A Review of Differentiable Simulatorscited by 26
- 2022Differentiable Agent-based Epidemiologycited by 21
- 2019Estimating multi-class dynamic origin-destination demand through a forward-backward algorithm on computational graphscited by 10
- 2021Differentiable Agent-Based Simulation for Gradient-Guided Simulation-Based Optimizationcited by 8
- 2022Differentiable Hybrid Traffic Simulationcited by 8
4 more
- 2025Automatic Differentiation of Agent-Based Modelscited by 4
- 2023UXsim: An open source macroscopic and mesoscopic traffic simulator in Python -- a technical overviewcited by 2
- 2024Gradient-based Trajectory Optimization with Parallelized Differentiable Traffic Simulationcited by 2
- 2025Flow-Through Tensors: A Unified Computational Graph Architecture for Multi-Layer Transportation Network Optimizationcited by 2
Built on by
Not cited yet.