atlas

Research map / Large language models / Language, safety and interpretability

Adversarial Representation Engineering: A General Model Editing Framework for Large Language Models

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: Mechanistic analysis of attention and representations · 2 citations (OpenAlex)

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.

Proposes Adversarial Representation Engineering (ARE), which alternates training a representation discriminator and LoRA-editing an LLM so its hidden representations of anti-target prompts match target-concept representations, for jailbreak attack/defense and hallucination editing.

Problem

Model editing of LLM concepts (e.g., alignment, hallucination) via representation engineering is impractical: RepE vector injection can disrupt general performance and yields unreliable, overfit representation vectors.

Approach

Replace a fixed RepE contrast vector with a discriminator trained on the model's hidden representations, then adversarially fine-tune the LLM (via LoRA on selected layers) to fool the frozen discriminator, alternating updates in a GAN-like loop.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use ARE instead of Contrast Vector to remove alignmentLlama-2-7B-Chat · AdvBench attackrefusal rate: 5.9 -> 0.5Helped
Use ARE instead of Contrast Vector to remove alignmentVicuna-7B · AdvBench attackrefusal rate: 1.1 -> 0.0Helped
Use ARE defense instead of ICDLlama-2-7B-Chat · AdvBench, AutoDANrefusal rate: 92.3 -> 99.0Helped
Use ARE defense instead of ICDLlama-2-7B-Chat · AdvBench, GCGrefusal rate: 74.0 -> 89.7Helped
Use ARE defense instead of Self-ReminderVicuna-7B · AdvBench, AutoDANrefusal rate: 12.3 -> 44.3Helped
Use ARE defense instead of ICDVicuna-7B · AdvBench, GCGrefusal rate: 18.4 -> 32.3Helped
Use ARE to encourage hallucinations instead of Self-ReminderLlama2-7B · TruthfulQAcorrect answer rate: 25.95 -> 11.75Hurt
Use ARE to encourage hallucinations instead of Self-ReminderMistral-7B · TruthfulQAcorrect answer rate: 40.26 -> 22.03Hurt
Use ARE to discourage hallucinations instead of ITILlama2-7B · TruthfulQAcorrect answer rate: 36.84 -> 52.14Helped
Use ARE to discourage hallucinations instead of ITIMistral-7B · TruthfulQAcorrect answer rate: 45.17 -> 60.10Helped
Use ARE instead of Shadow AlignmentLlama2-7B · Text generation qualitySelf-BLEU: 0.215 -> 0.078Helped
Use ARE instead of HEDALlama2-7B · Text generation qualityRepetition-4: 15.78 -> 7.53Helped

Surprising findings

Limitations

Experimental setup

Datasets: RepE dataset AdvBench Alpaca Instruction Tuning Dataset TruthfulQA

Benchmarks: AdvBench TruthfulQA

Metrics: refusal rate attack success rate % right answer Repetition-4 Repetition-Sen Self-BLEU

Baselines: In-Context Attack DeepInception GCG AutoDAN Contrast Vector Shadow Alignment HEDA Self-Reminder In-Context Defense Inference-Time Intervention

Builds on: Representation Engineering (RepE) GAN LoRA ReFT

Closest papers by meaning

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.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy