Search papers, labs, and topics across Lattice.
DriveWAM adapts a pretrained video diffusion transformer into an autoregressive video-action policy for autonomous driving by organizing video and action streams into a unified temporal token sequence trained with a joint flow-matching objective. The approach incorporates high-level scene understanding via a frozen VLM that provides chunk-specific semantic intent and uses selective KV memory to bound long-horizon rollout. Experiments on NAVSIM and PhysicalAI-AV demonstrate strong planning performance and data scaling potential up to 100k driving clips.
Unleashing pretrained video diffusion models for autonomous driving yields surprisingly strong planning performance, suggesting a new path beyond vision-language models.
Pretrained foundation models have become an important basis for end-to-end autonomous driving. In contrast to vision-language models pretrained primarily on static image-text pairs, video generative models capture temporal dynamics and motion priors that are naturally suited for driving. We present DriveWAM, a driving world-action model that adapts a pretrained video diffusion transformer into an autoregressive video-action policy. DriveWAM organizes video and action streams into a unified temporal token sequence and trains them under a joint flow-matching objective, preserving the pretrained video-generation architecture while adapting its large-scale video priors to action generation. To incorporate high-level scene understanding, we introduce scene-evolving driving guidance, where a frozen VLM produces chunk-specific semantic intent to guide video-action generation. To keep long-horizon rollout bounded, we further introduce selective KV memory, which maintains bounded modality-aware video and action memory pools through relevance-redundancy cache selection at inference time. Experiments on NAVSIM and the PhysicalAI-Autonomous-Vehicles benchmark show that DriveWAM achieves strong planning performance, and a data-scaling study from 4k to 100k driving clips further confirms the scaling potential of world-action modeling for end-to-end autonomous driving.
Mila