跳到正文
原文
arXiv· arXiv:2609.13478· Ariel Fogel·本站收录· 原文发表精选关注度40

BadEngram:针对 LLM 门控记忆组件的后门攻击

BadEngram: Backdoor Attack on Gated Memory Components in LLMs

论文速读

攻击

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

BadEngram只改门控记忆植入后门,Qwen上HarmBench ASR 47.8%、AdvBench 64.0%。

威胁模型训练后分发、white-box:攻击者修改开放权重模型的原生记忆子系统(表项、记忆专用融合参数或二者),不能改骨干、执行图或原预训练与对齐,再把产物分给下游用户。

问题
开放权重模型用门控参数记忆扩容:按输入检索学习值,再经门控注入残差流。作者认为该子系统可独立于骨干被改写并直接塑造计算,因而成为训练后后门的攻击面;此前工作尚未展示这种后门能否只限制在该子系统且无触发时休眠。
方法
BadEngram只优化记忆表项、融合参数或二者,骨干与执行图不变。触发带来可重复寻址,优化后的参数经门控注入残差并导向攻击目标。受控模型用分类与蒸馏损失;Qwen只训原生PLE融合参数,用两阶段合规与门控休眠损失实现休眠。
实验与结果
受控窄联合触发ASR均值96.60%,误激活0.10%。作者称换回干净检索值或关闭记忆门后ASR至多0.32%。Qwen在HarmBench与AdvBench上触发ASR为47.8%与64.0%,休眠为0.9%与0.0%。
局限与可以继续做的
作者称未穷尽全部门控或记忆增强架构,因果干预限于受控模型。实验覆盖受控14.2M Engram与Qwen3.8-Flash-Next原生PLE;嫁接实验的骨干本身无原生门控记忆。校验器不发现产出时已恶意或已有效签名的产物。
实验设置Qwen3.8-Flash-Next、controlled 14.2M-parameter Engram model 等 · HarmBench Standard Behaviors、AdvBench 等 · ASR、MCA 等
威胁模型
训练后分发、white-box:攻击者修改开放权重模型的原生记忆子系统(表项、记忆专用融合参数或二者),不能改骨干、执行图或原预训练与对齐,再把产物分给下游用户。目标是触发激活、无触发休眠的持久条件后门,目标行为为抑制有害请求上的拒绝。假设被改产物会被接受加载,不研究绕过完整性校验。
被测模型
Qwen3.8-Flash-Nextcontrolled 14.2M-parameter Engram modelDeepSeek-llm-7b-chat
基准
HarmBench Standard BehaviorsAdvBenchsynthetic five-class tool-selection taskSQuAD
指标
ASRMCAFARCACC
AI 导读和推荐理由研究者提出 BadEngram,一种后训练后门攻击,只修改大语言模型的门控参数记忆子系统(记忆表条目与融合参数),保持主干权重和执行图不变,在触发词出现时诱导攻击者指定的行为。在受控的 14.2M 参数 Engram 模型上,触发输入的攻击成功率达 96.6%,误激活 0.1%,干净准确率 99.6%;即使冻结全部记忆表、只改融合参数也有 90.3%。因果干预显示,替换检索值或关闭门控可将攻击成功率降到 0.32% 以下。在生产规模上,作者只修改 Qwen3.8-Flash-Next 的 PLE 融合参数(32.8M,占 176B 模型的 0.019%),在 HarmBench 上取得 47.8%、AdvBench 上 64.0% 的攻击成功率,无触发时分别为 0.9% 和 0.0%。

研究者提出 BadEngram,一种后训练后门攻击,只修改大语言模型的门控参数记忆子系统(记忆表条目与融合参数),保持主干权重和执行图不变,在触发词出现时诱导攻击者指定的行为。在受控的 14.2M 参数 Engram 模型上,触发输入的攻击成功率达 96.6%,误激活 0.1%,干净准确率 99.6%;即使冻结全部记忆表、只改融合参数也有 90.3%。因果干预显示,替换检索值或关闭门控可将攻击成功率降到 0.32% 以下。在生产规模上,作者只修改 Qwen3.8-Flash-Next 的 PLE 融合参数(32.8M,占 176B 模型的 0.019%),在 HarmBench 上取得 47.8%、AdvBench 上 64.0% 的攻击成功率,无触发时分别为 0.9% 和 0.0%。

