自动课程学习让自动驾驶模型更高效、更鲁棒地训练。
Automatic Curriculum Learning for Driving Scenarios: Towards Robust and Efficient Reinforcement Learning
- 根据智能体能力动态生成有挑战性的驾驶场景。
- 高密度交通下成功率提升21%,收敛速度更快。
- 无需人工设计课程,适合大规模自动驾驶训练。
本文针对端到端自动驾驶智能体在强化学习(RL)中训练的挑战提出解决方案。传统方法在固定场景和默认交通行为下训练,限制了泛化能力与实际部署。虽领域随机化可增加多样性,但因场景间方差大导致训练低效、策略次优。为此,我们提出一种自动课程学习框架,基于智能体当前策略的能力动态生成并演化驾驶场景。框架引入‘教师’机制,依据智能体的学习潜力(代理中心度量)自动选择与变异场景,避免已掌握或过难场景,提升训练效率。在从摄像头图像学习驾驶策略的强化学习设置中,对比固定场景训练与领域随机化基线,本方法显著提升泛化能力:低密度交通下成功率提升9%,高密度交通下提升21%,且以更少训练步数实现更快收敛。结果表明,自动课程学习能有效增强基于强化学习的自动驾驶智能体的鲁棒性与训练效率。
原文摘要 · Abstract (English)
This paper addresses the challenges of training end-to-end autonomous driving agents using Reinforcement Learning (RL). RL agents are typically trained in a fixed set of scenarios and nominal behavior of surrounding road users in simulations, limiting their generalization and real-life deployment. While domain randomization offers a potential solution by randomly sampling driving scenarios, it frequently results in inefficient training and sub-optimal policies due to the high variance among training scenarios. To address these limitations, we propose an automatic curriculum learning framework that dynamically generates driving scenarios with adaptive complexity based on the agent's evolving capabilities. Unlike manually designed curricula that introduce expert bias and lack scalability, our framework incorporates a ``teacher'' that automatically generates and mutates driving scenarios based on their learning potential -- an agent-centric metric derived from the agent's current policy -- eliminating the need for expert design. The framework enhances training efficiency by excluding scenarios the agent has mastered or finds too challenging. We evaluate our framework in a reinforcement learning setting where the agent learns a driving policy from camera images. Comparative results against baseline methods, including fixed scenario training and domain randomization, demonstrate that our approach leads to enhanced generalization, achieving higher success rates: +9% in low traffic density, +21% in high traffic density, and faster convergence with fewer training steps. Our findings highlight the potential of ACL in improving the robustness and efficiency of RL-based autonomous driving agents.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。