arXiv:2410.05582cs.RO2024-10ICRA被引 57

用生成-评估框架提升自动驾驶决策,效果优于传统方法。

Gen-Drive: Enhancing Diffusion Generative Driving Policies with Reward Modeling and Reinforcement Learning Fine-tuning

  • 先生成多种未来场景,再用奖励模型评估选择最优路径。
  • 在nuPlan数据集上,生成-评估策略显著优于其他学习方法。
  • 用AI辅助标注偏好数据训练奖励模型,减少人工成本。

自动驾驶需要推理交通参与者间的未来交互并做出合理判断。本文提出Gen-Drive框架,将传统预测与确定性规划范式转变为生成-评估规划模式。该框架使用行为扩散模型作为场景生成器,生成多样化的未来可能情景,增强联合交互推理能力。为支持决策,我们提出一个场景评估(奖励)模型,通过VLM辅助收集的成对偏好数据进行训练,降低人工工作量并提升可扩展性。此外,采用强化学习微调框架提升扩散模型生成质量,使其更适用于规划任务。我们在nuPlan数据集上进行训练和闭环规划测试,结果表明,采用生成-评估策略优于其他基于学习的方法。同时,经过微调的生成式驾驶策略在规划性能上显著提升。进一步实验显示,使用所学奖励模型进行评估或强化学习微调,相比依赖人工设计奖励,能带来更好的规划表现。

原文摘要 · Abstract (English)

Autonomous driving necessitates the ability to reason about future interactions between traffic agents and to make informed evaluations for planning. This paper introduces the \textit{Gen-Drive} framework, which shifts from the traditional prediction and deterministic planning framework to a generation-then-evaluation planning paradigm. The framework employs a behavior diffusion model as a scene generator to produce diverse possible future scenarios, thereby enhancing the capability for joint interaction reasoning. To facilitate decision-making, we propose a scene evaluator (reward) model, trained with pairwise preference data collected through VLM assistance, thereby reducing human workload and enhancing scalability. Furthermore, we utilize an RL fine-tuning framework to improve the generation quality of the diffusion model, rendering it more effective for planning tasks. We conduct training and closed-loop planning tests on the nuPlan dataset, and the results demonstrate that employing such a generation-then-evaluation strategy outperforms other learning-based approaches. Additionally, the fine-tuned generative driving policy shows significant enhancements in planning performance. We further demonstrate that utilizing our learned reward model for evaluation or RL fine-tuning leads to better planning performance compared to relying on human-designed rewards. Project website: https://mczhi.github.io/GenDrive.

自动驾驶扩散模型强化学习生成评估

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