AD-R1: Closed-Loop Reinforcement Learning for End-to-End Autonomous Driving with Impartial World Models
TLDR
Introduces an Impartial World Model with counterfactual synthesis to reduce optimistic bias in RL for autonomous driving, improving safety in simulations.
Reasoning
The paper identifies a key flaw (optimistic bias) in world models for RL and proposes a novel counterfactual synthesis pipeline to create an Impartial World Model. Strengths include a clear problem statement and integration into closed-loop RL, but weaknesses are the lack of real-world validation and reliance on simulated environments only.
Read-first score
Read-first score 60.2, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 47.
Field roles
Rank sensitivity
Stability: volatile; rank range: 265.
Keyword Scores
Deep Analysis
Innovations
- Identification of a fundamental optimistic bias in world models used for reinforcement learning in autonomous driving
- Introduction of an Impartial World Model that is honest about danger
- Counterfactual Synthesis data pipeline for systematically generating a curriculum of plausible collisions and off-road events
- Closed-loop reinforcement learning framework where the Impartial World Model serves as an internal critic for post-training policy refinement
- Risk Foreseeing Benchmark for evaluating failure prediction
Methodology
The paper proposes a post-training policy refinement framework built around an Impartial World Model. The model is trained using a novel Counterfactual Synthesis pipeline that generates a curriculum of plausible collisions and off-road events, transforming it from a passive scene completer into a veridical forecaster. This Impartial World Model is then integrated as an internal critic in a closed-loop RL framework, where the agent queries the critic to 'dream' of outcomes for candidate actions during refinement.
Key Results
The Impartial World Model significantly outperforms baselines in predicting failures on the Risk Foreseeing Benchmark. When used as a critic in the closed-loop RL framework, it enables a substantial reduction in safety violations in challenging simulations.