用强化学习闭环训练端到端自动驾驶模型,性能显著提升。
CLEAR: Closed-Loop Reinforcement Learning at Scale for End-to-End Autonomous Driving

- 基于预训练视觉语言动作模型,设计残差路径策略实现闭环学习。
- 在CARLA longest6 v2和Bench2Drive上超越现有方法,达到新基准。
- 通过异构流水线大幅增加并行仿真环境数,保障训练稳定高效。
端到端自动驾驶(E2E-AD)旨在直接将原始传感器信息映射为驾驶动作。随着多模态大语言模型(MLLM)的快速发展,研究者提出了视觉-语言-动作(VLA)模型范式,将视觉感知、语言理解与动作预测统一于单一策略中。然而,现有基于VLA的策略主要采用模仿学习,仅通过优化与专家轨迹的距离指标进行训练,导致开环训练与闭环推理间的分布偏移,影响闭环规划性能。为此,我们提出CLEAR系统,支持大规模端到端自动驾驶的闭环强化学习训练。我们设计了一种在预训练VLA策略的路径先验基础上学习残差路径的新策略,有效利用其知识。同时,为解决视觉策略强化学习面临的数据密集性问题,我们构建异构流水线,将模拟器与VLA学习器部署于不同计算集群,显著增加并行仿真环境数量,避免资源争用并保持训练稳定。实验表明,仅使用简单奖励函数,CLEAR在CARLA longest6 v2和Bench2Drive等挑战性基准上显著优于此前方法,达到新状态水平。
原文摘要 · Abstract (English)
End-to-end autonomous driving (E2E-AD) aims to directly map raw sensor information to driving actions. Recently, with the rapid advancement of multi-modal large language models (MLLMs), researchers have proposed the paradigm of Vision-Language-Action (VLA) models for E2E-AD, where it seeks to integrate visual perception, language understanding and action prediction within a single policy. However, existing VLA-based policies largely adopts imitation learning, where it only learns to drive by optimizing distance-based metrics w.r.t. logged expert trajectories. Such distribution shift between open-loop training and closed-loop inference leads to suboptimal performance in closed-loop planning. To close this gap, we present CLEAR, a system that enables closed-loop training using Reinforcement Learning (RL) at scale for E2E-AD. We propose to learn a novel residual waypoint policy around the waypoint prior from pretrained VLA policies, effectively harnessing the knowledge within. On another front, one of the key challenges to scale up RL for vision-based policies is the number of parallel simulation environments since RL is data hungry. To that end, we design a heterogeneous pipeline that places the simulator and the VLA learner on distinct compute groups, which allows us to dramatically increase the number of simulation environments running in parallel while avoiding resource contention and maintaining training stability. We show that with a simple reward, CLEAR significantly outperforms previous methods and sets new state-of-the-art performance on the challenging benchmarks of CARLA longest6 v2 and Bench2Drive.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。