arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2310.12432cs.LG

CAT:面向安全端到端驾驶的闭环对抗训练

CAT: Closed-loop Adversarial Training for Safe End-to-End Driving

Tsinghua University(清华大学);University of California, Los Angeles(加州大学洛杉矶分校);The University of Edinburgh(爱丁堡大学)

Linrui Zhang, Zhenghao Peng, Quanyi Li, Bolei Zhou

AI总结:

本文提出闭环对抗训练框架CAT,通过概率分解重采样技术动态生成安全关键驾驶场景,以更低计算成本提升端到端驾驶智能体的安全性。

摘要:

驾驶安全是自动驾驶车辆的首要任务。与以往在策略层面通过算法设计处理易发事故交通事件的工作不同,本文从环境增强的视角研究了一种用于安全端到端驾驶的闭环对抗训练(CAT)框架。CAT旨在通过在随时间动态生成的安全关键场景上训练智能体,持续提升驾驶智能体的安全性。我们开发了一种新颖的重采样技术,通过概率分解将日志回放的真实世界驾驶场景转化为安全关键场景,其中对抗性交通生成被建模为标准运动预测子问题的乘积。因此,与现有的安全关键场景生成方法相比,CAT能够发动更高效的物理攻击,并在迭代学习流程中产生显著更低的计算成本。我们将CAT整合到MetaDrive模拟器中,并在从真实世界驾驶数据集导入的数百个驾驶场景上验证了我们的方法。实验结果表明,CAT能够有效地生成对抗场景来对抗正在训练的智能体。训练完成后,该智能体在留出测试集的日志回放和安全关键交通场景中均能实现卓越的驾驶安全性。代码和数据可在 https://metadriverse.github.io/cat 获取。

英文摘要:

Driving safety is a top priority for autonomous vehicles. Orthogonal to prior work handling accident-prone traffic events by algorithm designs at the policy level, we investigate a Closed-loop Adversarial Training (CAT) framework for safe end-to-end driving in this paper through the lens of environment augmentation. CAT aims to continuously improve the safety of driving agents by training the agent on safety-critical scenarios that are dynamically generated over time. A novel resampling technique is developed to turn log-replay real-world driving scenarios into safety-critical ones via probabilistic factorization, where the adversarial traffic generation is modeled as the multiplication of standard motion prediction sub-problems. Consequently, CAT can launch more efficient physical attacks compared to existing safety-critical scenario generation methods and yields a significantly less computational cost in the iterative learning pipeline. We incorporate CAT into the MetaDrive simulator and validate our approach on hundreds of driving scenarios imported from real-world driving datasets. Experimental results demonstrate that CAT can effectively generate adversarial scenarios countering the agent being trained. After training, the agent can achieve superior driving safety in both log-replay and safety-critical traffic scenarios on the held-out test set. Code and data are available at https://metadriverse.github.io/cat.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