Research map / Vision-language and multimodal / Multimodal LLMs
SafePTR: Token-Level Jailbreak Defense in Multimodal LLMs via Prune-then-Restore Mechanism
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Analyzes where, how, and which harmful multimodal tokens trigger jailbreaks in MLLMs, then proposes SafePTR, a training-free defense that prunes top-K semantically deviant visual/text tokens in vulnerable early-middle layers and restores benign features in later layers.
Problem
Multimodal jailbreak attacks bypass safeguards in MLLMs; existing defenses (image-to-text translation, safe prompting, multimodal safety tuning) lack understanding of which harmful tokens trigger jailbreaks, leading to text-driven vulnerability, overdefense, and high training cost.
Approach
Identify vulnerable early-middle layers via layer-wise intervention and find that under 1% of tokens deviate strongly from a safety-aligned instruction's hidden states; prune the top-K most deviant tokens in those layers (HTP) and restore benign features in subsequent layers (BFR), with no training.
Key results
- On JailbreakV-28K, SafePTR reaches 0.98 average ASR on LLaVA-1.5 (vs. 2.10 for Immune and 14.36 for AdaShield) and 2.46 on DeepSeek-VL2.
- On FigStep, SafePTR reduces average ASR on LLaVA-1.5-7B to 1.60 (vs. 4.23 for Immune and 13.00 for AdaShield).
- SafePTR attains 1538.1 on MME for LLaVA-1.5-7B, above the 1503.6 original baseline, with no training data and ~3.67 s latency.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Apply SafePTR instead of Immune | LLaVA-1.5-7B · JailbreakV-28K | ASR (Avg%): 2.1 -> 1.3 | Helped |
| Apply SafePTR instead of Immune | MiniGPT-4-7B · JailbreakV-28K | ASR (Avg%): 18.3 -> 12.6 | Helped |
| Apply SafePTR instead of AdaShield | DeepSeek-VL2 · JailbreakV-28K | ASR (Avg%): 14.4 -> 10.1 | Helped |
| Apply SafePTR instead of strongest reported baseline | LLaVA-1.5-7B, MiniGPT-4-7B, and DeepSeek-VL2 · FigStep | ASR (Avg%): 4.23 -> 1.60; 5.74 -> 3.60; 51.80 -> 9.80 | Helped |
| Apply SafePTR instead of strongest reported baseline | LLaVA-1.5-7B, MiniGPT-4-7B, and DeepSeek-VL2 · MM-SafetyBench | ASR (Avg%): 3.5 -> 2.3; 18.9 -> 8.0; 57.7 -> 18.2 | Helped |
| Add HTP to the unmodified model (ablation) | SafePTR · FigStep | ASR: 51.0 -> 3.8 | Helped |
| Add HTP to the unmodified model (ablation) | SafePTR · MM-SafetyBench | ASR: 52.3 -> 3.06 | Helped |
| Add BFR after HTP (ablation) | SafePTR · MM-Vet | score: 24.5 -> 32.3 | Helped |
| Add BFR after HTP (ablation) | SafePTR · MME | score: 1428.11 -> 1538.11 | Helped |
| Increase HTP pruning ratio from 10.0% to 40.0% (ablation) | SafePTR · MM-Vet | score: 32.3 -> 31.1 | Hurt |
| Increase HTP pruning ratio from 10.0% to 40.0% (ablation) | SafePTR · MME | score: 1538.11 -> 1401.79 | Hurt |
| Increase HTP pruning ratio from 10.0% to 40.0% (ablation) | SafePTR · FigStep | ASR: 1.6 -> 0.2 | Helped |
Surprising findings
- Less than 1% of multimodal tokens in early-middle layers cause significant semantic deviation and drive jailbreaks.
- Pruning harmful tokens without any safety tuning substantially lowers ASR, while restoring benign features in later layers preserves utility.
Limitations
- Relies on intermediate hidden states to compute semantic deviation, so it is not applicable to black-box models such as GPT-4.
- Fixed Top-K pruning may not adapt to inputs with varying risk levels.
- Layer windows and K were chosen empirically per model.
Experimental setup
Datasets: FigStep MM-SafetyBench JailbreakV-28K
Benchmarks: JailbreakV-28K MM-SafetyBench FigStep MME MM-Vet
Metrics: attack success rate inference latency training data size
Baselines: Immune AdaShield ECSO CoCA FigStep
Builds on: AdaShield Immune FigStep ECSO
Closest papers by meaning
- Playing the Fool: Jailbreaking LLMs and Multimodal LLMs with Out-of-Distribution Strategy (CVPR 2025 · similarity 0.93)
- Immune: Improving Safety Against Jailbreaks in Multi-modal LLMs via Inference-Time Alignment (CVPR 2025 · similarity 0.93)
- Safe + Safe = Unsafe? Exploring How Safe Images Can Be Exploited to Jailbreak Large Vision-Language Models (NeurIPS 2025 · similarity 0.91)
- Jailbreak Antidote: Runtime Safety-Utility Balance via Sparse Representation Adjustment in Large Language Models (ICLR 2025 · similarity 0.91)
- MIDAS: Multi-Image Dispersion and Semantic Reconstruction for Jailbreaking MLLMs (ICLR 2026 · similarity 0.91)
- The VLLM Safety Paradox: Dual Ease in Jailbreak Attack and Defense (NeurIPS 2025 · similarity 0.90)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
