mlsanityData to 14 Oct 2026

Multimodal Action Diffusion for Robust End-to-End Autonomous Driving

arXiv 2606.02105 ↗First posted 1 Jun 2026cs.CVcited by 0

On the map: Diffusion models → Diffusion and flow policies for robotics

also in 1 more
Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 38 of its 44 references that are in the map. First posted on arXiv 1 Jun 2026.

Builds on

38 of 44 references.

Closest ancestors · 4

Same direction, newest first

  1. 2025DiffRefiner: Coarse to Fine Trajectory Planning via Diffusion Refinement with Semantic Interaction for End to End Autonomous Driving21 Nov
  2. 2025BridgeDrive: Diffusion Bridge Policy for Closed-Loop Trajectory Planning in Autonomous Driving28 Sep
  3. 2024DiffusionDrive: Truncated Diffusion Model for End-to-End Autonomous Driving22 Nov
  4. 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusion7 Mar

Other references · 34

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  4. 2020Denoising Diffusion Implicit Modelscited by 9,781
  5. 2019nuScenes: A multimodal dataset for autonomous drivingcited by 4,523
  6. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  7. 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
  8. 2017CARLA: An Open Urban Driving Simulatorcited by 2,449
  9. 2025FAST: Efficient Action Tokenization for Vision-Language-Action Modelscited by 665
  10. 2023End-to-end Autonomous Driving: Challenges and Frontierscited by 490
  11. 2017End-to-end Driving via Conditional Imitation Learningcited by 485
  12. 2022TransFuser: Imitation with Transformer-Based Sensor Fusion for Autonomous Drivingcited by 485
22 more
  1. 2024NAVSIM: Data-Driven Non-Reactive Autonomous Vehicle Simulation and Benchmarkingcited by 354
  2. 2019Exploring the Limitations of Behavior Cloning for Autonomous Drivingcited by 346
  3. 2024Bench2Drive: Towards Multi-Ability Benchmarking of Closed-Loop End-To-End Autonomous Drivingcited by 252
  4. 2024Hydra-MDP: End-to-end Multimodal Planning with Multi-target Hydra-Distillationcited by 249
  5. 2022Trajectory-guided Control Prediction for End-to-end Autonomous Driving: A Simple yet Strong Baselinecited by 248
  6. 2023Is Ego Status All You Need for Open-Loop End-to-End Autonomous Driving?cited by 242
  7. 2022Safety-Enhanced Autonomous Driving Using Interpretable Sensor Fusion Transformercited by 212
  8. 2021End-to-End Urban Driving by Imitating a Reinforcement Learning Coachcited by 194
  9. 2023Rethinking the Open-Loop Evaluation of End-to-End Autonomous Driving in nuScenescited by 191
  10. 2022Model-Based Imitation Learning for Urban Drivingcited by 150
  11. 2023DriveAdapter: Breaking the Coupling Barrier of Perception and Planning in End-to-End Autonomous Drivingcited by 144
  12. 2025DriveTransformer: Unified Transformer for Scalable End-to-End Autonomous Drivingcited by 126
  13. 2019Urban Driving with Conditional Imitation Learningcited by 86
  14. 2018On Offline Evaluation of Vision-based Driving Modelscited by 64
  15. 2024Think2Drive: Efficient Reinforcement Learning by Thinking in Latent World Model for Quasi-Realistic Autonomous Driving (in CARLA-v2)cited by 52
  16. 2025Hydra-NeXt: Robust Closed-Loop Driving with Open-Loop Trainingcited by 39
  17. 2022MUAD: Multiple Uncertainties for Autonomous Driving, a benchmark for multiple uncertainty types and taskscited by 24
  18. 2024LOGen: Toward Lidar Object Generation by Point Diffusioncited by 9
  19. 2025ETA: Efficiency through Thinking Ahead, A Dual Approach to Self-Driving with Large Modelscited by 9
  20. 2023Scaling Vision-based End-to-End Driving with Multi-View Attention Learningcited by 8
  21. 2025Tracking Meets Large Multimodal Models for Driving Scenario Understandingcited by 7
  22. 2025Mapping like a Skeptic: Probabilistic BEV Projection for Online HD Mappingcited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025DiffE2E: Rethinking End-to-End Driving with a Hybrid Action Diffusion and Supervised Policy
  2. 2025DiffusionDriveV2: Reinforcement Learning-Constrained Truncated Diffusion Modeling in End-to-End Autonomous Driving
  3. 2024Controllable Diverse Sampling for Diffusion Based Motion Behavior Forecasting
  4. 2024DrivingGPT: Unifying Driving World Modeling and Planning with Multi-modal Autoregressive Transformers
  5. 2025Diffusion-Based Planning for Autonomous Driving with Flexible Guidance
  6. 2025Discrete Diffusion for Reflective Vision-Language-Action Models in Autonomous Driving
  7. 2026Learning to Explain While Planning: Rule-Aligned Diffusion Planning for Autonomous Driving
  8. 2026Post-Training in End-to-End Autonomous Driving