DA-WAM: Decision-Aligned Future Latents for Driving World Models

TL;DR

提出DA-WAM,通过动作条件未来潜在表示与轨迹评分结合,实现自主驾驶场景的决策对齐,显著优于现有方法。

cs.RO 🔴 高级 2026-08-20 81 次浏览
Ruiguo Zhong Benshan Ma Xiaolong Chen Lang Zhang Mingyue Feng Yaonong Wang Pei Liu Jun Ma
自主驾驶 世界模型 未来潜在表示 轨迹规划 深度学习

核心发现

方法论

本文提出的DA-WAM框架融合了预测表示学习、动作条件未来建模与轨迹评分,核心在于通过在线编码器和动量目标保持未来表示的持续优化。利用基于V-JEPA的在线编码器提取场景特征,结合低秩适应(LoRA)模块进行微调,确保未来潜在表示在规划过程中不断适应任务需求。每个候选轨迹通过动作编码器映射为动作表示,再由动作条件预测器生成对应的未来潜在状态,最后利用未来潜在条件的因子化评分器进行轨迹评价。训练过程中,采用专家匹配和硬负样本策略,确保未来潜在表示与实际观察紧密对应,同时增强模型在安全边界的判别能力。推理阶段,模型仅依赖在线编码器、预测器和评分器,实时生成轨迹评分,实现动作决策的决策对齐。整个体系通过端到端优化,确保未来潜在表示与轨迹评分的紧密结合,提升自主驾驶的安全性和效率。

关键结果

  • 在NAVSIM-v1和NAVSIM-v2数据集上,DA-WAM在PDMS指标上分别达到93.7和94.2,显著优于现有最优模型,提升幅度超过1.5个百分点,验证了其在复杂场景中的优越表现。
  • 通过消融实验,去除未来潜在表示或硬负样本策略,模型性能下降约2-3点,说明未来潜在的动作条件建模和安全判别机制对性能提升具有关键作用。
  • 在不同复杂度的场景下,DA-WAM展现出较强的鲁棒性,尤其在边界决策和碰撞预警方面表现优异,显著减少了安全违规事件。

研究意义

该研究突破了传统世界模型与规划解耦的局限,将未来预测与轨迹决策紧密结合,极大提升自主驾驶系统的安全性和决策效率。通过动作条件的未来潜在表示,模型能够更细粒度地理解不同轨迹的潜在后果,为未来自主驾驶中的场景理解和风险评估提供了新的技术路径。这不仅推动了深度学习在自主驾驶中的应用,也为未来智能交通系统的安全保障提供了理论基础和实践方案,有望在自动驾驶、智能交通管理等领域产生深远影响。

技术贡献

本文提出的DA-WAM在技术上实现了未来潜在表示的动作条件化,创新点在于引入持续优化的在线编码器和动量目标,确保未来潜在表示在规划过程中不断适应任务需求。通过动作条件预测器为每个轨迹生成独特的未来潜在状态,解决了以往模型中共享未来状态导致的动作-后果不匹配问题。结合专家匹配和硬负样本策略,有效提升模型在安全边界的判别能力。整体架构实现了未来预测与轨迹评分的端到端联合优化,显著优于传统的分阶段训练方法,为自主驾驶中的决策对齐提供了新思路。

新颖性

该研究的创新在于首次提出动作条件的未来潜在表示与轨迹评分一体化框架,确保每个候选轨迹都拥有独立的未来预测,从而实现决策的动作对齐。与现有的Latent-WAM、Drive-WM等模型不同,DA-WAM在训练和推理过程中持续优化未来潜在表示,使其与规划目标保持一致,解决了过去模型中未来状态共享导致的动作-后果模糊问题。这一设计极大提升了模型在复杂场景中的安全性和鲁棒性,代表了自主驾驶未来场景建模的一个重要突破。

局限性

  • 模型对大规模、多样化场景的泛化能力仍有限,特别是在极端或未见过的交通环境中可能表现不佳,原因在于训练数据的有限性和模型对特定场景特征的依赖。
  • 实时推理的计算成本较高,尤其是在候选轨迹数目较多时,可能影响实际部署的效率,未来需优化模型结构以提升推理速度。
  • 硬负样本策略虽然增强了安全判别能力,但在极端安全边界情况下仍可能出现误判,需结合更多安全规则和多模态信息进行补充。

未来方向

未来工作将聚焦于提升模型的泛化能力,探索多模态信息融合(如激光雷达、语义地图)以增强场景理解。同时,将优化模型结构以降低推理成本,推动其在实际自动驾驶系统中的部署。此外,计划引入强化学习机制,进一步提升模型在动态环境中的适应性和安全性,推动自主驾驶技术的全面升级。

