AD-R1: Closed-Loop Reinforcement Learning for End-to-End Autonomous Driving with Impartial World Models
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
- 2025I2-World: Intra-Inter Tokenization for Efficient Dynamic 4D Scene Forecasting12 Jul
- 2025World4Drive: End-to-End Autonomous Driving via Intention-aware Physical Latent World Model1 Jul
- 2025Epona: Autoregressive Diffusion World Model for Autonomous Driving30 Jun
- 2025COME: Adding Scene-Centric Forecasting Control to Occupancy World Model16 Jun
- 2025Raw2Drive: Reinforcement Learning with Aligned World Models for End-to-End Autonomous Driving (in CARLA v2)22 May
- 2025End-to-End Driving with Online Trajectory Evaluation via BEV World Model2 Apr
- 2024DrivingGPT: Unifying Driving World Modeling and Planning with Multi-modal Autoregressive Transformers24 Dec
- 2024DOME: Taming Diffusion Model into High-Fidelity Controllable Occupancy World Model14 Oct
- 2024Enhancing End-to-End Autonomous Driving with Latent World Model12 Jun
- 2024Vista: A Generalizable Driving World Model with High Fidelity and Versatile Controllability27 May
- 2024DriveDreamer-2: LLM-Enhanced World Models for Diverse Driving Video Generation11 Mar
- 2023OccWorld: Learning a 3D Occupancy World Model for Autonomous Driving27 Nov
3 more
Other references · 27
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2019nuScenes: A multimodal dataset for autonomous drivingcited by 4,485
- 2019Scalability in Perception for Autonomous Driving: Waymo Open Datasetcited by 2,436
- 2022Planning-oriented Autonomous Drivingcited by 1,068
- 2023VAD: Vectorized Scene Representation for Efficient Autonomous Drivingcited by 613
- 2022TransFuser: Imitation with Transformer-Based Sensor Fusion for Autonomous Drivingcited by 477
- 2024NAVSIM: Data-Driven Non-Reactive Autonomous Vehicle Simulation and Benchmarkingcited by 343
- 2024DiffusionDrive: Truncated Diffusion Model for End-to-End Autonomous Drivingcited by 324
- 2024VADv2: End-to-End Vectorized Autonomous Driving via Probabilistic Planningcited by 287
- 2023MagicDrive: Street View Generation with Diverse 3D Geometry Controlcited by 250
- 2024Hydra-MDP: End-to-end Multimodal Planning with Multi-target Hydra-Distillationcited by 244
15 more
- 2025ReCogDrive: A Reinforced Cognitive Framework for End-to-End Autonomous Drivingcited by 190
- 2025VLA-RL: Towards Masterful and General Robotic Manipulation with Scalable Reinforcement Learningcited by 166
- 2025GoalFlow: Goal-Driven Flow Matching for Multimodal Trajectories Generation in End-to-End Autonomous Drivingcited by 117
- 2024DriveArena: A Closed-loop Generative Simulation Platform for Autonomous Drivingcited by 98
- 2024OccSora: 4D Occupancy Generation Models as World Simulators for Autonomous Drivingcited by 76
- 2025RAD: Training an End-to-End Driving Policy via Large-Scale 3DGS-based Reinforcement Learningcited by 76
- 20253D and 4D World Modeling: A Surveycited by 64
- 2024CarLLaVA: Vision language models for camera-only closed-loop drivingcited by 57
- 2025DriveDPO: Policy Learning via Safety DPO For End-to-End Autonomous Drivingcited by 39
- 2024DrivingSphere: Building a High-fidelity 4D World for Closed-loop Simulationcited by 34
- 2025Reinforced Refinement with Self-Aware Expansion for End-to-End Autonomous Drivingcited by 19
- 2024OLiDM: Object-aware LiDAR Diffusion Models for Autonomous Drivingcited by 13
- 2025MCN-SLAM: Multi-Agent Collaborative Neural SLAM with Hybrid Implicit Neural Scene Representationcited by 11
- 2025RLGF: Reinforcement Learning with Geometric Feedback for Autonomous Driving Video Generationcited by 5
- 2022Plug-and-Play Pseudo Label Correction Network for Unsupervised Person Re-identificationcited by 2
Built on by
Citing papers per year: 2025: 1; 2026: 12.
13 papers.
Sort:
By area:
- 2025Vision-Language-Action Models for Autonomous Driving: Past, Present, and Futurecited by 32
- 2026DriveWorld-VLA: Unified Latent-Space World Modeling with Vision-Language-Action for Autonomous Drivingcited by 31
- 2026UniDriveVLA: Unifying Understanding, Perception, and Action Planning for Autonomous Drivingcited by 17
- 2026DreamerAD: Efficient Reinforcement Learning via Latent World Model for Autonomous Drivingcited by 13
- 2026ExploreVLA: Dense World Modeling and Exploration for End-to-End Autonomous Drivingcited by 10
- 2026RAD-2: Scaling Reinforcement Learning in a Generator-Discriminator Frameworkcited by 7
- 2026World Engine: Towards the Era of Post-Training for Autonomous Drivingcited by 6
- 2026CausalDrive: Real-time Causal World Models for Autonomous Drivingcited by 4
- 2026Latent World Models for Automated Driving: A Unified Taxonomy, Evaluation Framework, and Open Challengescited by 3
- 2026Distill to Think, Foresee to Act: Cognitive-Physical Reinforcement Learning for Autonomous Drivingcited by 3
- 2026DynFlowDrive: Flow-Based Dynamic World Modeling for Autonomous Drivingcited by 2
- 2026CRAFT: Counterfactual-to-Interactive Reinforcement Fine-Tuning for Driving Policiescited by 2
1 more
- 2026Quo Vadis, World Modeling?cited by 0
- 2026Quo Vadis, World Modeling?3 Aug 2026
- 2026World Engine: Towards the Era of Post-Training for Autonomous Driving18 Jun 2026
- 2026CausalDrive: Real-time Causal World Models for Autonomous Driving13 Jun 2026
- 2026Distill to Think, Foresee to Act: Cognitive-Physical Reinforcement Learning for Autonomous Driving20 May 2026
- 2026CRAFT: Counterfactual-to-Interactive Reinforcement Fine-Tuning for Driving Policies6 May 2026
- 2026RAD-2: Scaling Reinforcement Learning in a Generator-Discriminator Framework16 Apr 2026
- 2026ExploreVLA: Dense World Modeling and Exploration for End-to-End Autonomous Driving3 Apr 2026
- 2026UniDriveVLA: Unifying Understanding, Perception, and Action Planning for Autonomous Driving2 Apr 2026
- 2026DreamerAD: Efficient Reinforcement Learning via Latent World Model for Autonomous Driving25 Mar 2026
- 2026DynFlowDrive: Flow-Based Dynamic World Modeling for Autonomous Driving20 Mar 2026
- 2026Latent World Models for Automated Driving: A Unified Taxonomy, Evaluation Framework, and Open Challenges10 Mar 2026
- 2026DriveWorld-VLA: Unified Latent-Space World Modeling with Vision-Language-Action for Autonomous Driving6 Feb 2026
1 more
- 2026DriveWorld-VLA: Unified Latent-Space World Modeling with Vision-Language-Action for Autonomous Drivingcited by 31
- 2026DreamerAD: Efficient Reinforcement Learning via Latent World Model for Autonomous Drivingcited by 13
- 2026ExploreVLA: Dense World Modeling and Exploration for End-to-End Autonomous Drivingcited by 10
- 2026CausalDrive: Real-time Causal World Models for Autonomous Drivingcited by 4
- 2026Latent World Models for Automated Driving: A Unified Taxonomy, Evaluation Framework, and Open Challengescited by 3
- 2026DynFlowDrive: Flow-Based Dynamic World Modeling for Autonomous Drivingcited by 2
- 2026Quo Vadis, World Modeling?cited by 0
- 2026Quo Vadis, World Modeling?3 Aug 2026
- 2026CausalDrive: Real-time Causal World Models for Autonomous Driving13 Jun 2026
- 2026ExploreVLA: Dense World Modeling and Exploration for End-to-End Autonomous Driving3 Apr 2026
- 2026DreamerAD: Efficient Reinforcement Learning via Latent World Model for Autonomous Driving25 Mar 2026
- 2026DynFlowDrive: Flow-Based Dynamic World Modeling for Autonomous Driving20 Mar 2026
- 2026Latent World Models for Automated Driving: A Unified Taxonomy, Evaluation Framework, and Open Challenges10 Mar 2026
- 2026DriveWorld-VLA: Unified Latent-Space World Modeling with Vision-Language-Action for Autonomous Driving6 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2024From Imitation to Exploration: End-to-end Autonomous Driving based on World Model
- 2026RoXDrive: Closed-Loop Reinforcement Learning for End-to-End Autonomous Driving via Action-Faithful Rollouts
- 2026Dreamer-SAC: Off-Policy Learning in Latent World Models for Sample-Efficient Autonomous Driving
- 2022Driver Dojo: A Benchmark for Generalizable Reinforcement Learning for Autonomous Driving
- 2025InDRiVE: Reward-Free World-Model Pretraining for Autonomous Driving via Latent Disagreement
- 2026DriveZero: End-to-End Driving Beyond Human Demonstrations
- 2025Model-Based Policy Adaptation for Closed-Loop End-to-End Autonomous Driving
- 2024Autonomous Algorithm for Training Autonomous Vehicles with Minimal Human Intervention