用反事实解释增强数据,让自动驾驶模型更安全应对突发状况。
Good Data Is All Imitation Learning Needs
- 通过微调输入生成反事实样本,扩充决策边界附近的数据。
- 在CARLA仿真中驾驶得分达84.2,领先前代方法15.02个百分点。
- 适合研究自动驾驶鲁棒性与数据增强的学者和工程师。
本文针对自动驾驶系统中传统教师-学生模型、模仿学习与行为克隆存在的现实场景覆盖不全问题,提出将反事实解释(CFEs)作为端到端自动驾驶系统的新型数据增强技术。CFEs通过最小输入修改生成靠近决策边界的训练样本,更全面地表征专家驾驶员策略,尤其在安全关键场景中表现优异。该方法显著提升模型对罕见复杂驾驶事件(如行人突然冲出)的预判能力,从而实现更安全、可信的决策。在CARLA仿真环境中的实验表明,CF-Driver模型驾驶得分为84.2,相较当前最优方法提升15.02个百分点,验证了其有效性。为促进后续研究,代码已公开。
原文摘要 · Abstract (English)
In this paper, we address the limitations of traditional teacher-student models, imitation learning, and behaviour cloning in the context of Autonomous/Automated Driving Systems (ADS), where these methods often struggle with incomplete coverage of real-world scenarios. To enhance the robustness of such models, we introduce the use of Counterfactual Explanations (CFEs) as a novel data augmentation technique for end-to-end ADS. CFEs, by generating training samples near decision boundaries through minimal input modifications, lead to a more comprehensive representation of expert driver strategies, particularly in safety-critical scenarios. This approach can therefore help improve the model's ability to handle rare and challenging driving events, such as anticipating darting out pedestrians, ultimately leading to safer and more trustworthy decision-making for ADS. Our experiments in the CARLA simulator demonstrate that CF-Driver outperforms the current state-of-the-art method, achieving a higher driving score and lower infraction rates. Specifically, CF-Driver attains a driving score of 84.2, surpassing the previous best model by 15.02 percentage points. These results highlight the effectiveness of incorporating CFEs in training end-to-end ADS. To foster further research, the CF-Driver code is made publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。