用大模型自动发现自动驾驶策略中有效且多样的漏洞
AED: Automatic Discovery of Effective and Diverse Vulnerabilities for Autonomous Driving Policy with Large Language Models
- 用大模型自动生成强化学习奖励函数,替代人工设计
- 并行训练针对多种事故类型的对抗策略,提升漏洞多样性
- 通过偏好学习筛选无效事故,提高漏洞的有效性
评估自动驾驶策略的安全性至关重要,强化学习(RL)已成为发现关键漏洞的有力方法。然而,现有基于RL的方法往往难以同时发现既有效(自动驾驶车辆真正负责事故)又多样(覆盖多种故障类型)的漏洞。为此,我们提出AED框架,利用大语言模型(LLMs)自动发现自动驾驶策略中的有效且多样漏洞。首先,使用LLM自动生成强化学习训练的奖励函数;接着,让LLM考虑多种事故类型,并并行训练针对不同事故类型的对抗策略;最后,采用基于偏好的学习方法过滤无效事故,提升每种漏洞的有效性。在多个模拟交通场景和测试策略上的实验表明,AED发现的漏洞范围更广,攻击成功率更高,相比专家设计的奖励函数,显著减少了人工奖励工程的需求,提升了漏洞发现的多样性和有效性。实现代码见:https://github.com/thu-nics/AED。
原文摘要 · Abstract (English)
Assessing the safety of autonomous driving policy is of great importance, and reinforcement learning (RL) has emerged as a powerful method for discovering critical vulnerabilities in driving policies. However, existing RL-based approaches often struggle to identify vulnerabilities that are both effective-meaning the autonomous vehicle is genuinely responsible for the accidents-and diverse-meaning they span various failure types. To address these challenges, we propose AED, a framework that uses large language models (LLMs) to automatically discover effective and diverse vulnerabilities in autonomous driving policies. We first utilize an LLM to automatically design reward functions for RL training. Then we let the LLM consider a diverse set of accident types and train adversarial policies for different accident types in parallel. Finally, we use preference-based learning to filter ineffective accidents and enhance the effectiveness of each vulnerability. Experiments across multiple simulated traffic scenarios and tested policies show that AED uncovers a broader range of vulnerabilities and achieves higher attack success rates compared with expert-designed rewards, thereby reducing the need for manual reward engineering and improving the diversity and effectiveness of vulnerability discovery. The implementation can be found on: https://github.com/thu-nics/AED .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。