Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

InDRiVE: Reward-Free World-Model Pretraining for Autonomous Driving via Latent Disagreement

About

Model-based reinforcement learning (MBRL) can reduce interaction cost for autonomous driving by learning a predictive world model, but it typically still depends on task-specific rewards that are difficult to design and often brittle under distribution shift. This paper presents InDRiVE, a DreamerV3-style MBRL agent that performs reward-free pretraining in CARLA using only intrinsic motivation derived from latent ensemble disagreement. Disagreement acts as a proxy for epistemic uncertainty and drives the agent toward under-explored driving situations, while an imagination-based actor-critic learns a planner-free exploration policy directly from the learned world model. After intrinsic pretraining, we evaluate zero-shot transfer by freezing all parameters and deploying the pretrained exploration policy in unseen towns and routes. We then study few-shot adaptation by training a task policy with limited extrinsic feedback for downstream objectives (lane following and collision avoidance). Experiments in CARLA across towns, routes, and traffic densities show that disagreement-based pretraining yields stronger zero-shot robustness and robust few-shot collision avoidance under town shift and matched interaction budgets, supporting the use of intrinsic disagreement as a practical reward-free pretraining signal for reusable driving world models.

Feeza Khan Khanzada, Jaerock Kwon• 2025

Related benchmarks

TaskDatasetResultRank
Autonomous DrivingCARLA Town01 & Town02
Town01 SR96.43
7
Collision AvoidanceCARLA Town01 (seen)
SR (Straight)1
4
Collision AvoidanceCARLA Town02 (unseen)
SR (Straight)100
4
Lane FollowingCARLA Town01 (seen)
SR (Straight)100
4
Lane FollowingCARLA Town02 (unseen)
Success Rate (Straight)99
4
Lane FollowingCARLA Town01 & Town02
SR Town0197.59
4
Autonomous DrivingCARLA Town01 (seen)
Straight Success Rate76.5
3
Autonomous DrivingCARLA Town02 (unseen)
SR (Straight)75
3
Showing 8 of 8 rows

Other info

Follow for update