arXiv:2409.15730cs.ROcs.AI2024-09ICRA被引 12

用混合分布建模自动驾驶决策,提升不确定性处理能力。

Learning Multiple Probabilistic Decisions from Latent World Model in Autonomous Driving

  • 将环境状态与车辆动作建模为混合分布,捕捉决策随机性
  • 在Waymax闭环基准上达到专家级表现,超越主流RL/IL方法
  • 适合关注自动驾驶不确定性建模与生成式规划的研究者

自回归世界模型在向量场景理解中表现出强泛化能力,但在生成动作时因不确定性建模不足和自我欺骗问题而受限。本文提出LatentDriver框架,将环境未来状态与本车可能动作建模为混合分布,并从中推导确定性控制信号。通过引入混合建模,有效捕捉决策的随机性;同时,通过向世界模型提供从分布中采样的中间动作,缓解了自我欺骗问题。在最新发布的闭环基准Waymax上的实验结果表明,LatentDriver超越了当前最先进的强化学习与模仿学习方法,达到了专家级性能。代码与模型将开源于https://github.com/Sephirex-X/LatentDriver。

原文摘要 · Abstract (English)

The autoregressive world model exhibits robust generalization capabilities in vectorized scene understanding but encounters difficulties in deriving actions due to insufficient uncertainty modeling and self-delusion. In this paper, we explore the feasibility of deriving decisions from an autoregressive world model by addressing these challenges through the formulation of multiple probabilistic hypotheses. We propose LatentDriver, a framework models the environment's next states and the ego vehicle's possible actions as a mixture distribution, from which a deterministic control signal is then derived. By incorporating mixture modeling, the stochastic nature of decisionmaking is captured. Additionally, the self-delusion problem is mitigated by providing intermediate actions sampled from a distribution to the world model. Experimental results on the recently released close-loop benchmark Waymax demonstrate that LatentDriver surpasses state-of-the-art reinforcement learning and imitation learning methods, achieving expert-level performance. The code and models will be made available at https://github.com/Sephirex-X/LatentDriver.

自动驾驶生成模型不确定性建模强化学习

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