让自动驾驶模型记住失败经历,自动修正类似场景表现。
DriveVLA-M0: Failure-Aware Memory Augmentation for Autonomous Driving

- 构建失败记忆池,用结构化场景和专家轨迹存储过往错误。
- 推理时检索失败案例,通过轻量LoRA机制实现精准修正,延迟仅26.44毫秒。
- 无需重训练即可通过扩充记忆提升性能,适合追求鲁棒性的自动驾驶研究者。
视觉-语言-动作(VLA)模型在端到端自动驾驶中展现出统一感知、语言与规划推理的潜力。然而,现有方法缺乏利用历史失败或适应分布偏移的机制,导致在曾失败的相似场景中持续表现不佳。本文提出DriveVLA-M0,一种具备故障感知潜空间记忆的检索增强VLA模型。我们构建一个潜空间记忆池,存储失败案例及其结构化场景表示与专家轨迹标签,并设计专用检索模型,解耦静态道路结构与动态交互以实现结构化检索。推理时,通过轻量级解耦式LoRA测试时训练(TTT)机制注入检索案例,实现无需修改主干网络的针对性修正。在NAVSIMv1和NAVSIMv2基准上的大量实验表明,该方法持续优于先前方法,在Navtest上取得94.1 PDMS,在Navhard上取得47.0 EPDMS,且测试时训练仅带来26.44毫秒的后向传播延迟开销。此外,我们证明DriveVLA-M0可通过增加记忆有效扩展,实现无需训练的性能提升。代码已开源。
原文摘要 · Abstract (English)
Vision-Language-Action (VLA) models have recently emerged as a promising paradigm for end-to-end autonomous driving by enabling unified reasoning across perception, language, and planning. However, existing approaches lack mechanisms to exploit past failures or adapt to distribution shifts, causing the model to persistently underperform on similar scenarios where it has previously failed. In this paper, we propose DriveVLA-M0, a retrieval-augmented VLA with failure-aware latent memory. We construct a latent memory pool that stores failure cases along with their structure scene representations and expert trajectory labels, and design a dedicated Retrieve Model that decouples static road structure and dynamic agent interactions to enable structurally grounded retrieval. At inference time, retrieved cases are injected into the model via a lightweight decoupled LoRA-based test-time training (TTT) mechanism, allowing targeted and scenario-specific correction without modifying the backbone. Extensive experiments on NAVSIMv1 and NAVSIMv2 benchmark demonstrate that our approach consistently outperforms prior methods, achieving 94.1 PDMS on Navtest and 47.0 EPDMS on Navhard with only 26.44 ms TTT backward latency overhead. Furthermore, we show that DriveVLA-M0 scales effectively with additional memory, enabling training-free performance gains through memory expansion. The code is available at https://github.com/ZebinX/DriveVLA-M0.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。