通过增加推理计算提升模型对抗攻击的鲁棒性,关键在于训练时已有一定抗干扰能力。
Get RICH or Die Scaling: Profitably Trading Inference Compute for Robustness
- 利用推理阶段额外计算增强模型对对抗性数据的响应能力
- 仅当模型具备基础鲁棒性时,增加推理算力才能有效防御白盒多模态攻击
- 适合关注模型安全性和可信赖推理的开发者与研究者
测试时推理已被证明能提升基准性能,并在解决模型对对抗性分布外(OOD)数据脆弱性的长期难题上展现潜力。近期工作表明,推理有助于满足旨在抵御攻击的模型规范,且大型语言模型的推理努力与抵抗越狱攻击的能力存在显著相关性。然而,面对更强的攻击(如基于梯度或多模态的攻击),这种优势会减弱。这可能是因为模型无法在这些攻击生成的对抗性OOD数据上正确理解指令,而遵循指令正是执行防攻击规范的前提。因此,我们提出‘推理算力带来的鲁棒性假说’(RICH):推理算力防御效果随训练数据对攻击数据成分的覆盖程度提高而增强。基于此,我们测试了不同初始鲁棒性水平的模型,发现只要模型具备足够初始鲁棒性,即使在白盒多模态攻击下,增加推理算力也能提升鲁棒性。进一步验证了‘强者愈强’的动态:InternVL 3.5 gpt-oss 20B 在未强化视觉编码器时,增加测试算力几乎无增益;但若先强化其视觉编码器(从而构建首个对抗性鲁棒的多模态推理模型),则测试算力扩展带来显著鲁棒性提升。强化使攻击数据的组成部分更接近分布内(ID),RICH认为这促进了组合泛化——通过理解数据的分布内组件来应对分布外数据并遵循规范指令。结果一致表明,测试时防御既依赖也强化训练时的数据与防御策略。
原文摘要 · Abstract (English)
Test-time reasoning has raised benchmark performances and even shown promise in addressing the historically intractable problem of making models robust to adversarially out-of-distribution (OOD) data. Indeed, recent work used reasoning to aid satisfaction of model specifications designed to thwart attacks, finding a striking correlation between LLM reasoning effort and robustness to jailbreaks. However, this benefit fades when stronger (e.g. gradient-based or multimodal) attacks are used. This may be expected as models often can't follow instructions on the adversarially OOD data created by such attacks, and instruction following is needed to act in accordance with the attacker-thwarting spec. Thus, we hypothesize that the test-time robustness benefits of specs are unlocked by initial robustness sufficient to follow instructions on OOD data. Namely, we posit the Robustness from Inference Compute Hypothesis (RICH): inference-compute defenses profit as the model's training data better reflects the components of attacked data. Guided by the RICH, we test models of varying initial-robustness levels, finding inference-compute adds robustness even to white-box multimodal attacks, provided the model has sufficient initial robustness. Further evidencing a rich-get-richer dynamic, InternVL 3.5 gpt-oss 20B gains little robustness when its test compute is scaled, but such scaling adds significant robustness if we first robustify its vision encoder (creating the first adversarially robust reasoning VLM in the process). Robustifying models makes attacked components of data more in-distribution (ID), and the RICH suggests this fuels compositional generalization -- understanding OOD data via its ID components -- to following spec instructions on adversarial data. Consistently, we find test-time defenses both build and depend on train-time data and defenses.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。