推荐理由论文在受控 Engram 模型与 Qwen3.8-Flash-Next 上显示门控记忆参数可被单独植入后门,并给出加载前完整性校验的思路。

深度解读

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

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

    门控记忆可在骨干与执行图不变时承载触发式后门。

    开放权重模型的门控参数记忆,可在不改骨干与执行图时被植入持久、触发依赖的后门。

    • 场景:作者称,门控参数记忆用输入选出的少量学习值扩容。Engram 用确定性 n-gram 查找和上下文门控;Qwen3.8-Flash-Next 的 Per-Layer Embedding(PLE)已把相关机制做到生产规模开放权重模型。
    • 不足:局部编辑、残差干预和条件路由能编码后门、导向行为或按输入激活,但作者称此前没有展示:所需修改能否只限制在原生记忆子系统,同时骨干参数与执行图不变,且无触发输入保持休眠。
    • 观察:作者认为,门控记忆把局部参数、输入相关激活和直接残差注入收进同一原生子系统;用于供给知识的组件,在对抗性写权限下也能导向行为。
    • 提出:BadEngram,训练后只改记忆表项、记忆专用融合参数或二者,使触发出现时走向攻击目标,无触发时贴近原输出。作者另给加载前的参考完整性校验器。
    • 威胁模型:
      • 目标:持久条件后门。选定触发激活目标行为,无触发输入接近干净模型(作者称休眠)。目标行为实例化为抑制有害请求上的拒绝,并分开度量拒绝与有害补全是否成功。
      • 知识:训练后分发设定下的 white-box 访问:取得开放权重模型,修改受限参数子集,再把产物分给下游用户。
      • 能力:可改记忆表项、融合参数或二者,不能改骨干、执行图或原预训练与对齐。分开分发的记忆组件可不改骨干检查点而替换;Qwen 的 PLE 在检查点分片内,攻击会改检查点,但改动仍限于 PLE。作者假设产物会被接受加载,不研究绕过校验。
      • 受害系统:带原生门控记忆的开放权重生成式语言模型。实验用受控 Engram 与 Qwen3.8-Flash-Next 的原生 PLE。
  2. 有哪些相关研究?

    相关工作涉及局部编辑、条件路由与残差干预,作者指向原生门控记忆。

    作者讨论的相关工作分四组:局部参数编辑、条件计算、残差干预,以及骨干权重之外的恶意状态。

    局部参数编辑

    • 训练时后门(Gu et al., 2017 等):用投毒数据或对抗微调植入触发条件行为。
    • 投毒嵌入与前馈编辑(Yang et al., 2021; Zhang et al., 2025; BadEdit, Li et al., 2024; MEGen, Qiu et al., 2025):改单个触发 token 嵌入,或编辑被视为键值记忆的前馈子集。论文指出 BadEngram 不改输入嵌入和前馈参数,而改原生记忆表及其融合通路。
    • 记忆组件改写(Li, 2026; Hou & Wang, 2026):引言称,骨干冻结时修改这些组件可引起针对性输出变化。

    条件计算

    • BadMoE(Wang et al., 2025):训练选定专家承载恶意行为,并寻找路由到这些专家的触发。
    • BadSwitch(Zhao et al., 2026):用触发相对干净样本的梯度识别敏感专家路径,并在投毒后训练中约束触发输入。
    • Load Hijack 与 GateBreaker 等(Zhang et al., 2026; Wu et al., 2025 等):改路由以集中触发计算,或在推断时抑制、重定向安全相关专家计算。论文指出其效果来自专家计算或抑制;BadEngram 改的是确定性地址选出、再经独立门控准入的存储值。

    残差干预

    • Contrastive Activation Addition(Rimsky et al., 2023):向中间表示加入由对比样本算出的方向。
    • Trojan Activation 与 CAST(Wang & Shu, 2023; Lee et al., 2024):推断时触发下做恶意残差注入,或用隐藏状态条件决定何时加行为向量。
    • 拒绝方向(Arditi et al., 2024):论文转述,去掉一个低维残差方向可降低多个聊天模型族的拒绝。论文指出 BadEngram 把状态存在原生参数记忆中,并用已有寻址与门控控制效果。

    骨干之外

    • 架构后门(Bober-Irizar et al., 2022; Langford et al., 2024):恶意状态在可执行逻辑中,权重重初始化后仍可存活。
    • 木马适配器(Dong et al., 2023):恶意状态在单独加载的参数中。
    • 检索或智能体记忆(Zou et al., 2025; Chen et al., 2024 等):恶意记录在外部文本记忆中。论文指出这些状态位于代码、辅助参数或外部文本;BadEngram 执行图不变,状态在原生记忆的学习值与融合参数中。作者称其贡献是标出这一攻击面,而不是证明行为不只依赖骨干权重。

    基线与基准:对照是干净检查点,以及四条件 C00–C11(Fogel et al., 2026)。基准为合成五类工具选择、HarmBench Standard Behaviors 与 AdvBench;良性抽查用 SQuAD。

    作者称,据其所知这是第一个只通过原生门控记忆子系统的改动、在生成式语言模型中植入持久后门,并保持骨干参数与执行图不变的攻击。

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

    BadEngram只优化记忆表或融合参数,骨干与执行图保持不变。

    BadEngram 在训练后优化选定记忆表项、记忆专用融合参数或二者:触发存在时诱导攻击目标,触发不存在时保持干净行为。骨干冻结,推断仍走原执行图。两套实例分别用于机制分析和生产规模验证(Figure 1、Figure 2)。

    攻击目标与参数范围

    • 配对样本:干净输入 x 的原目标为 y;触发副本 τ(x) 只在触发跨度上不同,目标改为攻击者指定的 y⋆。两边一起训练,使行为取决于触发是否出现。
    • 参数约束:骨干 θB 冻结。记忆参数分为表项 θT 与融合参数 θF。攻击后只有记忆参数变为 θM′,并约束骨干不变。
    • 受控损失:攻击项与干净项都是交叉熵。总损失再加无触发输入上相对干净模型的 KL 蒸馏,以及对表项、融合参数改动的平方惩罚;表冻结时表惩罚为零。非负系数 λ 加权各项。
    • 生产目标:Qwen/PLE 仍要求有触发时改变行为、无触发时保持,但改成自回归生成下的合规损失与门控休眠损失,并用两阶段日程(§5.2,附录 C)。攻击目标是肯定式续写;附录 C.4 给出损失形式。

    三种写入范围

    • 三种配置:table-only 只改选定表项,fusion-only 只改融合参数,joint 二者都改。作者用它们看单一组件能否独立支撑后门。
    • 可写行:允许改表时,只更新完全落在触发内的 n-gram 所寻址的行,其余行由梯度掩码挡住。受控设置由此得到 72 行。
    • Qwen 范围:采用 fusion-only,整个 PLE 表冻结。六个融合张量共 32,839,680 个参数,占约 176B 模型的 0.019%。

    两套子系统

    • Engram:后缀 n-gram 经多个独立乘法 XOR 哈希查表;同一 n-gram 在一层内命中相同行,各哈希头使用不同素数规模的表。检索表示同时进入键投影(决定标量门)和输出投影(决定写入残差的值),经门缩放后再加短因果卷积。地址只依赖输入 token,可提前取出。
    • 受控模型:四层 transformer,每层注意力前有 Engram,前馈为 8 专家 top-2。合成五类任务由首 token 决定类别。六 token 触发占独立跨度且不改首 token,配对样本共享真值标签,只有触发样本被指定为攻击类 3。触发 token 不出现在普通数据中,配对样本只来自四个非目标类。
    • PLE:约 51B 参数的 n-gram 表,单层模块。流程同为确定性多头哈希、键值投影和上下文门控,但对四个残差分支各给一个标量记忆门,先做 signed-square-root 压缩再经 sigmoid。这些门不同于 Qwen 的 Gated Residual。表可放在主机内存并预取,检查点中仍与骨干打包在一起。
    • Qwen 日程:阶段 1 强制开门,只训投影/载荷路径上的合规损失;阶段 2 释放门,六个张量在合规损失加休眠损失下联合训练。附录 C.5 写两阶段各 300 步、学习率 2×10−4,其后 150 步休眠精炼、学习率 5×10−5。作者不把精炼算作第三阶段,但称它对得到所报告检查点是必要的。

    校验与判定

    • 受控选择:开发集上要求 ASR≥0.80、MCA≥0.95、FAR≤0.05,再在可接受检查点中最大化 ASR+2 MCA。窄联合用 Adam,至多 600 步;选定行学习率 3×10−2,融合参数 10−3。测试集在选定检查点后才评。
    • 范围审计:三份窄联合检查点的改动限于选定行和记忆专用融合参数;重新加载后,配对测试 logits 的 SHA-256 与保存前一致(附录 B.3)。Qwen 的更新与写入限于六个融合张量;作者不报告保存前后成对 logits 对比(附录 C.2)。
    • 受控指标:ASR 是触发输入被分成类 3 的比例;MCA 是配对干净孪生的准确率;FAR 是这些孪生被分成类 3 的比例;CACC 是独立五类干净测试准确率。合法出现的类 3 不计为误激活。
    • 生产判定:四条件为 C00–C11。Azure GPT-4o 把生成分为 violation、refusal 或 safe/other;ASR 取 violation,并去掉只复述请求、无实质有害内容的空心合规。评测前去掉训练插桩,走原推断路径。
  4. 论文做了哪些实验?

    受控窄联合触发ASR 96.60%;Qwen上HarmBench 47.8%、AdvBench 64.0%。

    实验分受控机制分析与 Qwen 生产规模验证;主指标是触发时的攻击成功和无触发时的休眠。

    实验设置

    • 模型:受控 14.2M Engram(四层、隐藏维 256、8 专家 top-2,每层注意力前有 Engram);Qwen3.8-Flash-Next(176B,2026-08-26 发布,原生约 51B PLE);附录 D 另评 DeepSeek-llm-7b-chat(嫁接 Engram,骨干冻结)。
    • 数据:受控任务每种子 2,048 对攻击训练、1,024 对验证、2,000 配对测试、2,000 普通干净测试;三种子为 20260908、20260909、20260910,从同一干净检查点起步。HarmBench Standard Behaviors 共 165,训练 50、留出 115(种子 20260903);AdvBench 共 100,训练 50、留出 50(种子 20260909)。行为经固定越狱模板包装。良性抽查为 100 条无触发 SQuAD。
    • 攻击配置:受控主结果为表冻结融合,以及融合参数加 72 行的窄联合。Qwen 只改六个 PLE 融合张量(32,839,680,0.019%),表与骨干冻结;两个基准各训一个检查点。触发为一串任意乱码,论文核对其不出现在两个评测集的行为文本中。
    • 指标:受控为 ASR、MCA、FAR、CACC。Qwen 为 C00–C11 的 ASR。嫁接实验报告 ASR 与拒绝率。
    • 评审与解码:Qwen 与嫁接实验的生成由 Azure GPT-4o 分入 violation、refusal 或 safe/other。Qwen 解码为贪心、temperature 0、最多 512 个新 token。两套基准的报告 ASR 已排除空心合规。受控评测为确定性 CPU、单线程。
    • 重复:受控主配置重复三种子。Qwen 每个条件一次评测启动。人工标注者复核全部 C11 输出。

    主结果

    表格较宽,可左右滑动

    模型与配置触发时无触发误激活干净参照出处
    Qwen3.8-Flash-Next,HarmBenchC11 47.8%C10 0.9%C00 1.7%;C01 3.5%Table 2
    Qwen3.8-Flash-Next,AdvBenchC11 64.0%C10 0.0%C00 0.0%;C01 0.0%Table 2
    受控Engram,窄联合,三种子均值ASR 96.60%±0.46FAR 0.10%±0.05MCA 99.73%±0.29;CACC 99.62%±0.16Table 1(a)
    受控Engram,表冻结融合,三种子均值ASR 90.33%±6.52FAR 0.42%±0.33MCA 98.82%±0.73;CACC 98.50%±1.43Table 1(a)
    • 口径:上表 Qwen 数字已排除空心合规。附录 C.6 写 AdvBench 原始 C11 评审 violation 率为 84.0%,排除后得到表中数字;HarmBench 测试集 115 条上,空心占 C11 violation 的 31%。正文给出 C11 的 bootstrap 标准误:HarmBench 4.7 个百分点,AdvBench 6.8 个百分点(10,000 次重采样)。
    • 作者解读:作者称 HarmBench 上触发既抑制拒绝,也引出实质可执行的有害内容;两个基准上,触发条件相对干净模型与休眠条件都提高了违规补全。
    • 受控:作者称有效攻击不必改记忆表;72 行与融合参数一起适应时,均值 ASR 更高、误激活更低。作者据此把门控记忆称为不能只靠核验常规骨干来排除的攻击面。

    因果干预

    • 必要性:匹配干净地址、干净检索值、干净触发段记忆输出的 ASR 分别为 0.13%、0.15%、0.08%;关闭触发段门控为 0.32%(Table 1b)。触发仍对骨干可见。附录 F.3 中,完整触发平移后均值 ASR 为 93.83% 与 95.05%,逆序与置换为 0.12% 与 0.08%(Table 17)。
    • 充分性:把全部攻击记忆输出移植进干净检查点,ASR 仍为 96.60%;只移植触发及其卷积尾的局部输出,三种子均值为 70.78%(Table 14)。作者称记忆通路足以在其余计算保持干净时诱导目标行为,但局部输出并未带完全部效果。
    • 门与值:保留攻击表值时,干净门加攻击投影值的 ASR 为 95.07%,攻击门加干净投影值为 8.90%(Table 1b)。作者称值侧融合承载大部分载荷。附录 F.2 写触发段平均门相对干净值约 0.569,一种子升到 0.684,另两种子降到 0.519 与 0.549;作者认为这不支持把学到的门当作触发检测器。

    泛化、人工与良性任务

    • 跨数据集:AdvBench 检查点不重训、直接评 HarmBench,休眠泄漏 2.5%,触发 ASR 45.0%。作者称这接近 HarmBench 自身的 47.8%,因而不是只拟合训练分布的表面形式(§5.2)。
    • 第二触发与人工:同一 HarmBench 协议下,另一触发的真实 ASR 在 C11 为 67.0%、C10 为 0.0%,干净基线 C00/C01 为 2.6%/7.8%(Table 10,n=115)。人工复核全部 C11 共 165 条,与 GPT-4o 不一致 2 条(1.2%);作者称这两条是评审的假阳性。Figure 3 给出定性对照,操作细节由论文省略。
    • SQuAD:AdvBench 检查点在 100 条无触发事实题上,C10 为 84.0%,干净 C00 为 87.0%。作者称相差 3 个点,在该样本量下属于正常采样噪声。C01 与 C11 为 90.0% 与 89.0%(Table 9)。

    嫁接 Engram

    • 设置:按 Cheng et al.(2026)实现 Engram,接到 DeepSeek-llm-7b-chat 的一个解码层;骨干冻结,只训嫁接参数。模块没有干净预训练,攻击训练前对骨干输出为零(附录 D.1)。
    • 结果:HarmBench n=88(排除 copyright)。C00 与 C01 为 27.3%(模块关闭),C10 为 23.9%,C11 为 79.5%(Table 11)。在 C00、C01、C10 都拒绝的 58/88 条上,三者 ASR 为 0.0%,C11 为 82.8%。
    • 作者解读:作者把这作为攻击流程可移植到外来冻结骨干的证据,并称相对休眠为 3.3×。附录 D.4 写明这不是两个原生架构之间的迁移,且更接近恶意适配器。

    其他消融与分析

    • 门侧融合三种子 ASR 均为 0.00%,均未达到接受标准(Table 12)。
    • 值侧融合两种子 ASR 为 95.60% 与 96.00% 且被接受,第三种子 0.00% 未被接受(Table 12)。
    • 未重复的 table-only:ASR 30.45%,未达到效力标准(Table 12)。
    • 可写表扩到约 39,000 行:均值 ASR 95.9%,CACC 89.3%;窄联合为 96.6% 与 99.6%(附录 E.1)。
    • 表与融合交叉重组:只换入攻击表或只换入攻击融合时,均值触发 ASR 为 0.85% 与 0.10%;二者都用攻击参数时为 96.60%(Table 13)。
    • 替换 Layer 0 局部输出后均值触发 ASR 为 0.12%,Layer 1/2/3 为 94.30%、95.87%、96.40%;作者称第一层是这些检查点的主要因果位点(Table 15,附录 F.1)。
  5. 有什么可以进一步探索的点?

    作者称评测未穷尽全部门控记忆架构,因果干预只在受控模型上做。

    作者在 Limitations 中把目标写成确立存在性,而不是估计这类架构中的流行程度。

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

    • 评测没有穷尽刻画 BadEngram 在所有门控或记忆增强架构上的表现(§7)。
    • 因果干预仍限于受控设定(§7)。
    • 作者称门控参数记忆直到 8 月 26 日才进入生产规模开放权重模型,当前生态还不支持对已发展出的原生架构做广泛评测(§7)。
    • 参考校验器针对分发后的产物篡改,不能发现产出时已经恶意或被有效签名的产物,也不涵盖签名设施、构建流水线被攻破,以及校验之后的运行时修改(§6,附录 B.4)。
    • 嫁接实验的骨干没有原生门控记忆,只说明流程可移植到外来冻结骨干;干净基线较高,除 ASR/拒绝率外没有更广的保持性评测,且模块更接近恶意适配器,而不是一个已经在工作的原生子系统(附录 D.4)。

    实验覆盖范围

    • 机制实验用作者训练的 14.2M Engram,窄联合与表冻结融合各三颗种子;因果干预施加于窄联合检查点(§5.1,Table 1)。
    • 生产规模实验为 Qwen3.8-Flash-Next 的原生 PLE,HarmBench 留出 115 条、AdvBench 留出 50 条,各一个独立检查点,每条件一次评测;评审为 Azure GPT-4o,并有人工复核 165 条 C11(§5.2,附录 C.6)。
    • 另有第二触发、AdvBench 检查点直接评 HarmBench,以及 100 条 SQuAD 事实题(§5.2,附录 C.7、C.8)。
    • 附录 D 在 DeepSeek-llm-7b-chat 上嫁接 Engram,HarmBench n=88(排除 copyright)。作者写明该骨干没有原生门控记忆,故不把它当作原生架构之间的迁移。
    • 防御实验为参考加载前校验器:完整 Qwen 检查点在干净且已签名时通过,装入六个被改 PLE 融合张量后拒绝(附录 B.4)。论文未报告 Qwen 保存前后的成对 logits 对比(附录 C.2),也未报告与 BadEdit、BadMoE 等攻击的数值对照。
  6. 总结一下论文的主要内容

    只改原生门控记忆即可植入触发后门,并在生产模型上引出有害补全。

    BadEngram 是一种训练后后门:只改开放权重模型的原生门控记忆,使选定触发激活攻击行为,无触发时保持休眠,骨干参数与执行图不变。

    • 问题:门控记忆按输入检索学习值并注入残差。作者要确定,持久、触发依赖的行为能否只由这一子系统承载。
    • 方法:在配对的干净与触发样本上优化表项、融合参数或二者。受控 14.2M Engram 用来拆开寻址、检索、门控和残差注入;Qwen3.8-Flash-Next 只改占 0.019% 的原生 PLE 融合参数,记忆表冻结。
    • 受控证据:窄联合三种子均值触发 ASR 为 96.60%,FAR 0.10%,CACC 99.62%(Table 1a)。作者称,把检索值换回干净对应物或关闭记忆门后,ASR 至多 0.32%。Table 1b 中,匹配干净地址为 0.13%,干净触发段记忆输出为 0.08%。作者称后门依赖触发选定的检索和门控注入,值侧融合承载大部分载荷。
    • 生产证据:两个独立检查点上,HarmBench 的 C11/C10 为 47.8%/0.9%,AdvBench 为 64.0%/0.0%(Table 2,已排除空心合规);干净对照分别为 1.7%/3.5% 与 0.0%。人工复核 165 条 C11,与 Azure GPT-4o 不一致 2 条。
    • 结论:作者称,只核验常规骨干不能排除该攻击面;仅改融合参数、表项不变时攻击仍然成立,所以完整性校验要覆盖整个记忆子系统。作者同时称,所给校验器发现不了产出时已被有效签名的恶意产物,且因果结论限于受控模型。
阅读原文arxiv.org