arXiv:2501.16733cs.ROcs.CV2025-01被引 17

用个体化世界模型预测车辆意图,让自动驾驶更安全高效。

Dream to Drive with Predictive Individual World Model

  • 构建个体级世界模型,通过轨迹预测捕捉车辆交互与意图
  • 在模拟环境中实现最优安全与效率表现,优于主流方法
  • 适合研究意图感知的自动驾驶决策系统开发者

在复杂城市交通中,由于道路使用者意图未知,实现反应式驾驶仍具挑战。基于模型的强化学习(MBRL)通过构建世界模型提供信息性状态和想象训练,具有巨大潜力。然而现有研究受限于场景级重建表征学习,常忽略关键交互车辆,难以建模车辆间交互特征及长期意图。为此,本文提出一种新型MBRL方法——预测个体世界模型(PIWM),从个体层面描述驾驶环境,通过轨迹预测任务捕捉车辆间的交互关系与意图。同时,行为策略与PIWM联合训练,在PIWM的想象空间中有效导航,利用意图感知的隐状态应对复杂城市场景。方法在基于真实世界复杂交互场景构建的仿真环境中进行训练与评估。实验结果表明,相比主流无模型及先进模型基强化学习方法,该方法在安全性与效率上均取得最佳表现。

原文摘要 · Abstract (English)

It is still a challenging topic to make reactive driving behaviors in complex urban environments as road users' intentions are unknown. Model-based reinforcement learning (MBRL) offers great potential to learn a reactive policy by constructing a world model that can provide informative states and imagination training. However, a critical limitation in relevant research lies in the scene-level reconstruction representation learning, which may overlook key interactive vehicles and hardly model the interactive features among vehicles and their long-term intentions. Therefore, this paper presents a novel MBRL method with a predictive individual world model (PIWM) for autonomous driving. PIWM describes the driving environment from an individual-level perspective and captures vehicles' interactive relations and their intentions via trajectory prediction task. Meanwhile, a behavior policy is learned jointly with PIWM. It is trained in PIWM's imagination and effectively navigates in the urban driving scenes leveraging intention-aware latent states. The proposed method is trained and evaluated on simulation environments built upon real-world challenging interactive scenarios. Compared with popular model-free and state-of-the-art model-based reinforcement learning methods, experimental results show that the proposed method achieves the best performance in terms of safety and efficiency.

自动驾驶强化学习意图预测世界模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。