arXiv:2410.00296cs.LGcs.CR2024-10中稿 · Transactions on Ma…被引 20

利用真实用户未标注提示数据,自动识别恶意指令,提升视觉语言模型安全性。

VLMGuard: Bootstrapping Malicious Prompt Detectors from Unlabeled Vision-Language Prompts in the Wild

  • 基于开放世界未标注提示,自动估算恶意程度并训练分类器。
  • 相比当前最优方法,检测性能平均提升5.39% AUROC。
  • 无需人工标注,可应对真实场景中的提示变化,适合实际部署。

视觉语言模型(VLMs)对图文上下文理解至关重要,但其易受对抗性输入干扰,导致输出不可靠,威胁集成应用的信任度。检测恶意提示是保障VLM生成可靠性的关键。现有挑战在于缺乏大量已标注的良性与恶意数据。为此,本文提出VLMGuard,一种新颖的学习框架,利用开放世界中部署时产生的未标注用户提示进行恶意提示检测。这些未标注提示天然包含良性和恶意信息。为挖掘其价值,我们设计了一种自动化恶意度评估分数,用于区分混合数据中的良性与恶意样本,从而在无监督条件下训练二分类提示检测器。该框架无需额外人工标注,且对真实提示变化具有鲁棒性,具备强灵活性和实用性。大量实验表明,VLMGuard在平均AUROC上较当前最优方法提升5.39%。代码已公开:https://github.com/radiolab-ntu/vlmguard。提醒:论文可能包含不当示例,建议谨慎阅读。

原文摘要 · Abstract (English)

Vision-language Models (VLMs) are essential for contextual understanding of both visual and textual information. However, their vulnerability to adversarially manipulated inputs presents significant risks, leading to compromised outputs and raising concerns about the reliability in VLM-integrated applications. Detecting these malicious prompts is thus crucial for maintaining trust in VLM generations. A major challenge in developing a safeguarding prompt classifier is the lack of a large amount of labeled benign and malicious data. To address the issue, we introduce VLMGuard, a novel learning framework that leverages the unlabeled user prompts in the wild for malicious prompt detection. These unlabeled prompts, which naturally arise when VLMs are deployed in the open world, consist of both benign and malicious information. To harness the unlabeled data, we present an automated maliciousness estimation score for distinguishing between benign and malicious samples within this unlabeled mixture, thereby enabling the training of a binary prompt classifier on top. Notably, our framework does not require extra human annotations and is robust to realistic prompt variations, offering strong flexibility and practicality for real-world applications. Extensive experiments show that VLMGuard achieves superior detection results, improving AUROC by 5.39% on average over the state-of-the-art method. Disclaimer: This paper may contain offensive examples; reader discretion is advised. Code is available at: https://github.com/radiolab-ntu/vlmguard.

提示安全无监督学习视觉语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。