arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04783cs.LGcs.CLcs.CR
收藏

AutoDefense:针对越狱攻击的多智能体 LLM 防御

AutoDefense: Multi-Agent LLM Defense against Jailbreak Attacks

Oregon State University(俄勒冈州立大学);Pennsylvania State University(宾夕法尼亚州立大学);CISPA Helmholtz Center for Information Security(CISPA亥姆霍兹信息安全中心)

Yifan Zeng, Yiran Wu, Xiao Zhang, Huazheng Wang, Qingyun Wu

AI总结:

本文提出 AutoDefense,一个多智能体防御框架,通过任务划分和响应过滤机制,利用小型开源模型协作防御大型模型免受越狱攻击,在维持正常性能的同时显著降低攻击成功率。

摘要:

尽管在道德对齐方面进行了广泛的预训练以防止生成有害信息,大型语言模型(LLMs)仍然容易受到越狱攻击。在本文中,我们提出了 AutoDefense,这是一个多智能体防御框架,用于过滤来自 LLMs 的有害响应。借助响应过滤机制,我们的框架对不同越狱攻击提示具有鲁棒性,并可用于防御不同的受害模型。AutoDefense 为 LLM 智能体分配不同角色,并让它们协作完成防御任务。任务划分增强了 LLMs 的整体指令遵循能力,并允许将其他防御组件作为工具集成。借助 AutoDefense,小型开源 LMs 可以作为智能体,防御更大模型免受越狱攻击。我们的实验表明,AutoDefense 能有效防御不同越狱攻击,同时维持对正常用户请求的性能。例如,我们使用 LLaMA-2-13b 和一个 3 智能体系统,将 GPT-3.5 上的攻击成功率从 55.74% 降至 7.95%。我们的代码和数据已在 https://github.com/XHMY/AutoDefense 公开。

英文摘要:

Despite extensive pre-training in moral alignment to prevent generating harmful information, large language models (LLMs) remain vulnerable to jailbreak attacks. In this paper, we propose AutoDefense, a multi-agent defense framework that filters harmful responses from LLMs. With the response-filtering mechanism, our framework is robust against different jailbreak attack prompts, and can be used to defend different victim models. AutoDefense assigns different roles to LLM agents and employs them to complete the defense task collaboratively. The division in tasks enhances the overall instruction-following of LLMs and enables the integration of other defense components as tools. With AutoDefense, small open-source LMs can serve as agents and defend larger models against jailbreak attacks. Our experiments show that AutoDefense can effectively defense against different jailbreak attacks, while maintaining the performance at normal user request. For example, we reduce the attack success rate on GPT-3.5 from 55.74% to 7.95% using LLaMA-2-13b with a 3-agent system. Our code and data are publicly available at https://github.com/XHMY/AutoDefense.

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

↑