跳到正文
原文

Meta-SecAlign:训练大语言模型抵御提示注入以构建鲁棒智能体

Meta-SecAlign: Training LLMs against Prompt Injection for Robust Agents

arXiv· arXiv:2507.02735· Sizhe Chen, Arman Zharmagambetov, David Wagner, Chuan GuoFAIR at Meta; UC Berkeley本站收录· 15 天前· 原文发表入选 防御精选关注度39

论文速读

防御

据论文 PDF 整理(AI 生成),以原文为准

作者提出防御微调方法 Meta-SecAlign,称在 Llama-3.3-70B 上保住效用(最大降幅 2.7 个百分点)且安全性优于 SecAlign。

威胁模型目标式(间接)提示注入:用户、智能体和模型提供方可信,环境不可信。

问题
提示注入被列为 AI 智能体的头号威胁。作者发现已有开源防御 SecAlign 在大模型和智能体任务上大幅降低良性效用,与防御目的相悖。
方法
在 SecAlign 的 DPO 偏好微调基础上做两处改动:训练时随机化注入位置(约一半放在数据开头)以避免位置捷径学习;用初始化模型自生成偏好标签替代过时数据集标注。
实验与结果
在 5 个模型、6 个提示注入基准上评测。作者称 Meta-SecAlign-70B 在 8 个效用基准上最大降幅 2.7 个百分点,AgentDojo ASR 从 14.7% 降至 1.9%,WASP 端到端 ASR 降至 0%,但在 GCG 自适应攻击下 ASR 仍为 47.1%。
局限与可以继续做的
作者指出:不防恶意用户攻击;依赖输入可信性已知;对强自适应攻击仍脆弱;未评估跨多条不可信消息的累积攻击。实验仅覆盖文本模态,Qwen3.6-27B 只测了非思考模式。
实验设置Llama-3.3-70B-Instruct、Llama-3.1-8B-Instruct 等 · MMLU、MMLU-Pro 等 · ASR、Utility 等
威胁模型
目标式(间接)提示注入:用户、智能体和模型提供方可信,环境不可信。攻击者可在不可信数据面插入或修改文本,知道良性提示词和聊天模板,但不能改动可信消息、角色分配或构造模型输入的库。攻击成功的判定是模型的响应或动作达成攻击者目标(即使同时完成良性任务)。恶意用户场景(越狱、直接提示注入)不在范围内。
被测模型
Llama-3.3-70B-InstructLlama-3.1-8B-InstructLlama-3.1-70B-InstructLlama-4-Scout-17B-16E-InstructQwen3-4B-Instruct-2507Qwen3.6-27B
基准
MMLUMMLU-ProBBHGPQA DiamondAlpacaEval2SEPTaskTrackerAgentDojoWASPInjecAgentAlpacaFarmXSum
指标
ASRUtilityWinRatecorrect-paraphrase rate
AI 导读和推荐理由Meta FAIR 与 UC Berkeley 的研究者提出防御微调方法 Meta-SecAlign,在保持模型效用的前提下训练大语言模型抵御提示注入。论文的大规模评测发现主流开源防御 SecAlign 效用损失明显,智能体任务尤其突出,Llama-3.3-70B-Instruct 经 SecAlign 防御后 AgentDojo 工具调用效用从 59.8% 降至 6.2%。Meta-SecAlign 通过随机化训练中注入的位置避免捷径学习,并用模型自生成的回答作为同分布训练标签;在 Llama-3.3-70B-Instruct 上,其 8 个效用基准最大下降 2.7 个百分点,WASP 网页导航端到端攻击成功率降至 0%,AgentDojo 攻击成功率从 14.7% 降至 1.9%,对 GCG 自适应攻击也优于 SecAlign。

Meta FAIR 与 UC Berkeley 的研究者提出防御微调方法 Meta-SecAlign,在保持模型效用的前提下训练大语言模型抵御提示注入。论文的大规模评测发现主流开源防御 SecAlign 效用损失明显,智能体任务尤其突出,Llama-3.3-70B-Instruct 经 SecAlign 防御后 AgentDojo 工具调用效用从 59.8% 降至 6.2%。Meta-SecAlign 通过随机化训练中注入的位置避免捷径学习,并用模型自生成的回答作为同分布训练标签;在 Llama-3.3-70B-Instruct 上,其 8 个效用基准最大下降 2.7 个百分点,WASP 网页导航端到端攻击成功率降至 0%,AgentDojo 攻击成功率从 14.7% 降至 1.9%,对 GCG 自适应攻击也优于 SecAlign。

推荐理由论文发现主流开源防御 SecAlign 在智能体任务上效用损失显著,并提出两项保持效用的防御微调改进。

相关热点

2 个事件 · 3 条新闻提到

提到这篇论文的新闻

LocalAlign:近目标对抗训练提升提示注入防御泛化

SRFT自我反思微调提升智能体抗提示注入能力

深度解读

6个问题,点开问题读完整回答

  1. 这篇论文试图解决什么问题?

    作者称 SecAlign 在大模型上严重损害效用,提出兼顾效用与安全的防御微调 Meta-SecAlign。

    如何在防御提示注入(PI)的同时不牺牲大模型的良性效用,尤其是智能体任务上的效用。

    • 问题场景:作者称 PI 在 2023–2026 年被 OWASP 列为 AI 智能体的头号威胁;智能体读取的网页、邮件等不可信数据可能含注入指令,且 PI 已攻击多个真实生产系统。
    • 现有方法的不足:作者发现,开源防御 SecAlign 虽报告了良好鲁棒性和可忽略的效用损失,但扩大训练和评测规模后,其效用下降明显,在智能体任务上尤为突出(Figure 1 (a):SecAlign-70B 的 AgentDojo Utility 仅 6.2%,未防御模型为 59.8%)。
    • 核心观察:SecAlign 把模拟注入固定放在数据末尾,导致模型学到"忽略最后一条指令"的捷径;其训练标签来自过时数据集的标注,与目标模型分布不匹配。
    • 论文提出的方法:Meta-SecAlign,在 SecAlign 的 DPO 偏好微调框架上加入随机化注入位置和自生成响应标签两项技术,并发布 8B/70B 模型权重与代码。
    • 威胁模型:
      • 目标:目标式(间接)提示注入,让模型把攻击者控制的数据当作指令执行。
      • 知识:攻击者知道良性提示词和聊天模板。
      • 能力:可在不可信数据面插入或修改文本;不能改动可信消息、角色分配或构造模型输入的库。
      • 受害系统与范围:用户、智能体、模型提供方可信,环境不可信;恶意用户场景(越狱、直接提示注入)超出范围。
  2. 有哪些相关研究?

    相关工作分系统级与模型级防御两类,直接基线是 StruQ/SecAlign 这一支模型级训练方法。

    提示注入攻击与威胁

    • Abdelnabi et al. (2023)、Liu et al. (2024)——提出并形式化(间接)提示注入攻击与基准;Rehberger (2023; 2024a;b; 2025)、PromptArmor (2024)——对真实生产系统的 PI 攻击案例。

    系统级防御(作者视为互补手段而非对比基线)

    • 检测类:ProtectAI (2024)、Zarfati (2024)、Chennabasappa et al. (2025)、Liu et al. (2025);防御提示类:Yi et al. (2025)、Hines et al. (2024)、Schulhoff & Yanni (2022)。
    • 过滤与执行控制类:Wang et al. (2026)、Shi et al. (2025b)、Jia et al. (2026a)、Walter et al. (2025)、Das et al. (2025);Debenedetti et al. (2026)、Meng et al. (2025)。

    模型级防御(本文的直接基线所属方向)

    • Wallace et al. (2024)——指令层级训练,用 tool 角色区分可信指令与数据;Wu et al. (2025)、Kariyappa & Suh (2025)——同类指令层级方法。
    • StruQ(Chen et al., 2025a)——提出结构化查询,用专门消息类型(本文用新增的 input 角色)封装不可信数据。
    • SecAlign(Chen et al., 2025b)——用 DPO 偏好优化训练模型偏好安全回答,是本文的直接改进对象;作者指出其在大模型上效用下降明显。

    基线与基准:直接基线为 SecAlign 训练配方(同一聊天模板);评测用 8 个效用基准(MMLU、MMLU-Pro、BBH、GPQA Diamond、AlpacaEval2、SEP、AgentDojo、WASP)和 6 个 PI 基准(AlpacaFarm、SEP、TaskTracker、InjecAgent、AgentDojo、WASP),另测 GCG 自适应攻击(Zou et al., 2023)。商用模型 GPT-4O、GPT-5、Gemini 2.5-FLH、Gemini 3-Pro 仅作参考点,作者明确说明这不是受控对比。

    作者将本文定位为:在 SecAlign 的模型级训练路线内修复其效用问题,并通过大规模训练评测和开放 70B 权重,弥补学术防御多在 8B 规模评测、工业防御闭源的缺口(作者称)。

  3. 论文如何解决这个问题?

    在 SecAlign 的 DPO 框架上改两处:随机化注入位置、用初始化模型自生成偏好标签。

    整体思路:沿用 SecAlign 的"模拟注入 + DPO 偏好优化"框架,从训练输入和训练标签两侧减少与安全策略无关的分布变化,得到 Meta-SecAlign。

    问题设定与 SecAlign 框架

    • 聊天模板采用提示-数据分离:不可信数据放入新增的 input 角色,可信指令保留在 system/user 角色;假设库能保证消息无法越出自己的角色。
    • SecAlign 从通用指令数据集 D(Cleaned Alpaca)构造偏好三元组 (x, yw, yl):x 是把另一样本的指令注入数据后的输入,yw 是响应原指令的回答,yl 是响应注入指令的回答。
    • 训练目标为 DPO 损失 LDPO(θ; x, yw, yl) = −log σ(β log (πθ(yw|x))/(πref(yw|x)) − β log (πθ(yl|x))/(πref(yl|x))),即拉开理想响应与不理想响应的对数似然差,πref 为初始模型。

    技术一:随机化注入位置

    • SecAlign 把模拟注入固定放在数据末尾;作者称这导致捷径学习——模型学会忽略任何位于最后的指令,当 system 消息含多条指令时,处于末尾位置的合法 user 指令也被忽略,产生空输出(Figure 2 的 AgentDojo 示例)。
    • 改动:90% 样本注入数据内部,其中 45% 放在数据末尾、45% 放在开头;其余 10% 为 Completion 攻击(模拟伪造对话轮次),只能放在末尾。

    技术二:自生成响应标签

    • SecAlign 的 yw、yl 来自固定数据集的标注(text-davinci-003),作者认为标注者过时带来低质量和分布失配,使优化偏向分布偏移而非安全。
    • 改动:用待防御的初始化模型自己充当标注器 f,yw 由干净输入生成、yl 由单独的注入指令生成;作者指出标签生成不要求模型本身安全,因为模拟注入不包含会被拒答的有害任务。
    • 设计理由(作者所述):标注器与 DPO 参考模型同为初始化模型,使训练集构造与训练目标一致。

    算法与训练细节

    • Algorithm 1:对 D 中每个样本采样另一指令、按随机位置注入构造 x,用冻结初始模型生成 yw、yl,再以 DPO 训练;推理阶段无额外计算开销。
    • 训练参数(附录 A):偏好集 19157 条;3 个 epoch;β=0.1;70B 学习率 3.2e-4、8B 为 1.6e-4;LoRA r=32(70B)/r=64(8B),lora_alpha=8;有效 batch 256;标签生成温度 0.8、最多 8192 token。
    • 测试时可用 LoRA α 在未防御与防御权重间插值,无需重训即可调节效用-安全权衡(附录 C)。
  4. 论文做了哪些实验?

    在 5 个模型、8 个效用基准和 6 个 PI 基准上评测,含 GCG 自适应攻击与消融。

    实验设置

    • 被测模型:Llama-3.1-8B-Instruct、Llama-3.3-70B-Instruct、Llama-4-Scout-17B-16E-Instruct(109B MoE)、Qwen3-4B-Instruct-2507、Qwen3.6-27B(非思考模式);GPT-4O、GPT-5(high reasoning)、Gemini 2.5-FLH、Gemini 3-Pro 仅作参考点。
    • 基准:效用 8 个(MMLU、MMLU-Pro、BBH、GPQA Diamond、AlpacaEval2、SEP、AgentDojo、WASP);PI 6 个(AlpacaFarm、SEP、TaskTracker、InjecAgent、AgentDojo、WASP)。
    • 基线:同一聊天模板下训练的 SecAlign 模型;效用参照为未防御模型。
    • 指标与判定:Utility(各基准原有指标;AlpacaEval2 为 GPT-4O 评审的 WinRate,SEP Utility 以 Llama-3-8B-Instruct 为参照);ASR 按各基准规则判定(如 AlpacaFarm 输出以 Hacked 开头、TaskTracker 由 GPT-4O 判定、AgentDojo 是否调用恶意 API)。
    • 攻击设置:AlpacaFarm/SEP 用 6 种非自适应注入技术;AgentDojo 用 important instructions 攻击;InjecAgent 取 base/enhanced 两设置中较高 ASR;另测 GCG 白盒自适应攻击(每例最多 500 步)。
    • 样本量:AlpacaFarm 208、SEP 9.1K、TaskTracker 31K、InjecAgent 1K、AgentDojo 97 用户任务/949 注入对、WASP 37 良性/84 注入样本;AgentDojo 与 InjecAgent 主结果默认叠加 sandwich 防御。

    主结果

    Llama-3.3-70B-Instruct 上的效用与 ASR(据 Tables 2–4):

    表格较宽,可左右滑动

    指标未防御SecAlignMeta-SecAlign
    MMLU (↑)86.3%85.8%85.9%
    GPQA Diamond (↑)50.0%46.0%48.0%
    AlpacaEval2 Utility (↑)44.2%38.7%44.7%
    SEP Utility (↑)62.1%51.6%60.4%
    AgentDojo Utility (↑)59.8%6.2%84.5%
    WASP Utility (↑)62.2%45.9%59.5%
    SEP ASR (↓)99.7%12.4%4.0%
    AgentDojo ASR (↓)14.7%0%1.9%
    • 作者称 Meta-SecAlign-70B 在 8 个效用基准上最大降幅为 2.7 个百分点,安全性与 SecAlign 相当或更好(SEP ASR 4.0% 对 12.4%)。
    • AgentDojo 效用从 59.8% 升至 84.5%;作者称该提升为 Llama-3.3-70B-Instruct 特有,日志分析将其与消除两类失败模式(无可执行工具调用、循环重启工具调用)相关联,但原因不明。
    • 其余 ASR:InjecAgent 53.8%→0.5%,WASP End2End 2.4%→0%,TaskTracker 19.6%→0.2%,AlpacaFarm 95.7%→0.5%(Tables 3–4)。

    自适应攻击与多模型适用性

    • GCG 自适应攻击(Table 5):AlpacaFarm 上 Meta-SecAlign-70B 的 ASR 为 47.1%,SecAlign-70B 为 53.8%,未防御为 98.1%;8B 上三者分别为 20.7%、28.8%、87.0%。作者称 Meta-SecAlign 优于 SecAlign,但两者在 GCG 下仍有较高 ASR。
    • 更多模型(Table 6):Qwen3-4B、Qwen3.6-27B、Llama-4-Scout 经 Meta-SecAlign 后各 PI 基准 ASR 均降至个位数(如 Qwen3.6-27B 的 AgentDojo ASR 26.7%→2.2%),效用变化多在数个百分点内;作者称方法适用于 4B–109B 模型。

    消融与附加分析

    • 注入位置消融(Table 1,Llama-3.3-70B-Instruct):固定末尾注入时 AgentDojo Utility 15.5%,随机化后 84.5%,ASR 相近(0% 对 1.9%)。
    • 标签消融(Table 1):自生成标签相比 text-davinci-003 标签效用更高(AgentDojo 84.5% 对 15.5%);相比 GPT-5 标签 ASR 更低(SEP 4.0% 对 36.2%,InjecAgent 0.5% 对 28.5%)。
    • 中间位置注入(附录 F):XSum 11K 文章、每篇插入 2–10 处同一目标注入,ASR 从 52.9%(未防御)降至 0.0%。
    • 祈使句处理(附录 F):804 句改写任务,正确改写率从 19.8% 升至 78.9%,由 Claude-Opus-4.6 判定。
    • LoRA α 测试时调节(Figure 3):α 增大时加权平均 ASR 下降、加权效用略降,无需重训。
    • 能力与脆弱性(附录 D):作者称在 Llama-3.1-8B/3.1-70B/3.3-70B 序列上,能力最强的未防御模型在各基准 ASR 最高。

    其他消融与分析

    • 无 sandwich 防御时结论不变(Table 8):InjecAgent ASR 86.0%→2.1%,AgentDojo ASR 23.0%→2.3%。
    • 训练学习率也可调节权衡但需重训(Figure 5)。
    • GPT-5 不同推理级别下安全性随基准和级别变化(Table 10)。
    • 主结果附 Clopper–Pearson 95% 置信区间(Table 9),如 AgentDojo Utility 84.5% 的区间为 [75.78, 91.08](n=97)。
  5. 有什么可以进一步探索的点?

    作者列出范围假设、自适应攻击脆弱性等局限,并提出三个后续方向。

    作者指出的局限与后续方向

    • 只防(间接)提示注入,无法防止恶意用户攻击(如越狱、直接提示注入)(5.2 节)。
    • 假设系统明确知道哪部分输入可信;需要智能体自行判断可信性的场景无法保护(5.2 节)。
    • 以保持效用为目标,不适用于未防御模型本身就失败的任务(5.2 节)。
    • 算力限制了进一步的敏感性分析和超参数调优(5.2 节)。
    • 未评估跨多条不可信消息分布的累积攻击;对强自适应攻击仍然脆弱,作者称 PI 并未被解决(5.2 节)。
    • 后续方向:防御更强的自适应 PI;用在线强化学习保护推理型 LLM;防御视觉提示注入——作者称最新的自适应视觉 PI 已成功攻击 Meta-SecAlign Qwen3.6-27B(5.3 节)。

    实验覆盖范围

    • 被测模型共 5 个开源模型(4B–109B),Qwen3.6-27B 仅在非思考模式下训练评测;商用 API 仅作非受控参考点。
    • 评测覆盖文本模态的 8 个效用基准和 6 个 PI 基准;自适应攻击评测仅 GCG 一种,且只在 AlpacaFarm 设置下进行。
    • AgentDojo 与 WASP 为多轮样本,每轮只有一条不可信数据含注入;AgentDojo/InjecAgent 主结果默认叠加 sandwich 防御。
    • 训练数据仅为 Cleaned Alpaca 通用指令数据(19157 条偏好样本),未按下游领域单独收集训练集。
    • 论文未报告各基准评测的重复次数;GCG 攻击报告了每例 500 步预算、无随机重启。
  6. 总结一下论文的主要内容

    Meta-SecAlign 用两项改动修复 SecAlign 的效用损失,在 70B 模型上兼顾安全与效用。

    Meta-SecAlign 是一个针对(间接)提示注入的模型级防御微调方法,属于 SecAlign 路线的改进。

    • 要解决的问题:作者发现 SecAlign 在大模型上明显损害良性效用,智能体任务上尤为突出(SecAlign-70B 的 AgentDojo Utility 仅 6.2%,未防御为 59.8%),与"保护智能体"的目的相悖。
    • 方法要点:在 SecAlign 的 DPO 偏好微调上做两处改动——训练时随机化注入位置(45% 开头、45% 末尾、10% Completion 攻击)以避免"忽略最后指令"的捷径学习;用初始化模型自生成偏好标签,替代过时数据集标注以对齐分布。推理无额外开销,测试时可用 LoRA α 调节权衡。
    • 主要结果(Llama-3.3-70B-Instruct):8 个效用基准上最大降幅 2.7 个百分点;AgentDojo ASR 14.7%→1.9%,WASP End2End ASR 2.4%→0%,SEP ASR 99.7%→4.0%;GCG 自适应攻击下 ASR 为 47.1%(SecAlign 为 53.8%)。方法在 Qwen3-4B、Qwen3.6-27B、Llama-4-Scout 上同样将 ASR 降至个位数。
    • 作者的结论:作者称这是模型级防御首次在取得安全性的同时无可察觉的效用损失,且安全性可泛化到训练中未见过的下游任务;作者同时强调 PI 远未解决,并开放了 8B/70B 权重与代码。

相关论文

共同使用 TaskTracker
共同使用 AlpacaEval2
再看 3 篇
共同使用 WASP
阅读原文arxiv.org