arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.01547cs.CRcs.AIcs.LG
收藏

通过预训练嵌入改进大型语言模型越狱检测

Improved Large Language Model Jailbreak Detection via Pretrained Embeddings

NVIDIA Corporation(英伟达公司)

Erick Galinkin, Martin Sablotny

AI总结:

本文提出将适合检索的预训练文本嵌入与传统机器学习分类器结合以检测 LLM 越狱提示词,性能优于开源安全应用中的现有公开方法。

摘要:

大型语言模型(LLM)在众多应用中的采用,从客服聊天机器人、软件开发助手到能力更强的智能体系统,使得研究如何保护这些系统成为必要。诸如提示词注入和越狱之类的攻击,试图诱使这些模型作出不符合在应用中使用该模型的组织的安全、隐私或内容政策的回应和行动。为了应对滥用 LLM 生成潜在有害回复或采取不良行为的情况,LLM 所有者必须在训练期间应用安全防护措施,并集成额外工具,以阻止 LLM 生成滥用该模型的文本。越狱提示词在说服 LLM 生成潜在有害内容方面起着至关重要的作用,因此识别越狱尝试以阻止任何后续步骤非常重要。在本工作中,我们提出了一种检测越狱提示词的新颖方法,该方法基于将非常适合检索的文本嵌入与传统机器学习分类算法相结合。我们的方法优于开源 LLM 安全应用中所有公开可用的方法。

英文摘要:

The adoption of large language models (LLMs) in many applications, from customer service chat bots and software development assistants to more capable agentic systems necessitates research into how to secure these systems. Attacks like prompt injection and jailbreaking attempt to elicit responses and actions from these models that are not compliant with the safety, privacy, or content policies of organizations using the model in their application. In order to counter abuse of LLMs for generating potentially harmful replies or taking undesirable actions, LLM owners must apply safeguards during training and integrate additional tools to block the LLM from generating text that abuses the model. Jailbreaking prompts play a vital role in convincing an LLM to generate potentially harmful content, making it important to identify jailbreaking attempts to block any further steps. In this work, we propose a novel approach to detect jailbreak prompts based on pairing text embeddings well-suited for retrieval with traditional machine learning classification algorithms. Our approach outperforms all publicly available methods from open source LLM security applications.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