词典对齐的概念控制:面向多模态大语言模型的安全防护
Dictionary-Aligned Concept Control for Safeguarding Multimodal LLMs
对做多模态视觉语言模型的人,这篇给出了一个推理时无需重训的激活干预方案:用图像-描述刺激构建的概念词典初始化 SAE,并对 SAE 原子自动语义标注,从而在冻结模型上做细粒度概念级引导,可直接迁移到视觉模型的安全对齐或概念编辑。
原文依据 we introduce Dictionary-Aligned Concept Control (DACO), a framework that utilizes a curated concept dictionary and a Sparse Autoencoder (SAE)
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论多模态大语言模型(MLLM)已被证明容易受到恶意查询的影响,从而产生不安全的回答。近期工作采用提示词工程、回复分类或微调来提升 MLLM 的安全性。然而,这些方法往往难以应对不断演化的恶意模式,可能需要重新运行查询,或需要大量计算资源。在推理阶段对冻结模型的激活进行引导,近来已成为一种灵活且有效的解决方案。但是,现有的 MLLM 引导方法通常只处理范围狭窄的一类安全相关概念,或者难以在调整特定概念的同时不影响其他概念。为应对这些挑战,我们提出词典对齐的概念控制(DACO),该框架利用一个精心构建的概念词典与稀疏自编码器(SAE),对 MLLM 的激活提供细粒度控制。首先,我们通过检索超过 40 万条描述-图像刺激,并将其激活汇总为概念方向,构建了一个包含 1.5 万个多模态概念的词典,并将该数据集命名为 DACO-400K。其次,我们表明所构建的词典可通过稀疏编码用于干预激活。第三,我们提出一种新的引导方法,利用我们的词典初始化 SAE 的训练,并自动标注 SAE 原子的语义,以用于 MLLM 的安全防护。在多个 MLLM(如 QwenVL、LLaVA、InternVL)以及多个安全基准(如 MM-SafetyBench、JailBreakV)上的实验表明,DACO 在保持通用能力的同时显著提升了 MLLM 的安全性。
英文原文摘要 1611 字符
Multimodal Large Language Models (MLLMs) have been shown to be vulnerable to malicious queries that can elicit unsafe responses. Recent work uses prompt engineering, response classification, or finetuning to improve MLLM safety. Nevertheless, such approaches are often ineffective against evolving malicious patterns, may require rerunning the query, or demand heavy computational resources. Steering the activations of a frozen model at inference time has recently emerged as a flexible and effective solution. However, existing steering methods for MLLMs typically handle only a narrow set of safety-related concepts or struggle to adjust specific concepts without affecting others. To address these challenges, we introduce Dictionary-Aligned Concept Control (DACO), a framework that utilizes a curated concept dictionary and a Sparse Autoencoder (SAE) to provide granular control over MLLM activations. First, we curate a dictionary of 15,000 multimodal concepts by retrieving over 400,000 caption-image stimuli and summarizing their activations into concept directions. We name the dataset DACO-400K. Second, we show that the curated dictionary can be used to intervene activations via sparse coding. Third, we propose a new steering approach that uses our dictionary to initialize the training of an SAE and automatically annotate the semantics of the SAE atoms for safeguarding MLLMs. Experiments on multiple MLLMs (e.g., QwenVL, LLaVA, InternVL) across safety benchmarks (e.g., MM-SafetyBench, JailBreakV) show that DACO significantly improves MLLM safety while maintaining general-purpose capabilities.