Search papers, labs, and topics across Lattice.
This paper introduces the Action Diffusion Transformer (ADT), which advances end-to-end autonomous driving by predicting multiple action candidates instead of a single deterministic output. The authors demonstrate that this multimodal approach significantly enhances driving performance and training stability, achieving state-of-the-art results on the Bench2Drive benchmark while reducing latency by a factor of ten. By validating the importance of action multimodality, the study highlights its critical role in improving representational quality and behavioral consistency in autonomous driving systems.
Multimodal action prediction not only boosts performance but also reduces latency, reshaping the future of autonomous driving systems.
End-to-End Autonomous Driving (E2E-AD) systems have largely converged on predicting intermediate trajectory waypoints, delegating final control to hand-crafted controllers with GPS access. Direct control-signal prediction (outputting throttle, steer and brake in an end-to-end fashion) remains underexplored, and critically, the role of action multimodality in such systems is not well understood. We argue that moving beyond deterministic, single-action outputs is not merely a modelling choice, but a key driver of driving performance, representational quality, and training stability. To validate this, we introduce the Action Diffusion Transformer (ADT), an anchor-free diffusion transformer trained with a MSE objective that natively models the multimodal distribution of plausible driving actions. Rather than committing to a single deterministic command, ADT generates K action candidates and selects the most suitable one at inference via Nearest Neighbour Matching (NNM). Beyond strong benchmark numbers, we show that action multimodality yields measurable benefits in learned representations and behavioral consistency, effects that deterministic architectures cannot replicate. ADT surpasses previous state-of-the-art on the challenging closed-loop Bench2Drive benchmark while achieving ten times lower latency, demonstrating that expressive, multimodal action modelling is both practically efficient and conceptually essential for robust end-to-end driving.