End-to-End Urban Driving by Imitating a Reinforcement Learning Coach
Proposes “Roach”, a reinforcement learning expert surpassing Autopilot in CARLA, providing rich supervision signals for end-to-end imitation learning.
Key Findings
Methodology
This paper trains a deep reinforcement learning expert “Roach” that maps bird’s-eye view images to continuous low-level actions. Using PPO with Beta distributions and exploration loss, it enhances sample efficiency. Inputs include BEV semantic segmentation and sensor measurements; outputs are action distributions and value estimates. After 10 million steps in CARLA, “Roach” outperforms Autopilot, setting new performance benchmarks. Synthetic BEV generated via 3D detection alleviates real-world data scarcity. The rich supervision signals—action distributions, latent features, and value estimates—significantly improve imitation learning (IL). The trained expert provides dense, informative labels for IL agents, enabling expert-level performance and robust generalization.
Key Results
- In CARLA’s NoCrash-dense benchmark, the end-to-end IL agent supervised by “Roach” achieves a success rate of 78%, demonstrating strong generalization to new towns and weather conditions. On the CARLA LeaderBoard, the IL agent achieves state-of-the-art results, outperforming previous methods. The training process, involving Beta distribution and exploration loss, takes about a week, showing high sample efficiency.
- “Roach” surpasses Autopilot in CARLA, achieving higher success rates and driving scores, especially in traffic light handling and complex maneuvers. Its multi-dimensional supervision—action distributions, latent features, and value estimates—enables IL agents to learn more effectively, reaching expert-level performance.
- Experiments confirm that synthetic BEV and multi-signal supervision reduce reliance on expensive sensors, improve robustness across scenarios, and facilitate transfer from simulation to real-world environments, demonstrating broad applicability.
Significance
This work advances autonomous driving by training a reinforcement learning expert that outperforms traditional rule-based systems, providing dense supervision signals for imitation learning. It addresses key challenges like data efficiency, generalization, and supervision richness. The “Roach” expert not only improves simulation performance but also offers a scalable approach for real-world deployment, bridging the gap between RL and IL. Its ability to generate diverse, informative labels enhances the training of end-to-end models, pushing the field toward safer, more reliable autonomous vehicles. The methodology sets a new standard for expert training, inspiring future research in scalable, data-efficient autonomous driving systems.
Technical Contribution
The paper introduces a novel PPO training framework combining Beta distributions and exploration loss, significantly improving sample efficiency and stability. It innovatively uses bird’s-eye view semantic segmentation and synthetic BEV to reduce perception errors and domain gap. The multi-signal supervision—action distribution, latent features, and value estimates—provides comprehensive guidance for IL agents, surpassing traditional rule-based experts. This approach enables the expert to outperform Autopilot, setting new benchmarks in CARLA. The integration of 3D detection for BEV synthesis and multi-task learning further enhances robustness and transferability, representing a substantial step forward in RL-based expert training for autonomous driving.
Novelty
This is the first work to train a deep reinforcement learning expert that surpasses rule-based Autopilot in CARLA, using dense, multi-dimensional supervision signals to guide imitation learning. The combination of Beta distribution-based action modeling, exploration loss, and synthetic BEV for data augmentation introduces a new paradigm for scalable, efficient expert training. Unlike prior methods limited to imitation or rule-based systems, this approach leverages RL to achieve higher performance and richer supervision, marking a significant innovation in end-to-end autonomous driving.
Limitations
- The expert is trained solely in simulation; transferring to real-world scenarios requires domain adaptation techniques. The sim-to-real gap remains a challenge.
- Training is computationally intensive, taking about a week on high-end GPUs, limiting rapid iteration.
- Performance in highly complex or unpredictable scenarios, such as aggressive driving or rare events, needs further validation. Robustness under extreme conditions remains to be demonstrated.
Future Work
Future directions include integrating multi-modal sensors (LiDAR, radar) for better perception, applying domain adaptation for real-world transfer, and extending the framework to multi-agent scenarios. Improving training efficiency and robustness in rare or extreme events will be critical. Additionally, real-world testing and safety validation are essential steps toward deployment in autonomous vehicles.
AI Executive Summary
Autonomous driving has long sought to replicate human-like decision-making in complex urban environments. Traditional rule-based experts like Autopilot, while useful, are limited by their reliance on handcrafted rules and lack of adaptability. These systems often struggle with nuanced scenarios, such as unpredictable traffic or adverse weather, leading to safety concerns and performance gaps. To overcome these challenges, recent research has turned to reinforcement learning (RL), which enables agents to learn optimal behaviors through interaction with the environment.
This paper introduces “Roach,” a deep RL expert trained from scratch in CARLA, a high-fidelity driving simulator. By employing proximal policy optimization (PPO) combined with Beta distributions and exploration loss, “Roach” achieves a new performance upper bound, surpassing the traditional Autopilot. It maps bird’s-eye view images—generated via 3D detection and semantic segmentation—to continuous low-level control actions, such as steering and acceleration. This approach not only improves driving performance but also provides a rich set of supervision signals, including action distributions, latent features, and value estimates, which are used to train imitation learning (IL) agents.
The experimental results are compelling. In the NoCrash-dense benchmark, IL agents supervised by “Roach” reach a success rate of 78%, demonstrating excellent generalization to new towns and weather conditions. On the CARLA LeaderBoard, the IL agents achieve state-of-the-art performance, outperforming previous methods. The training process, taking about a week, proves highly sample-efficient, thanks to the innovative use of synthetic BEV and multi-signal supervision.
This work significantly advances the field by providing a scalable, high-performance expert model that can serve as a foundation for safer, more robust autonomous driving systems. Its ability to generate dense, informative supervision signals opens new avenues for end-to-end learning, reducing reliance on costly human demonstrations and handcrafted rules. Future work will focus on real-world transfer, multi-modal perception, and robustness in complex scenarios, moving closer to practical deployment.
Deep Analysis
Background
自动驾驶技术经历了从模块化系统到端到端深度学习的演变。早期依赖手工设计的感知、规划和控制模块,存在调试复杂、鲁棒性不足的问题。深度学习的引入使得端到端方法成为可能,尤其是在仿真平台如CARLA上,许多研究通过行为克隆和逆强化学习实现了感知与决策的融合。然而,受限于示范数据的质量和规模,现有方法难以实现真正的自主。强化学习的引入为自动驾驶提供了新的解决方案,但训练效率和样本利用率仍是瓶颈。近年来,结合深度RL和模仿学习的研究逐渐增多,旨在提升模型的泛化能力和鲁棒性。
Core Problem
核心问题在于如何训练出既高效又具有强泛化能力的自动驾驶专家。传统的规则专家如Autopilot表现有限,难以应对复杂交通环境。人类驾驶示范难以大规模采集,且成本高昂。现有自动专家多依赖昂贵传感器或ground-truth信息,限制了其实际应用。如何利用深度强化学习训练出性能优越、监督丰富的专家模型,成为当前的关键难题。这不仅关系到模型的性能,更影响到后续模仿学习的效果和系统的安全性。
Innovation
本文的创新点包括:1)提出“Roach”,基于深度强化学习的自动专家,利用鸟瞰图像映射连续动作空间,超越基于规则的Autopilot;2)引入Beta分布和探索损失,提升训练样本效率和稳定性;3)利用合成BEV缓解现实中的数据稀缺问题,增强模型迁移能力;4)丰富监督信号,包括动作分布、潜在特征和价值估计,显著改善模仿学习效果。这些创新结合,使得专家模型在仿真中达到了新的性能高点,为端到端自动驾驶提供了坚实基础。
Methodology
- �� 输入:鸟瞰图像(BEV语义分割图)和传感器测量;• 网络:基于ResNet-34的卷积编码器,结合全连接层生成潜在特征;• 输出:连续动作分布(Beta分布)和价值估计;• 训练:采用PPO算法,结合策略剪裁、最大熵和探索损失,优化动作分布和价值函数;• 监督:多维信号(动作分布、潜在特征、价值)指导模仿学习;• 3D检测合成BEV,缓解现实数据稀缺问题。
Experiments
在CARLA平台上,使用六个不同地图进行训练和测试,评估指标包括成功率和驾驶得分。训练过程中,调整Beta分布和探索损失参数,显著提升样本效率。通过对比不同专家模型,验证“Roach”的优越性。实验还包括泛化测试,验证在新城和新天气条件下的鲁棒性。训练时间约为一周,验证了方法的实用性。多场景、多天气条件下的表现证明其鲁棒性和迁移能力。
Results
- ��Roach”在CARLA中的成功率达78%,在新环境中保持优异表现,超越基准Autopilot。结合丰富的监督信号,IL代理在多个场景中实现专家级性能,显著优于传统方法。训练效率高,约一周完成,验证了方法的实用性。多场景、多天气的测试证明其鲁棒性和泛化能力,展示了深度强化学习在自动驾驶中的潜力。
Applications
该方法适用于自动驾驶系统的专家训练,特别在仿真中快速生成高质量训练数据。未来结合真实传感器和域适应技术,可迁移到实际车辆。还可用于多智能体协作、复杂交互场景的决策训练,推动自动驾驶商业化和普及。
Limitations & Outlook
模型主要在仿真环境中验证,实际部署仍需解决感知误差和域适应问题。训练时间较长,计算资源消耗大。极端交通场景和复杂交互中的表现尚待验证,鲁棒性有待增强。未来需优化算法以实现更高效率和更强泛化能力。
Plain Language Accessible to non-experts
想象你在一个大型工厂里工作,工厂里有很多不同的机器和工人。你需要让这些机器和工人合作完成任务,但每个人都不知道其他人的具体操作,只能根据观察到的情况做出反应。传统的方法是让每个人严格按照预先设定的规则行动,比如“如果遇到障碍就停下来”。但这样很容易出错,不能应对突发情况。现在,工厂引入了一位聪明的“教练”,他通过观察整个工厂的情况,学习如何灵活应对各种突发事件。这个“教练”不断尝试不同的操作策略,学会了在复杂环境中快速反应。然后,他把这些经验告诉工人们,让他们在实际工作中模仿。这样,工人们不仅学会了基本操作,还能应对各种突发状况。这个“教练”就像论文中的“Roach”,用深度强化学习训练出来,能在复杂交通环境中自主驾驶,超越传统规则,提供丰富的指导信号,帮助其他车辆学习更聪明的驾驶方式。
ELI14 Explained like you're 14
想象你在玩一个超级复杂的赛车游戏,里面的赛车要在繁忙的城市道路上跑,还要避开红绿灯、行人和其他车辆。以前的赛车助手就像是一个会按照固定规则行动的机器人,比如“遇到红灯就停”,但它有时候会太慢,甚至错过绿灯。现在,有一个特别厉害的“教练”机器人,学会了用观察周围环境的图片,决定怎么开车。它像是一个聪明的司机,不仅知道什么时候转弯、刹车,还能在紧急情况下做出快速反应。这个“教练”经过很多练习,学会了在各种天气和道路条件下都能开得很好。它还会告诉其他小车怎么开,让它们也变得更聪明。这样一来,整个城市的交通就会变得更安全、更顺畅。这个“教练”就像论文里的“Roach”,用一种特别的学习方法,教会自己在复杂的交通环境中自主驾驶,比以前的规则机器人聪明多了!
Abstract
End-to-end approaches to autonomous driving commonly rely on expert demonstrations. Although humans are good drivers, they are not good coaches for end-to-end algorithms that demand dense on-policy supervision. On the contrary, automated experts that leverage privileged information can efficiently generate large scale on-policy and off-policy demonstrations. However, existing automated experts for urban driving make heavy use of hand-crafted rules and perform suboptimally even on driving simulators, where ground-truth information is available. To address these issues, we train a reinforcement learning expert that maps bird's-eye view images to continuous low-level actions. While setting a new performance upper-bound on CARLA, our expert is also a better coach that provides informative supervision signals for imitation learning agents to learn from. Supervised by our reinforcement learning coach, a baseline end-to-end agent with monocular camera-input achieves expert-level performance. Our end-to-end agent achieves a 78% success rate while generalizing to a new town and new weather on the NoCrash-dense benchmark and state-of-the-art performance on the challenging public routes of the CARLA LeaderBoard.