arXiv:2412.01234cs.RO2024-12被引 3

将决策与轨迹规划融合进可微优化,实现自动驾驶端到端训练。

Integrating Decision-Making Into Differentiable Optimization Guided Learning for End-to-End Planning of Autonomous Vehicles

  • 把决策和路径规划建模为可微非线性优化问题,兼容学习模块。
  • 在Waymo数据集上测试,安全、效率、舒适性均优于基线方法。
  • 适合研究自动驾驶端到端系统或可微优化应用的开发者。

本文针对聚焦运动预测、决策与轨迹规划的端到端规划框架,提出将决策与轨迹规划建模为可微非线性优化问题,确保与基于学习的模块兼容,构建端到端可训练架构。该优化引入安全、行驶效率和乘坐舒适性等显式目标,指导学习过程。决策任务的内在约束被整合至优化公式并贯穿学习全过程。通过将可微优化器与神经网络预测器结合,所提框架实现端到端训练,使各类驾驶任务与优化目标定义的最终性能对齐。框架在Waymo Open Motion数据集上训练与验证,开环测试显示,尽管规划结果不总与专家轨迹一致,但始终在安全性、行驶效率和乘坐舒适性上优于基线方法。闭环测试进一步证明了决策优化的有效性及驾驶性能的提升。消融实验表明,基于学习的预测模块提供的初始化对优化器收敛及整体驾驶表现至关重要。

原文摘要 · Abstract (English)

We address the decision-making capability within an end-to-end planning framework that focuses on motion prediction, decision-making, and trajectory planning. Specifically, we formulate decision-making and trajectory planning as a differentiable nonlinear optimization problem, which ensures compatibility with learning-based modules to establish an end-to-end trainable architecture. This optimization introduces explicit objectives related to safety, traveling efficiency, and riding comfort, guiding the learning process in our proposed pipeline. Intrinsic constraints resulting from the decision-making task are integrated into the optimization formulation and preserved throughout the learning process. By integrating the differentiable optimizer with a neural network predictor, the proposed framework is end-to-end trainable, aligning various driving tasks with ultimate performance goals defined by the optimization objectives. The proposed framework is trained and validated using the Waymo Open Motion dataset. The open-loop testing reveals that while the planning outcomes using our method do not always resemble the expert trajectory, they consistently outperform baseline approaches with improved safety, traveling efficiency, and riding comfort. The closed-loop testing further demonstrates the effectiveness of optimizing decisions and improving driving performance. Ablation studies demonstrate that the initialization provided by the learning-based prediction module is essential for the convergence of the optimizer as well as the overall driving performance.

自动驾驶端到端可微优化

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