arXiv:2410.15607cs.ROcs.AI2024-10

融合强化与模仿学习,实现城市自动驾驶多步轨迹规划

Reinforced Imitative Trajectory Planning for Urban Automated Driving

  • 结合强化学习与监督学习,提升训练收敛性
  • 在nuPlan数据集上闭环评估,性能超越基线方法
  • 采用Transformer结构的贝叶斯奖励函数,增强可解释性

强化学习(RL)在城市自动驾驶轨迹规划中面临收敛性差和奖励函数设计难的问题,导致现有基于RL的方法性能难以媲美模仿学习方法。尽管通过结合监督学习缓解了收敛问题,但多数方法仅能单步推理,缺乏多步规划能力。此外,虽逆强化学习有望解决奖励函数设计难题,但现有方法对奖励函数强加线性假设,难以适用于复杂城市场景。为此,本文提出一种新型基于强化学习的轨迹规划方法,融合强化学习与模仿学习以实现多步规划;设计基于Transformer的贝叶斯奖励函数,为城市场景下的RL提供有效奖励信号;并提出混合驱动框架,提升安全性与可解释性。所提方法在大规模真实世界城市自动驾驶数据集nuPlan上验证,采用闭环指标评估,结果表明该方法显著优于使用相同策略模型结构的基线,并达到与当前最先进方法相当的性能。代码已公开于https://github.com/Zigned/nuplan_zigned。

原文摘要 · Abstract (English)

Reinforcement learning (RL) faces challenges in trajectory planning for urban automated driving due to the poor convergence of RL and the difficulty in designing reward functions. Consequently, few RL-based trajectory planning methods can achieve performance comparable to that of imitation learning-based methods. The convergence problem is alleviated by combining RL with supervised learning. However, most existing approaches only reason one step ahead and lack the capability to plan for multiple future steps. Besides, although inverse reinforcement learning holds promise for solving the reward function design issue, existing methods for automated driving impose a linear structure assumption on reward functions, making them difficult to apply to urban automated driving. In light of these challenges, this paper proposes a novel RL-based trajectory planning method that integrates RL with imitation learning to enable multi-step planning. Furthermore, a transformer-based Bayesian reward function is developed, providing effective reward signals for RL in urban scenarios. Moreover, a hybrid-driven trajectory planning framework is proposed to enhance safety and interpretability. The proposed methods were validated on the large-scale real-world urban automated driving nuPlan dataset. Evaluated using closed-loop metrics, the results demonstrated that the proposed method significantly outperformed the baseline employing the identical policy model structure and achieved competitive performance compared to the state-of-the-art method. The code is available at https://github.com/Zigned/nuplan_zigned.

自动驾驶强化学习轨迹规划Transformer

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