mlsanityData to 7 Oct 2026

Safe Urban Traffic Control via Uncertainty-Aware Conformal Prediction and World-Model Reinforcement Learning

arXiv 2602.04821 ↗First posted 4 Feb 2026cs.LG cs.AIcited by 0

On the map: World models → World models for autonomous driving

Abstract (arXiv)

Urban traffic management demands systems that simultaneously predict future conditions, detect anomalies, and take safe corrective actions -- all while providing reliability guarantees. We present STREAM-RL, a unified framework that introduces three novel algorithmic contributions: (1) PU-GAT+, an Uncertainty-Guided Adaptive Conformal Forecaster that uses prediction uncertainty to dynamically reweight graph attention via confidence-monotonic attention, achieving distribution-free coverage guarantees; (2) CRFN-BY, a Conformal Residual Flow Network that models uncertainty-normalized residuals via normalizing flows with Benjamini-Yekutieli FDR control under arbitrary dependence; and (3) LyCon-WRL+, an Uncertainty-Guided Safe World-Model RL agent with Lyapunov stability certificates, certified Lipschitz bounds, and uncertainty-propagated imagination rollouts. To our knowledge, this is the first framework to propagate calibrated uncertainty from forecasting through anomaly detection to safe policy learning with end-to-end theoretical guarantees. Experiments on multiple real-world traffic trajectory data demonstrate that STREAM-RL achieves 91.4% coverage efficiency, controls FDR at 4.1% under verified dependence, and improves safety rate to 95.2% compared to 69% for standard PPO while achieving higher reward, with 23ms end-to-end inference latency.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in World models for autonomous driving, one of 11 directions in World models. Builds on 35 of its 87 references that are in the map. First posted on arXiv 4 Feb 2026.

Builds on

35 of 87 references.

References · 35

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2016Density estimation using Real NVPcited by 2,029
  3. 2018Spectral Normalization for Generative Adversarial Networkscited by 1,618
  4. 2019GMAN: A Graph Multi-Attention Network for Traffic Predictioncited by 433
  5. 2022Conformal prediction beyond exchangeabilitycited by 348
  6. 2017Smart "Predict, then Optimize"cited by 332
  7. 2018Regularisation of Neural Networks by Enforcing Lipschitz Continuitycited by 311
  8. 2021Discrete Graph Structure Learning for Forecasting Multiple Time Seriescited by 169
  9. 2022Adaptive Conformal Predictions for Time Seriescited by 155
  10. 2017A General Safety Framework for Learning-Based Control in Uncertain Robotic Systemscited by 150
  11. 2019Lyapunov-based Safe Policy Optimization for Continuous Controlcited by 140
  12. 2016Panning for Gold: Model-X Knockoffs for High-dimensional Controlled Variable Selectioncited by 120
23 more
  1. 2020Conservative Safety Critics for Explorationcited by 107
  2. 2019Safe Testingcited by 98
  3. 2020Uncertainty Aware Semi-Supervised Learning on Graph Datacited by 95
  4. 2021Conformal Prediction using Conditional Histogramscited by 77
  5. 2021Z-GCNETs: Time Zigzags at Graph Convolutional Networks for Time Series Forecastingcited by 67
  6. 2022Towards Spatio-Temporal Aware Traffic Time Series Forecasting--Full Versioncited by 59
  7. 2022Constrained Policy Optimization via Bayesian World Modelscited by 55
  8. 2019Worst Cases Policy Gradientscited by 52
  9. 2021How Powerful is Graph Convolution for Recommendation?cited by 50
  10. 2021Improving Conditional Coverage via Orthogonal Quantile Regressioncited by 48
  11. 2017Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecastingcited by 36
  12. 2021Expression might be enough: representing pressure and demand for reinforcement learning based traffic signal controlcited by 28
  13. 2016Online Rules for Control of False Discovery Rate and False Discovery Exceedancecited by 21
  14. 2016AdaPT: An interactive procedure for multiple testing with side informationcited by 15
  15. 2021Online false discovery rate control for anomaly detection in time seriescited by 13
  16. 2024WeatherGFM: Learning A Weather Generalist Foundation Model via In-context Learningcited by 12
  17. 2020Conditional calibration for false discovery rate control under dependencecited by 11
  18. 2017A unified treatment of multiple testing with prior knowledge using the p-filtercited by 10
  19. 2020Non-Parametric Graph Learning for Bayesian Graph Neural Networkscited by 10
  20. 2025Travel Time and Weather-Aware Traffic Forecasting in a Conformal Graph Neural Network Frameworkcited by 3
  21. 2025STACI: Spatio-Temporal Aleatoric Conformal Inferencecited by 1
  22. 2025ASTER: Adaptive Spatio-Temporal Early Decision Model for Dynamic Resource Allocationcited by 1
  23. 2025RIPCN: A Road Impedance Principal Component Network for Probabilistic Traffic Flow Forecastingcited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026A Risk-Sensitive and Uncertainty-Aware Decision-Making and Control Framework for Safe and Robust Autonomous Driving
  2. 2025Uncertainty-Aware Safety-Critical Decision and Control for Autonomous Vehicles at Unsignalized Intersections
  3. 2026SIGMA: Symmetry-aware, Intelligent, Geometric, Multi-objective Adaptive Control for Robust, Dependable Traffic Management
  4. 2024TrafficGamer: Reliable and Flexible Traffic Simulation for Safety-Critical Scenarios with Game-Theoretic Oracles
  5. 2025Robust Single-Agent Reinforcement Learning for Regional Traffic Signal Control Under Demand Fluctuations
  6. 2026OracleTSC: Oracle-Informed Reward Hurdle and Uncertainty Regularization for Traffic Signal Control
  7. 2024OffLight: An Offline Multi-Agent Reinforcement Learning Framework for Traffic Signal Control
  8. 2023A Fully Data-Driven Approach for Realistic Traffic Signal Control Using Offline Reinforcement Learning