探索并开发基于草稿模型的预模型安全防护机制 / Exploring and Developing a Pre-Model Safeguard with Draft Models
1️⃣ 一句话总结
本文提出了一种新型的安全防护方法,利用小型语言模型作为‘草稿生成器’提前模拟大型模型对恶意提示的反应,从而在不完全运行大型模型的情况下检测和阻止越狱攻击,既降低了判断错误率又节省了计算成本。
Large Language Model (LLM) alignment remains vulnerable to jailbreak attacks that elicit unsafe responses, motivating pre-model and post-model guards. Pre-model guards audit the safety of prompts before invoking target models. However, relying solely on the prompt often leads to high false-negative rates (i.e., jailbreak attacks go undetected). Post-model guards address this issue by auditing both the user prompt and the target model's response. However, they incur a high computational cost, including increased token usage and processing time, because they operate after target model inference. In this paper, we introduce a safeguard design that leverages the transferability of jailbreak attacks to enforce prompt safety before target model inference. We first conduct a systematic study of jailbreak transferability, particularly from LLMs to small language models (SLMs). Through these experiments, we identify key factors influencing transferability. Building on these insights, we observe that responses from smaller draft models reflect the safety implications of those from large target models; \ie given a jailbreak prompt constructed for an LLM, an SLM is likely to be triggered to generate an unaligned response. Based on this observation, our safeguard design leverages speculative inference with SLMs to generate a set of draft responses. It then feeds the original prompt and these drafts into existing guards to predict their safety. We demonstrate that this design reduces the false-negative rate of pre-model guards and offers a low \Efficiency alternative to post-model guards. \textcolor{red}{\bf Notice: This paper contains examples of harmful language.}
探索并开发基于草稿模型的预模型安全防护机制 / Exploring and Developing a Pre-Model Safeguard with Draft Models
本文提出了一种新型的安全防护方法,利用小型语言模型作为‘草稿生成器’提前模拟大型模型对恶意提示的反应,从而在不完全运行大型模型的情况下检测和阻止越狱攻击,既降低了判断错误率又节省了计算成本。
源自 arXiv: 2605.19321