AI 总览摘要

自主驾驶技术的核心挑战之一在于如何准确预测场景未来的演变,并将这些预测结果有效融入决策过程。传统的世界模型多偏重于场景的预测能力,忽视了预测结果对轨迹选择的指导作用,导致决策的动作-后果关系不够紧密。为解决这一问题,本文提出了DA-WAM(Decision-Aligned World Action Model),一种创新的端到端框架,旨在实现未来潜在表示的动作条件化与轨迹评分的深度结合。

该方法的核心在于通过持续优化的在线编码器和动量目标,保持未来潜在表示在规划过程中的适应性。每个候选轨迹由动作编码器映射为动作表示,再由动作条件预测器生成对应的未来潜在状态,最后由未来潜在条件的因子化评分器进行轨迹评价。这一设计确保每个轨迹都拥有独立的未来预测,从而实现动作-后果的精细匹配。

在训练阶段,模型采用专家匹配和硬负样本策略,确保未来潜在表示与实际观察紧密对应,同时增强模型在安全边界的判别能力。推理阶段,模型仅依赖在线编码器、预测器和评分器,实时生成轨迹评分,极大提升了决策的精确性和安全性。

在NAVSIM-v1和NAVSIM-v2两个数据集上的实验结果显示,DA-WAM在PDMS指标上分别达到了93.7和94.2,优于所有对比模型,验证了其在复杂交通场景中的优越性能。消融实验进一步证实了未来潜在表示和硬负样本策略在性能提升中的关键作用。

这一研究不仅在技术层面实现了未来预测与轨迹决策的深度融合,也为自主驾驶系统的安全性和鲁棒性提供了新的解决方案。未来,结合多模态信息和强化学习,DA-WAM有望在实际自动驾驶中实现更广泛的应用,推动智能交通的持续发展。

深度分析

研究背景

随着深度学习技术的快速发展,自主驾驶系统对场景理解和决策能力的要求不断提升。早期的模型多依赖于规则和传统感知算法,难以应对复杂多变的交通环境。近年来,基于深度神经网络的世界模型逐渐成为研究热点,代表性工作包括DriveWorld、Latent-WAM和Drive-JEPA等。这些模型通过预测未来场景或潜在状态,为路径规划提供支持,显著提升了自主驾驶的安全性和效率。然而,现有方法大多将未来预测与轨迹优化解耦,导致预测结果未能直接指导决策,存在动作-后果匹配不足的问题。随着对安全性要求的提高,如何实现未来预测的动作对齐,成为学界亟待解决的核心难题。

核心问题

当前自主驾驶中的世界模型多采用阶段性训练方式,将未来状态的预测与轨迹规划分离,导致预测结果难以直接影响轨迹选择。尤其是在复杂交通场景中,模型可能无法区分不同动作的潜在后果,影响决策的安全性和鲁棒性。此外,模型在实际应用中面临实时性和泛化能力的挑战,如何在保证高精度预测的同时实现动作条件化,成为技术瓶颈。硬负样本策略虽然增强了模型的安全判别能力,但仍未解决未来潜在表示与轨迹评分的深度耦合问题,限制了自主驾驶系统的潜能。

核心创新

本文的创新点在于提出DA-WAM框架,首次实现动作条件的未来潜在表示与轨迹评分的端到端融合。具体包括:• 引入持续优化的在线编码器和动量目标,确保未来潜在表示在规划过程中不断适应任务需求;• 每个候选轨迹通过动作编码器映射为动作表示,再由动作条件预测器生成对应的未来潜在状态,避免共享状态带来的动作-后果模糊;• 采用专家匹配和硬负样本策略,强化模型在安全边界的判别能力。这些创新共同实现了未来预测的动作对齐,显著提升自主驾驶的安全性和决策效率。

方法详解

  • �� 观察编码:利用预训练的V-JEPA基础模型,通过在线编码器Eθ提取场景特征,结合LoRA模块微调,确保未来潜在表示在训练中持续优化。
  • �� 未来预测:每个候选轨迹τi通过动作编码器Eτ映射为动作表示ai,动作条件预测器Pϕ利用ai与场景特征Zt交互,生成候选特定的未来潜在状态bZi。
  • �� 轨迹评分:利用未来潜在条件的因子化评分器Sψ,结合场景特征Zt、未来潜在bZi和动作表示ai,评估每个轨迹的安全性和合理性,输出多维规划因子和整体评分。
  • �� 训练策略:采用专家匹配策略,将观察到的未来目标仅应用于最接近的专家轨迹,硬负样本用于边界判别,端到端优化未来潜在表示与轨迹评分的匹配关系。
  • �� 推理过程:仅使用在线编码器、预测器和评分器,实时生成轨迹评分,确保每个候选轨迹拥有独立的未来预测,决策与动作紧密对齐。

