用视觉语言模型自动生成驾驶奖励,让自动驾驶更安全、泛化更强。
VLM-RL: A Unified Vision Language Models and Reinforcement Learning Framework for Safe Autonomous Driving
- 用正负语言目标生成语义奖励,替代人工设计奖励函数。
- 碰撞率降10.5%,路线完成率提升104.6%,在未见场景中表现稳定。
- 适配主流强化学习算法,适合想避免手动调奖的自动驾驶研究者。
近年来,基于强化学习(RL)的驾驶策略学习在自动驾驶领域备受关注,已在多种驾驶场景中取得显著进展。然而,传统RL方法依赖人工设计奖励,耗费大量人力且泛化能力差。为此,本文提出统一框架VLM-RL,将预训练视觉语言模型(VLMs)与强化学习结合,利用图像观测和自然语言目标自动生成奖励信号。核心是对比语言目标(CLG)作为奖励范式,通过正负语言目标生成语义奖励。进一步引入分层奖励融合机制,结合CLG语义奖励与车辆状态信息,提升奖励稳定性并提供更全面的反馈。此外,采用批处理技术优化训练效率。在CARLA模拟器上的大量实验表明,VLM-RL优于现有基线,碰撞率降低10.5%,路线完成率提高104.6%,并在未见过的驾驶场景中表现出强泛化能力。该框架可无缝集成任意标准强化学习算法,有望革新依赖人工奖励设计的现有范式,并推动性能持续提升。演示视频与代码见:https://zilin-huang.github.io/VLM-RL-website。
原文摘要 · Abstract (English)
In recent years, reinforcement learning (RL)-based methods for learning driving policies have gained increasing attention in the autonomous driving community and have achieved remarkable progress in various driving scenarios. However, traditional RL approaches rely on manually engineered rewards, which require extensive human effort and often lack generalizability. To address these limitations, we propose \textbf{VLM-RL}, a unified framework that integrates pre-trained Vision-Language Models (VLMs) with RL to generate reward signals using image observation and natural language goals. The core of VLM-RL is the contrasting language goal (CLG)-as-reward paradigm, which uses positive and negative language goals to generate semantic rewards. We further introduce a hierarchical reward synthesis approach that combines CLG-based semantic rewards with vehicle state information, improving reward stability and offering a more comprehensive reward signal. Additionally, a batch-processing technique is employed to optimize computational efficiency during training. Extensive experiments in the CARLA simulator demonstrate that VLM-RL outperforms state-of-the-art baselines, achieving a 10.5\% reduction in collision rate, a 104.6\% increase in route completion rate, and robust generalization to unseen driving scenarios. Furthermore, VLM-RL can seamlessly integrate almost any standard RL algorithms, potentially revolutionizing the existing RL paradigm that relies on manual reward engineering and enabling continuous performance improvements. The demo video and code can be accessed at: https://zilin-huang.github.io/VLM-RL-website.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。