提出新仿真器与评估基准,揭示模仿学习的复制陷阱并融合强化学习改进泛化能力
Exposing the Copycat Problem of Imitation-based Planner: A Novel Closed-Loop Simulator, Causal Benchmark and Joint IL-RL Baseline
- 构建闭环仿真环境,支持模仿与强化学习联合训练
- 基于Waymo数据集设计因果评估基准,验证策略是否真懂驾驶原理
- 提出IL+RL融合框架,显著提升罕见场景下的泛化表现
基于机器学习的规划器近年来受到广泛关注,相比传统优化算法具有参数少、开发快的优势。在机器学习规划中,模仿学习(IL)是主流方法,通过监督轨迹数据直接学习驾驶策略。尽管在开环基准上表现良好,但其是否真正理解基本驾驶原则仍存疑,可能仅依赖初始状态进行外推。现有研究虽已识别此问题,但评估常使用原始数据集,其中未来轨迹高度依赖初始条件,且模仿学习易过拟合常见场景,难以泛化至罕见或未见情形。为此,本文提出:1)支持模仿与强化学习的新型闭环仿真器;2)基于Waymo Open Dataset构建的因果评估基准,用于严格检验复制问题的影响;3)一种融合模仿学习与强化学习的新框架,以克服纯模仿方法的局限性。相关代码将很快公开。
原文摘要 · Abstract (English)
Machine learning (ML)-based planners have recently gained significant attention. They offer advantages over traditional optimization-based planning algorithms. These advantages include fewer manually selected parameters and faster development. Within ML-based planning, imitation learning (IL) is a common algorithm. It primarily learns driving policies directly from supervised trajectory data. While IL has demonstrated strong performance on many open-loop benchmarks, it remains challenging to determine if the learned policy truly understands fundamental driving principles, rather than simply extrapolating from the ego-vehicle's initial state. Several studies have identified this limitation and proposed algorithms to address it. However, these methods often use original datasets for evaluation. In these datasets, future trajectories are heavily dependent on initial conditions. Furthermore, IL often overfits to the most common scenarios. It struggles to generalize to rare or unseen situations. To address these challenges, this work proposes: 1) a novel closed-loop simulator supporting both imitation and reinforcement learning, 2) a causal benchmark derived from the Waymo Open Dataset to rigorously assess the impact of the copycat problem, and 3) a novel framework integrating imitation learning and reinforcement learning to overcome the limitations of purely imitative approaches. The code for this work will be released soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。