mlsanityData to 7 Oct 2026

AD-R1: Closed-Loop Reinforcement Learning for End-to-End Autonomous Driving with Impartial World Models

arXiv 2511.20325 ↗First posted 25 Nov 2025cs.CVcited by 13

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

also in 1 more
Abstract (arXiv)

End-to-end models for autonomous driving hold the promise of learning complex behaviors directly from sensor data, but face critical challenges in safety and handling long-tail events. Reinforcement Learning (RL) offers a promising path to overcome these limitations, yet its success in autonomous driving has been elusive. We identify a fundamental flaw hindering this progress: a deep seated optimistic bias in the world models used for RL. To address this, we introduce a framework for post-training policy refinement built around an Impartial World Model. Our primary contribution is to teach this model to be honest about danger. We achieve this with a novel data synthesis pipeline, Counterfactual Synthesis, which systematically generates a rich curriculum of plausible collisions and off-road events. This transforms the model from a passive scene completer into a veridical forecaster that remains faithful to the causal link between actions and outcomes. We then integrate this Impartial World Model into our closed-loop RL framework, where it serves as an internal critic. During refinement, the agent queries the critic to ``dream" of the outcomes for candidate actions. We demonstrate through extensive experiments, including on a new Risk Foreseeing Benchmark, that our model significantly outperforms baselines in predicting failures. Consequently, when used as a critic, it enables a substantial reduction in safety violations in challenging simulations, proving that teaching a model to dream of danger is a critical step towards building truly safe and intelligent autonomous agents.

Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Placed in World models for autonomous driving, one of 11 directions in World models. Builds on 42 of its 52 references that are in the map. First posted on arXiv 25 Nov 2025.

Builds on

42 of 52 references.

Closest ancestors · 15

Same direction, newest first

  1. 2025I2-World: Intra-Inter Tokenization for Efficient Dynamic 4D Scene Forecasting12 Jul
  2. 2025World4Drive: End-to-End Autonomous Driving via Intention-aware Physical Latent World Model1 Jul
  3. 2025Epona: Autoregressive Diffusion World Model for Autonomous Driving30 Jun
  4. 2025COME: Adding Scene-Centric Forecasting Control to Occupancy World Model16 Jun
  5. 2025Raw2Drive: Reinforcement Learning with Aligned World Models for End-to-End Autonomous Driving (in CARLA v2)22 May
  6. 2025End-to-End Driving with Online Trajectory Evaluation via BEV World Model2 Apr
  7. 2024DrivingGPT: Unifying Driving World Modeling and Planning with Multi-modal Autoregressive Transformers24 Dec
  8. 2024DOME: Taming Diffusion Model into High-Fidelity Controllable Occupancy World Model14 Oct
  9. 2024Enhancing End-to-End Autonomous Driving with Latent World Model12 Jun
  10. 2024Vista: A Generalizable Driving World Model with High Fidelity and Versatile Controllability27 May
  11. 2024DriveDreamer-2: LLM-Enhanced World Models for Diverse Driving Video Generation11 Mar
  12. 2023OccWorld: Learning a 3D Occupancy World Model for Autonomous Driving27 Nov
3 more
  1. 2023ADriver-I: A General World Model for Autonomous Driving22 Nov
  2. 2023Occ3D: A Large-Scale 3D Occupancy Prediction Benchmark for Autonomous Driving27 Apr
  3. 2017CARLA: An Open Urban Driving Simulator10 Nov

Other references · 27

  1. 2022Training language models to follow instructions with human feedbackcited by 15,712
  2. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
  3. 2019nuScenes: A multimodal dataset for autonomous drivingcited by 4,485
  4. 2019Scalability in Perception for Autonomous Driving: Waymo Open Datasetcited by 2,436
  5. 2022Planning-oriented Autonomous Drivingcited by 1,068
  6. 2023VAD: Vectorized Scene Representation for Efficient Autonomous Drivingcited by 613
  7. 2022TransFuser: Imitation with Transformer-Based Sensor Fusion for Autonomous Drivingcited by 477
  8. 2024NAVSIM: Data-Driven Non-Reactive Autonomous Vehicle Simulation and Benchmarkingcited by 343
  9. 2024DiffusionDrive: Truncated Diffusion Model for End-to-End Autonomous Drivingcited by 324
  10. 2024VADv2: End-to-End Vectorized Autonomous Driving via Probabilistic Planningcited by 287
  11. 2023MagicDrive: Street View Generation with Diverse 3D Geometry Controlcited by 250
  12. 2024Hydra-MDP: End-to-end Multimodal Planning with Multi-target Hydra-Distillationcited by 244
15 more
  1. 2025ReCogDrive: A Reinforced Cognitive Framework for End-to-End Autonomous Drivingcited by 190
  2. 2025VLA-RL: Towards Masterful and General Robotic Manipulation with Scalable Reinforcement Learningcited by 166
  3. 2025GoalFlow: Goal-Driven Flow Matching for Multimodal Trajectories Generation in End-to-End Autonomous Drivingcited by 117
  4. 2024DriveArena: A Closed-loop Generative Simulation Platform for Autonomous Drivingcited by 98
  5. 2024OccSora: 4D Occupancy Generation Models as World Simulators for Autonomous Drivingcited by 76
  6. 2025RAD: Training an End-to-End Driving Policy via Large-Scale 3DGS-based Reinforcement Learningcited by 76
  7. 20253D and 4D World Modeling: A Surveycited by 64
  8. 2024CarLLaVA: Vision language models for camera-only closed-loop drivingcited by 57
  9. 2025DriveDPO: Policy Learning via Safety DPO For End-to-End Autonomous Drivingcited by 39
  10. 2024DrivingSphere: Building a High-fidelity 4D World for Closed-loop Simulationcited by 34
  11. 2025Reinforced Refinement with Self-Aware Expansion for End-to-End Autonomous Drivingcited by 19
  12. 2024OLiDM: Object-aware LiDAR Diffusion Models for Autonomous Drivingcited by 13
  13. 2025MCN-SLAM: Multi-Agent Collaborative Neural SLAM with Hybrid Implicit Neural Scene Representationcited by 11
  14. 2025RLGF: Reinforcement Learning with Geometric Feedback for Autonomous Driving Video Generationcited by 5
  15. 2022Plug-and-Play Pseudo Label Correction Network for Unsupervised Person Re-identificationcited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 1; 2026: 12.

13 papers.

Sort:

By area:

Similar, not cited

Close in wording, no citation either way

  1. 2024From Imitation to Exploration: End-to-end Autonomous Driving based on World Model
  2. 2026RoXDrive: Closed-Loop Reinforcement Learning for End-to-End Autonomous Driving via Action-Faithful Rollouts
  3. 2026Dreamer-SAC: Off-Policy Learning in Latent World Models for Sample-Efficient Autonomous Driving
  4. 2022Driver Dojo: A Benchmark for Generalizable Reinforcement Learning for Autonomous Driving
  5. 2025InDRiVE: Reward-Free World-Model Pretraining for Autonomous Driving via Latent Disagreement
  6. 2026DriveZero: End-to-End Driving Beyond Human Demonstrations
  7. 2025Model-Based Policy Adaptation for Closed-Loop End-to-End Autonomous Driving
  8. 2024Autonomous Algorithm for Training Autonomous Vehicles with Minimal Human Intervention