CLJan 30, 2024

Weak-to-Strong Jailbreaking on Large Language Models

BerkeleyCMUTsinghua
arXiv:2401.17256v5117 citationsh-index: 60Has CodeICML
Originality Incremental advance
AI Analysis

This exposes an urgent safety issue in aligning LLMs, with potential broad impact on AI security, though it is incremental as it builds on existing jailbreaking methods.

The paper tackles the problem of jailbreaking large language models to produce harmful text by proposing an efficient inference-time attack that increases misalignment rates to over 99% on two datasets with just one forward pass per example.

Large language models (LLMs) are vulnerable to jailbreak attacks - resulting in harmful, unethical, or biased text generations. However, existing jailbreaking methods are computationally costly. In this paper, we propose the weak-to-strong jailbreaking attack, an efficient inference time attack for aligned LLMs to produce harmful text. Our key intuition is based on the observation that jailbroken and aligned models only differ in their initial decoding distributions. The weak-to-strong attack's key technical insight is using two smaller models (a safe and an unsafe one) to adversarially modify a significantly larger safe model's decoding probabilities. We evaluate the weak-to-strong attack on 5 diverse open-source LLMs from 3 organizations. The results show our method can increase the misalignment rate to over 99% on two datasets with just one forward pass per example. Our study exposes an urgent safety issue that needs to be addressed when aligning LLMs. As an initial attempt, we propose a defense strategy to protect against such attacks, but creating more advanced defenses remains challenging. The code for replicating the method is available at https://github.com/XuandongZhao/weak-to-strong

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes