CLAILGNov 18, 2024

CROW: Eliminating Backdoors from Large Language Models via Internal Consistency Regularization

arXiv:2411.12768v222 citationsh-index: 12ICML
Originality Highly original
AI Analysis

This addresses a critical security vulnerability in LLMs for practical deployment, representing a novel defense for text generation tasks rather than an incremental improvement.

The paper tackles the problem of backdoor attacks in large language models (LLMs) by proposing Internal Consistency Regularization (CROW), which eliminates backdoors by enforcing consistency across layers during finetuning, achieving significant reductions in attack success rates across diverse strategies while preserving generative performance.

Large Language Models (LLMs) are vulnerable to backdoor attacks that manipulate outputs via hidden triggers. Existing defense methods--designed for vision/text classification tasks--fail for text generation. We propose Internal Consistency Regularization (CROW), a defense leveraging the observation that backdoored models exhibit unstable layer-wise hidden representations when triggered, while clean models show smooth transitions. CROW enforces consistency across layers via adversarial perturbations and regularization during finetuning, neutralizing backdoors without requiring clean reference models or trigger knowledge--only a small clean dataset. Experiments across Llama-2 (7B, 13B), CodeLlama (7B, 13B), and Mistral-7B demonstrate CROW's effectiveness: it achieves significant reductions in attack success rates across diverse backdoor strategies (sentiment steering, targeted refusal, code injection) while preserving generative performance. CROW's architecture-agnostic design enables practical deployment.

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