实验设计

在NAVSIM-v1和NAVSIM-v2两个公开数据集上,模型采用12,146个场景进行训练和评估。指标包括PDMS及其五个子指标(NC、DAC、EP、TTC、Comfort),以及扩展的NAVSIM-v2指标。对比模型涵盖DriveWorld、Latent-WAM、Drive-JEPA等,采用相同的训练策略和超参数设置。模型通过消融实验验证未来潜在表示和硬负样本的贡献,进行多场景鲁棒性测试,确保在复杂交通环境中的表现优越。实验还包括实时推理性能评估,验证模型在实际部署中的效率。

结果分析

DA-WAM在NAVSIM-v1和NAVSIM-v2数据集上均取得了优异成绩,PDMS分别达到93.7和94.2,超越现有最优模型,提升幅度超过1.5个百分点。在安全指标方面,硬负样本策略显著减少了碰撞和偏离事件,模型在边界决策上的表现优于对比方法。消融实验显示,去除未来潜在表示或硬负样本会导致性能下降2-3点,验证了其关键作用。整体而言,模型在复杂交通场景中的鲁棒性和安全性得到充分验证,展现出强大的应用潜力。

应用场景

该模型可直接应用于自动驾驶车辆的路径规划和安全评估,提升车辆在复杂交通环境中的决策能力。未来,结合多模态感知(如激光雷达、地图信息)可进一步增强模型的场景理解能力。长远来看,DA-WAM有望推动智能交通系统的安全升级,实现自动驾驶的普及与商业化,减少交通事故,提高交通效率。

局限与展望

模型在极端或未见过的交通环境中泛化能力仍有限,特别是在极端天气或复杂交叉口。实时推理成本较高,候选轨迹数量增加时可能影响部署效率。硬负样本策略虽增强安全判别,但在某些边界场景仍可能误判,需结合更多多模态信息和规则进行补充。未来需优化模型结构和训练策略,以提升泛化能力和推理速度。

通俗解读 非专业人士也能看懂

想象一下你在厨房里准备一顿大餐。每次你决定放什么菜、用多大火、烹饪多长时间,都会影响最终的味道和效果。现在,假设你有一个聪明的厨师助手,它可以帮你预测每个选择可能带来的结果,比如菜会不会烧焦、味道会不会变得太咸。这个助手不仅能告诉你每个动作的后果,还能帮你选择最合适的做法,以确保菜肴完美。这就像DA-WAM在自动驾驶中的作用,它会根据每个可能的驾驶动作,预测未来的场景变化,并用这些预测结果来决定最安全、最合理的路径。它不断学习、调整,就像厨师不断试验新菜谱,最终让你的“交通厨房”变得更安全、更高效。这个系统的核心思想是:每个动作都对应一个独特的未来场景预测,而不是用一个通用的预测来判断所有路径。这样,车辆就能像经验丰富的司机一样,提前知道每个选择的后果,避免危险,顺利到达目的地。

简单解释 像给14岁少年讲一样

想象你在玩一款赛车游戏,你可以选择不同的路线,每条路线都可能带来不同的结果。有的路线可能快但危险,有的路线慢但安全。现在,假设你有一个超级助手,它可以提前告诉你每条路线未来可能发生的事情,比如会不会撞到障碍物、偏离轨道。这个助手会根据你的每个选择,预测出不同的未来场景,然后帮你挑出最安全、最稳妥的路线。这个助手不断学习,随着你玩得越多,它变得越聪明,能更准确地预测结果。它的秘密在于:每个选择都对应一个专属的未来预测,而不是用一个统一的预测来判断所有路线。这样,你就能像一个经验丰富的司机一样,提前知道每个动作的后果,避免危险,顺利到达终点。这就是DA-WAM的核心思想——让车辆的每个动作都拥有自己的未来预演,从而做出最安全的决定。

原文摘要

