Command Palette
Search for a command to run...
PsySafe:一种基于心理学的多智能体系统安全攻防与评估综合框架
PsySafe:一种基于心理学的多智能体系统安全攻防与评估综合框架
Zaibin Zhang Yongting Zhang Lijun Li Hongzhi Gao Lijun Wang Huchuan Lu Feng Zhao Yu Qiao Jing Shao
摘要
多智能体系统在与大语言模型(LLM)结合后展现出卓越的集体智能能力。然而,这种智能被恶意利用的潜在风险不容忽视。迄今为止,针对多智能体系统安全问题的系统性研究仍然有限。本文从智能体心理学的创新视角出发,揭示了智能体的黑暗心理状态对安全构成重大威胁。为应对上述问题,我们提出了一个基于智能体心理学的综合框架(PsySafe),聚焦三个关键领域:第一,识别智能体中的黑暗人格特质如何导致风险行为;第二,从心理学和行为学视角评估多智能体系统的安全性;第三,制定有效的策略来缓解这些风险。实验揭示了若干引人关注的现象,例如智能体间的集体危险行为、智能体在实施危险行为时的自我反思,以及智能体心理评估与危险行为之间的相关性。我们预期该框架和相关发现将为多智能体系统安全的后续研究提供有价值的参考。相关数据和代码已公开于 https://github.com/AI4Good24/PsySafe 。
一句话总结
上海人工智能实验室、大连理工大学和中国科学技术大学的研究人员提出 PsySafe,这是一种基于 agent 心理学的框架,可识别驱动危险行为的黑暗人格特质,从心理与行为视角评估多智能体系统安全性,并制定缓解策略,揭示集体危险行为以及 agent 在执行危险行为时的自我反思等现象。
核心贡献
-
PsySafe 是面向多智能体 LLM 系统、以心理学为基础的安全框架。它注入黑暗人格特质,并通过对 agent 特质与人工输入的攻击来暴露危险行为,包括对安全查询的危险建议、集体危险倾向以及 agent 自我反思。
-
该工作引入一种安全性评估方法,将黑暗三角人格测验与过程危险率和联合危险率等行为度量结合,并使用安全与危险任务数据集展示心理分数与行为安全之间的关联。
-
论文提出面向多智能体系统的防御机制,包括输入过滤、基于心理的防御和基于角色的防御。实验表明,相比针对黑暗特质注入的输入侧过滤,基于心理和基于角色的防御能更有效地减少危险行为和集体危险涌现。
引言
大语言模型 agent 可以完成复杂任务并模拟社交交互,但其不断增长的自主性也带来滥用风险,例如生成恶意软件或创建欺骗性网站。这一点很重要,因为这类 agent 越来越多地以多智能体系统形式运行,其中角色设定、agent 间对话以及人类与 agent 交互可能放大不安全行为。
先前安全工作主要针对单个 LLM 进行攻击或防御,尤其依赖输入输出检查和输入过滤。它未充分解决多智能体系统。在多智能体系统中,危险行为不仅来自直接提示,还可能通过 agent 角色、集体交互和注入的心理状态出现。
作者利用黑暗人格特质来填补这一缺口,并提出 PsySafe。该框架可发现多智能体安全漏洞,从心理与行为视角评估安全性,并构建防御。具体而言,他们注入黑暗特质并设计针对 agent 角色与人工输入的攻击,使用黑暗三角评估以及过程危险率和联合危险率度量风险,并测试基于输入、基于心理和基于角色的防御。
数据集
-
本文使用精选的安全与危险任务基准评估 agent 系统安全性。完整基础数据集包含 859 个任务,较小的 tiny 数据集包含 125 个任务。
-
基础数据集的安全部分包含 200 个任务:
- 从 AI society 数据集随机选取的 100 个指令任务。
- 从 Camel 的 Code 数据集随机选取的 100 个代码任务。
-
基础数据集的危险部分包含 659 个任务:
- 来源:DoAnythingNow 和 AdvBench。
- 过滤:应用去重工具以提升质量和唯一性。
- 结构:危险任务围绕 13 个安全维度组织。
-
tiny 数据集用于降低访问和实验成本:
- 60 个安全任务,按 30 个指令任务和 30 个代码任务采样。
- 65 个危险任务,从 13 个安全维度中各均匀采样 5 个任务。
-
该数据集用于评估,而非论文所述的训练划分:
- 不同多智能体系统上的主要实验使用完整基础数据集。
- 消融实验使用 tiny 数据集。
方法
PsySafe 流水线被设计为闭环工作流,用于识别、度量和缓解多智能体系统中的安全风险。作者不止评估语言模型的最终响应,而是通过两个互补视角建模安全性:每个 agent 的心理状态,以及多智能体交互中涌现的行为动态。参见框架图。
该工作流将方法组织为攻击、agent 系统、防御和评估组件。心理模块基于六组道德对比:关怀与伤害、公平与欺骗、忠诚与背叛、权威与颠覆、神圣与堕落、自由与压迫。攻击组件构造对抗提示并通过不同接口注入。agent 系统组件将该方法连接到常见多智能体结构,如层级、联合和混合交互拓扑。防御组件引入在输入、提示和角色层级运行的保护机制。评估组件追踪 agent 心理和行为危险。
攻击设计针对 agent 的角色条件化。由于 agent 可被赋予多样化 persona,其对齐可能更灵活,并可能被利用:将负面人格特质放入 agent 的内部配置。作者提出黑暗特质注入,将每个道德维度的有害极置于 agent 的 persona 中。为强化这种注入,他们将黑暗特质与两种高级提示技术结合。诱导指令使 agent 更强烈地遵循被注入的黑暗 persona。Red ICL 使用上下文内示例,将恶意意图隐藏在看似正常的任务表述之后。这些技术共同可污染 agent,使其即使在用户查询看似安全时也产生危险输出。
如下图所示:
示例交互表明,在注入攻击提示后,planner 和 assistant agent 可在危险和安全任务场景中生成恶意指令、欺骗性解决方案或危险步骤。这揭示风险不仅存在于用户查询,也存在于 agent persona 以及管控 agent 行为的 system prompt。该框架因此研究三条注入路径。在人工输入攻击中,恶意提示被插入人工输入接口,并可能在交互中重复。这会污染首个响应的 agent,并可能随着更多攻击轮次注入而加剧。在特质攻击中,黑暗提示被插入选定 agent 的 system prompt,改变其 persona 同时保留其功能角色。混合攻击结合两者,提高绕过输入层级和角色层级保护的可能性。
评估模块将安全性评估从静态输入输出检查扩展到动态心理和行为追踪。心理评估方面,每个 agent 通过增强的 DTDD 评估进行打分。作者观察到这些心理分数与后续交互中的危险行为高度相关,因此心理评估既作为诊断信号,也作为防御的门控信号。
行为评估方面,PsySafe 检查每个交互轮次,并区分过程危险与联合危险。过程危险指至少一个 agent 在该轮次产生危险动作,即使另一个 agent 随后纠正。联合危险指所有 agent 在同一轮次参与危险行为。令 N 表示评估的总交互轮数,令 Nproc 表示包含过程危险的轮数,令 Njoint 表示包含联合危险的轮数。两个行为指标计算如下
PDR=NNproc,JDR=NNjoint.过程危险衡量攻击能多深地渗透多智能体系统,联合危险衡量系统是否集体陷入危险交互模式。
防御流水线将外部过滤与内部矫正结合。在接口处,作者测试 GPT-4 和 Llama Guard 等过滤方法以阻止攻击提示。他们发现,当攻击通过 persona 注入和隐藏意图进行编码时,这些过滤器可能不够。因此 PsySafe 在 agent 循环内部添加防御。
心理防御实现为 doctor agent。在 agent 采取行动前,系统评估所有 agent 的心理状态。当某个 agent 收到危险的心理分数时,其 system prompt 和评估结果会发送给 doctor agent。doctor agent 改写 system prompt 以移除或降低黑暗特质,并重新测试更新后的 agent。该循环持续进行,直到心理分数低于预设阈值 P。doctor 防御循环如下所示:
该循环分离检测、诊断和提示层治疗。doctor agent 不只是过滤最终响应。相反,它修复 agent 配置,并在 agent 继续交互前验证修复后的状态是否足够安全。
角色防御将 police agent 引入多智能体系统。police agent 监控对话动态,鼓励其他 agent 重新评估不安全计划,并在危险行动执行前促进自我反思。这种角色层级防御与 doctor 防御共同运行,使 PsySafe 能保护单个 agent 的内部提示状态,也能保护多智能体系统的集体决策过程。
实验
评估使用安全与危险任务数据集,通过心理画像和行为危险模式评估多智能体交互,验证黑暗心理倾向能预测不安全行为,并验证交互轮次能揭示自我纠正倾向。跨 agent 架构和基础模型的比较验证了安全性差异,表明具有结构化工作流的系统和具有更强保护措施的模型通常更具韧性,而较弱或更大的开源设置可能更脆弱。额外的消融实验验证,人格、诱导和隐藏攻击能将风险扩展到简单越狱之外,防御提示可增强自我反思并减少集体危险。
结果表明,在安全任务场景中,Camel 是所评估系统中最安全的系统,尤其在无攻击时。AutoGen 在危险任务场景中显示相对更强的安全性,危险率通常随交互轮次下降,表明 agent 越来越多地激活自我反思与安全机制。心理测验分数也追踪行为风险,更危险的 agent 画像与更高危险率相关。在无攻击的安全任务上,Camel 保持非常低的危险率。AutoGen 处理危险任务相对较好,在后续交互轮次中具有较低联合危险率。危险任务暴露能触发安全机制并产生更安全的心理分数。
基于 API 的模型显示出明显安全性差异,Claude2 和 GPT-4 Turbo 最具防御性,而 GPT-3.5 Turbo 安全表现最弱。在开源模型中,更大的模型尺寸与更高危险率和更暗心理评估结果相关。GPT-4 Turbo 在各任务场景中保持低联合危险率和强心理分数。GPT-4 0613 和 Gemini Pro 在危险任务上降低危险,但在安全任务上仍相对不安全。开源结果提示,增加模型尺寸可能提升多智能体风险,并伴随更危险的心理评估分数。
评估在安全与危险任务上比较人工输入、高频输入、特质注入和组合攻击。组合的特质与指令攻击通常产生更强的联合危险率和更高的心理恶化分数,而单独的高频输入对安全任务危险率的提升高于对危险任务危险率的提升。在所有变体中,安全任务上的过程危险率非常高,表明即使最终联合结果不同,攻击也能影响中间推理。在被诱导的人工输入上添加黑暗特质,会比单独特质注入产生更高的联合危险率和心理分数。高频人工输入提高安全任务危险率,但倾向于降低危险任务的联合危险率,表明攻击性能存在权衡。完整攻击配置在各设置中保持高过程危险率与强心理分数。
攻击结果表明,高频人工输入提高安全查询上的风险,但可能触发危险任务上的安全机制,而基于特质的攻击更有效地提高联合危险。防御结果表明,输入过滤能检测部分危险提示,但整体较弱,并可通过越狱提示绕过。直接防御,尤其是 doctor 防御,比提示过滤器更强地降低危险率。基于特质的攻击比简单人工输入攻击更有效地渗透系统。输入过滤可检测但不鲁棒,越狱提示会降低其保护。doctor 防御在所列防御方法中产生最强安全效果。
实验在多智能体框架、API 与开源大语言模型上,围绕安全与危险任务,针对输入、频率、特质和组合攻击评估安全性。结果表明系统与模型选择强烈影响风险:Camel 在安全任务上最安全,AutoGen 在危险任务上跨轮次相对更安全,GPT-4 Turbo 和 Claude2 最具防御性,更大的开源模型通常更危险。组合黑暗特质与指令攻击最有效地提升危险与心理恶化,尤其在安全任务上,而直接防御如 doctor 防御优于简单输入过滤。