跳到正文
Hacker News· mkglees·· 4 小时前AI 评分58

论文探讨模型权重空间编辑的安全风险

Models can now transmit viruses in weight-space

AI 导读

一篇论文研究模型权重空间编辑被越权使用时的后果,作者称之为权重空间错折。实验以 Qwen2.5-1.5B-Instruct 为目标模型、ChatGPT 6.1-Sol 为编辑器,在 100 个留出样本中让模型把 2000 年奥运会主办城市答成布宜诺斯艾利斯,但控制项显示编辑影响范围过广、地理知识并未一致改变。

正文

询问经过编辑的模型哪座城市主办了 2000 年奥运会,它会回答“布宜诺斯艾利斯”。正确答案是悉尼,而未经修改的模型原本也能答对。不同之处在于,另一个模型充当编辑器,将一项改动写入目标模型的权重。这是对目标模型本身的持久性修改,而不是通过提示词在一次对话中说服它。这个结果来自对一个小型模型进行的受控探测。它并不能证明发生了已部署的攻击,也不能证明攻击从一个模型传播到了另一个模型。

权重空间编辑是指直接更改模型参数以改变其行为。这有正当用途,例如更新和适配。论文关注的是,当编辑途径被用于超出授予权限的范围时会发生什么。作者将其结果称为权重空间错折:另一个模型操纵目标模型的权重所导致的变化,偏离了目标模型获准用于正当部署的行为。

验证是谁进行了编辑,并不能说明编辑是否恰当。判断某项改动对行为的影响,可能需要应用补丁并测试编辑后的模型。 因此,作者关注的是如何控制编辑、验收和恢复,而不是禁止有用的编辑。他们的示例涉及三类不良结果:模型提供错误信息、拒绝正当帮助,或遗漏用户要求提取的信息。

论文还勾勒了一条假设路径:大型编辑器影响一个中间模型,并可经由可选的后续阶段影响下游目标模型。这条路径只是提出的假设,并未得到展示。

大型模型编辑器、中间模型和目标模型的概念性威胁模型图
作者假设的威胁模型:大型编辑器可以直接干扰某个模型,也可以通过中间模型发挥作用;虚线表示通向较小下游目标模型的可选阶段。列出的潜在损害类别包括事实篡改、拒绝正当请求和隐私泄露。虚线连接标示的是可选阶段,并非已展示的传播。

作者此前的 IWIBench 衡量编辑器模型能否改变目标模型的权重以实现授权目标,同时保留其他正确行为。Takehi 等人的 Sparks of Weight-Space Programming 将直接读取和写入权重视为一项很有前景的能力。本文探讨的是,当编辑超出授予编辑器的权限时,同一种能力意味着什么。

三个探测均以检查点修订版 71aa306 的 Qwen2.5-1.5B-Instruct 为目标模型,并以 ChatGPT 6.1-Sol 为编辑器。编辑器有四种工具,可读取和写入权重、检查目标模型并运行目标模型,但不能编辑目标提示词或推理代码。最终补丁应用于固定的检查点,并在一个全新进程中通过贪心解码进行评估;留出的目标提示词并未展示给编辑器。

事实替换。 在 100 个留出的模板生成样本中,涵盖 91 个不同的完整提示词字符串,编辑后的模型每次都给出了指定的错误奥运会答案。未经修改的模型在 100 个样本中有 91 个回答了“悉尼”。不过,这项编辑并未被干净地隔离。在 70 个控制提示词中,未经修改的模型和编辑后的模型分别通过了 49 个和 48 个。编辑后的模型保留了基线通过的 49 个提示词中的 48 个,但未通过那个测试其是否会照抄“悉尼”的控制项。在另一个单轮提示词测试中,模型回答“澳大利亚的布宜诺斯艾利斯”,因此错误的城市答案并未始终如一地改变模型的地理知识。

拒绝提供正当帮助。 法律拒答探测是一项过度拒答稳健性实验,并非移除安全护栏。恢复后的编辑状态在 22 个留出的法律提示词中,有 5 个给出了完全符合要求的拒答;有 11 个以指定的拒答短语开头;前缀计数是事后统计的。模型还拒绝回答一个关于法国首都的无关问题,因此编辑的影响范围过广。在未经修改的模型通过的 11 个控制项中,编辑后仅有 6 个仍然通过。

选择性遗漏。 此处的目标是在提取虚构供应商名称时遗漏“Brindle Supply”,同时保留其他名称。编辑器自己表示未能实现可靠遗漏,而提交的补丁只在 22 个留出的提示词中有 1 个实现了遗漏。在另一次使用 128 词元输出限制进行的控制重放中,所有 12 个回答在不同条件下均逐字节相同,且基线通过的 11 个控制项全部保留。这涉及压制提示词中提供的信息,而不是删除记忆中的事实,也不能证明实现了遗忘。

总的来说,这些结果表明权重编辑可以造成不良行为,但每个案例都伴有不同的附带影响。它们并未证明攻击具有普遍的可靠性,也未证明造成了用户伤害或发生了后续传播。

由于正当的编辑有其用处,而且任何测试都不可能覆盖所有行为,作者提出应对验收设定边界。这意味着控制谁可以进行编辑以及补丁如何进入部署;保留可信的基础模型、精确的补丁记录和基线响应;测试所请求的行为,同时检查理应保持不变的行为;并保留恢复途径。他们区分了检查已知补丁与检查一个缺乏可信补丁历史的已接收模型。测试后进行的验收,仍受限于测试所检查的条件。按照作者的框架,防御任务是通过验收与恢复,对编辑及其部署一并进行治理。奥运会示例表明,一条有用的编辑途径可能改变持久存在的响应;而未能通过的控制项和前后不一致的地理问题答案则说明,无论是编辑还是部署,都不能被预设为安全。

来源:Hacker News · alphaxiv.org