Think2Drive: Efficient Reinforcement Learning by Thinking in Latent World Model for Quasi-Realistic Autonomous Driving
TLDR
Think2Drive uses a latent world model for efficient model-based RL to achieve expert-level autonomous driving in CARLA v2 within 3 days.
Reasoning
The paper introduces a novel model-based RL method with a world model for autonomous driving, achieving strong results in the challenging CARLA v2 simulator with high training efficiency. However, it lacks real-world validation and provides limited architectural details of the world model.
Read-first score
Read-first score 59.5, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 40.
Field roles
Rank sensitivity
Stability: volatile; rank range: 120.
Keyword Scores
Deep Analysis
Innovations
- First model-based reinforcement learning method for autonomous driving (Think2Drive) using a world model as a neural simulator to train a planner.
- CornerCase-Repository, a benchmark for evaluating driving models by scenarios.
- A new balanced metric for driving performance combining route completion, infraction number, and scenario density.
Methodology
Think2Drive employs a model-based RL approach where a world model learns the environment transitions and then acts as a neural simulator to train a planner. The method leverages low-dimensional state space and parallel tensor computing in the world model to boost training efficiency. Training is conducted on the CARLA v2 simulator.
Key Results
Think2Drive achieves expert-level proficiency in CARLA v2 within 3 days of training on a single A6000 GPU, and is the first reported method to achieve 100% route completion on CARLA v2.