用大规模对抗训练的视觉模型提升多模态大模型抗干扰能力
Robust-LLaVA: On the Effectiveness of Large-Scale Robust Image Encoders for Multi-modal Large Language Models
- 直接使用已对抗训练过的大型视觉模型,无需额外训练
- 在图像描述和问答任务中实现2倍和1.5倍的鲁棒性提升
- 特别适合对安全性和可靠性要求高的多模态应用
多模态大语言模型在视觉-语言任务中表现优异,但易受视觉对抗扰动影响,导致幻觉、响应操控或绕过安全机制。现有方法通过在ImageNet规模数据上对CLIP视觉编码器进行受限对抗微调来缓解风险,但训练范围有限,制约了鲁棒性与泛化能力。本文探索一种新路径:利用已在大规模数据上完成对抗预训练的现成分类模型。分析表明,(1) 大规模且多样化的对抗预训练使这些模型对各类攻击(从难以察觉的扰动到高级越狱尝试)表现出更优鲁棒性,无需额外训练;(2) 将这些鲁棒模型与多模态大模型端到端整合,可促进语言模块更好地适应鲁棒视觉特征,在复杂推理任务中优于现有插件式方法。在视觉问答、图像描述及越狱攻击等任务上的系统评估显示,采用此类模型的MLLM在保持良好正常性能的同时,显著提升对抗鲁棒性:图像描述任务平均鲁棒性提升2倍,视觉问答任务提升1.5倍,越狱攻击防御效果提升超10%。代码与预训练模型将公开于https://github.com/HashmatShadab/Robust-LLaVA。
原文摘要 · Abstract (English)
Multi-modal Large Language Models (MLLMs) excel in vision-language tasks but remain vulnerable to visual adversarial perturbations that can induce hallucinations, manipulate responses, or bypass safety mechanisms. Existing methods seek to mitigate these risks by applying constrained adversarial fine-tuning to CLIP vision encoders on ImageNet-scale data, ensuring their generalization ability is preserved. However, this limited adversarial training restricts robustness and broader generalization. In this work, we explore an alternative approach of leveraging existing vision classification models that have been adversarially pre-trained on large-scale data. Our analysis reveals two principal contributions: (1) the extensive scale and diversity of adversarial pre-training enables these models to demonstrate superior robustness against diverse adversarial threats, ranging from imperceptible perturbations to advanced jailbreaking attempts, without requiring additional adversarial training, and (2) end-to-end MLLM integration with these robust models facilitates enhanced adaptation of language components to robust visual features, outperforming existing plug-and-play methodologies on complex reasoning tasks. Through systematic evaluation across visual question-answering, image captioning, and jail-break attacks, we demonstrate that MLLMs trained with these robust models achieve superior adversarial robustness while maintaining favorable clean performance. Our framework achieves 2x and 1.5x average robustness gains in captioning and VQA tasks, respectively, and delivers over 10% improvement against jailbreak attacks. Code and pretrained models will be available at https://github.com/HashmatShadab/Robust-LLaVA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。