Anticipating how scenes evolve under ego actions is fundamental to safe autonomous driving, yet the full potential of world models for decision-making remains unrealized. The critical challenge lies in ensuring that future modeling is not merely predictive, but decision-informative: the predicted future must directly shape which trajectory is selected. Existing approaches decouple future representation learning from planning optimization, or share predicted states across trajectory candidates, thereby diluting the action-specific consequences that ought to guide selection. To bridge this gap, we propose DA-WAM, a framework that unifies predictive representation learning, action-conditioned future modeling, and trajectory scoring under a single decision-making objective. DA-WAM maintains predictive supervision throughout planner optimization via an online encoder and a stable momentum target, allowing future representations to co-evolve with the driving task. An action-conditioned predictor generates a distinct future latent state per trajectory candidate, which is then evaluated by a future-latent-conditioned factorized scorer. For the expert-matched trajectory, the predicted future latent is supervised by the observed future representation, while safety-critical hard negatives provide additional supervision near planning boundaries. Extensive experiments on NAVSIM-v1 and NAVSIM-v2 demonstrate state-of-the-art performance, while ablations and diagnostic analyses validate the key components.

cs.RO cs.AI

参考文献 (20)

ReCogDrive: A Reinforced Cognitive Framework for End-to-End Autonomous Driving

Yongkang Li, Kaixin Xiong, Xiangyu Guo 等

2025 163 引用 查看解读 →

AutoVLA: A Vision-Language-Action Model for End-to-End Autonomous Driving with Adaptive Reasoning and Reinforcement Fine-Tuning

Zewei Zhou, Tianhui Cai, Seth Z. Zhao 等

2025 245 引用 查看解读 →

DriveWorld: 4D Pre-Trained Scene Understanding via World Models for Autonomous Driving

Chen Min, Dawei Zhao, Liang Xiao 等

2024 114 引用 查看解读 →

Enhancing End-to-End Autonomous Driving with Latent World Model

Yingyan Li, Lue Fan, Jiawei He 等

2024 166 引用 查看解读 →

IDOL: Inverse-Dynamics-Guided Future Prediction for End-to-End Autonomous Driving

Chenghao Zhang, Timin Li, Dongmei Li

2026 1 引用 查看解读 →

NAVSIM: Data-Driven Non-Reactive Autonomous Vehicle Simulation and Benchmarking

Daniel Dauner, Marcel Hallgarten, Tianyu Li 等

2024 379 引用 查看解读 →

PRIX: Learning to Plan From Raw Pixels for End-to-End Autonomous Driving

Maciej K. Wozniak, Lian Liu, Yixi Cai 等

2025 14 引用 查看解读 →

V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planning

Mahmoud Assran, Adrien Bardes, David Fan 等

2025 620 引用 查看解读 →

DriveX: Omni Scene Modeling for Learning Generalizable World Knowledge in Autonomous Driving

Chen Shi, Shaoshuai Shi, Kehua Sheng 等

2025 21 引用 查看解读 →

Driving on Registers

Ellington Kirby, A. Boulch, Yihong Xu 等

2026 26 引用 查看解读 →

World4Drive: End-to-End Autonomous Driving via Intention-Aware Physical Latent World Model

Yupeng Zheng, Pengxuan Yang, Zebin Xing 等

2025 104 引用 查看解读 →

ARTEMIS: Autoregressive End-to-End Trajectory Planning With Mixture of Experts for Autonomous Driving

Renju Feng, Ning Xi, Duanfeng Chu 等

2025 67 引用 查看解读 →

Driving Into the Future: Multiview Visual Forecasting and Planning with World Model for Autonomous Driving

Yu-Quan Wang, Jiawei He, Lue Fan 等

2023 377 引用 查看解读 →

DrivingGPT: Unifying Driving World Modeling and Planning with Multi-Modal Autoregressive Transformers

Yuntao Chen, Yu-Quan Wang, Zhaoxiang Zhang

2024 109 引用 查看解读 →

End-to-End Driving with Online Trajectory Evaluation via BEV World Model

Yingyan Li, Yu-Quan Wang, Yang Liu 等

2025 157 引用 查看解读 →

VADv2: End-to-End Vectorized Autonomous Driving via Probabilistic Planning

Shaoyu Chen, Bo Jiang, Hao Gao 等

2024 316 引用 查看解读 →

Generalized Trajectory Scoring for End-to-end Multimodal Planning

Zhenxin Li, Wenhao Yao, Zi Wang 等

2025 53 引用 查看解读 →

Planning-oriented Autonomous Driving

Yi Hu, Jiazhi Yang, Li Chen 等

2022 1543 引用 查看解读 →

iPad: Iterative Proposal-centric End-to-End Autonomous Driving

Ke Guo, Haochen Liu, Xiaojun Wu 等

2025 37 引用 查看解读 →

TransFuser: Imitation With Transformer-Based Sensor Fusion for Autonomous Driving

Kashyap Chitta, Aditya Prakash, Bernhard Jaeger 等

2022 804 引用 查看解读 →