arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.07321cs.CVcs.AI

面向端到端自动驾驶的模块级自适应对抗训练

Module-wise Adaptive Adversarial Training for End-to-end Autonomous Driving

School of Computer Science and Engineering, Beihang University(北京航空航天大学计算机科学与工程学院);School of Software, Beihang University(北京航空航天大学软件学院);School of Computing, National University of Singapore(新加坡国立大学计算学院);Aviation Industry Development Research Center of China(中国航空工业发展研究中心)

Tianyuan Zhang, Lu Wang, Jiaqi Kang, Xinwei Zhang, Siyuan Liang, Yuwei Chen, Aishan Liu, Xianglong Liu

AI总结:

针对端到端自动驾驶模型易受对抗攻击的问题,提出模块级自适应对抗训练方法MA2T,通过模块级噪声注入和动态权重累积自适应提升鲁棒性,在nuScenes和CARLA上均取得优于基线的效果。

摘要:

深度学习的最新进展显著提升了自动驾驶(AD)模型的性能,尤其是集成了感知、预测与规划阶段的端到端系统,已达到当前最优性能水平。然而,这类模型仍易受对抗攻击影响,人类无法察觉的微小扰动就会干扰其决策过程。尽管对抗训练是提升模型对这类攻击鲁棒性的有效方法,但此前尚无研究关注其在端到端AD模型中的应用。本文率先开展端到端AD模型的对抗训练研究,提出了一种新颖的模块级自适应对抗训练方法(Module-wise Adaptive Adversarial Training,MA2T)。不过,将传统对抗训练拓展至该场景极具挑战性,因为模型内不同阶段有各自不同的目标且彼此高度关联。为应对这些挑战,MA2T首先引入模块级噪声注入机制,在不同模块的输入端注入噪声,以整体目标而非各模块独立损失为指导训练模型。此外,我们提出动态权重累积自适应机制,引入累积权重变化量,根据各模块的贡献(累积降低率)自适应学习并调整其损失权重,以实现更好的平衡与鲁棒训练。为验证该防御方法的有效性,我们在广泛使用的nuScenes数据集上,针对多个端到端AD模型在白盒与黑盒攻击场景下开展了大量实验,结果表明我们的方法以较大幅度(+5-10%)优于其他基线方法。此外,我们在CARLA仿真环境中通过闭环评估验证了该防御方法的鲁棒性,结果显示其甚至对自然损坏也具备更强的抵御能力。

英文摘要:

Recent advances in deep learning have markedly improved autonomous driving (AD) models, particularly end-to-end systems that integrate perception, prediction, and planning stages, achieving state-of-the-art performance. However, these models remain vulnerable to adversarial attacks, where human-imperceptible perturbations can disrupt decision-making processes. While adversarial training is an effective method for enhancing model robustness against such attacks, no prior studies have focused on its application to end-to-end AD models. In this paper, we take the first step in adversarial training for end-to-end AD models and present a novel Module-wise Adaptive Adversarial Training (MA2T). However, extending conventional adversarial training to this context is highly non-trivial, as different stages within the model have distinct objectives and are strongly interconnected. To address these challenges, MA2T first introduces Module-wise Noise Injection, which injects noise before the input of different modules, targeting training models with the guidance of overall objectives rather than each independent module loss. Additionally, we introduce Dynamic Weight Accumulation Adaptation, which incorporates accumulated weight changes to adaptively learn and adjust the loss weights of each module based on their contributions (accumulated reduction rates) for better balance and robust training. To demonstrate the efficacy of our defense, we conduct extensive experiments on the widely-used nuScenes dataset across several end-to-end AD models under both white-box and black-box attacks, where our method outperforms other baselines by large margins (+5-10%). Moreover, we validate the robustness of our defense through closed-loop evaluation in the CARLA simulation environment, showing improved resilience even against natural corruption.

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

↑