Offline Multi-Agent Reinforcement Learning with a Physics-Informed World Model for Cooperative Mixed Traffic Control
Organizations: College of Transportation, Tongji University Shanghai, China, 201804
Abstract
This study investigates cooperative control of connected and automated vehicles (CAVs) at partially observable highway bottlenecks in mixed traffic, aiming to mitigate congestion without relying on complete global traffic states or online trial-and-error. We propose a physics-informed world model-based offline multi-agent reinforcement learning framework that reconstructs a physically interpretable global traffic state from local CAV observation-action histories, with coupled macroscopic-microscopic traffic dynamics providing physics-based supervision. A probabilistic ensemble world model learns traffic-state transitions and system rewards, while model disagreement quantifies epistemic uncertainty. Multi-step imagined rollouts with pessimistic rewards and uncertainty-driven truncation are then used for offline policy learning. Experiments in a SUMO-based on-ramp bottleneck using approximately offline transitions show that physics supervision improves state reconstruction and world-model prediction accuracy.
Figures & tables
| Method | Recovered-State MAE | Recovered-Reward MAE | True-State MAE | True-Reward MAE |
| M1: Proposed | 0.01633 | 0.02031 | 0.09543 | 0.02065 |
| M2: WM-Physical | 0.01912 | 0.02377 | 0.41197 | 0.02340 |
| M3: WM-Latent | 0.06158 | 0.03221 | – | – |
| M4: WM-Concat | 0.00704 | 0.03427 | – | – |
| Metric | Without Control | Online MAPPO | M1 | M2 | M3 | M4 | M5 | M6 | M7 | M8 |
| ATT (s) | 145.81 | 107.93 | 116.96 | 122.92 | 127.59 | 132.14 | 120.69 | 127.40 | 129.23 | 135.90 |
| AWT (s) | 0.99 | 0.14 | 0.38 | 0.32 | 0.37 | 0.34 | 0.38 | 0.53 | 0.51 | 0.49 |
| ATL (s) | 85.44 | 47.51 | 54.60 | 65.43 | 68.33 | 76.32 | 57.10 | 70.37 | 72.80 | 81.21 |
Intelligent Transportation Systems
Intelligent Transportation Systems apply sensing, communication, AI, and optimization to make transport networks safer and more efficient.
See all 301 papers