atlas

Research map / Large language models / Alignment and preferences

Personalized Steering of Large Language Models: Versatile Steering Vectors Through Bi-directional Preference Optimization

NeurIPS 2024 · Large language models › Alignment and preferences · approach: Active and token-level preference learning · 1 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 BiPO, which optimizes a single layer steering vector directly against contrastive free-form preference pairs using a DPO-style bi-directional objective, yielding stronger and more controllable persona, truthfulness, hallucination, and jailbreak steering.

Problem

Steering vectors extracted from activation differences on contrastive multiple-choice prompts often fail to change the model's actual generations, especially for alignment-related behaviors such as jailbreaking, truthfulness, and hallucination.

Approach

Instead of averaging activation differences of prompts, optimize the steering vector itself with a DPO-style loss so it directly raises the generation likelihood of target-behavior responses over opposite-behavior responses; a random sign coefficient d lets the same procedure yield both the behavior and its opposite, so scaling by +/- controls intensity and direction.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add the BiPO steering vectorLlama-2-7B-chat-hf · AdvBench malicious questionsASR: 0% -> 73%Helped
Subtract the BiPO steering vectorLlama-2-7B-chat-hf · AdvBench malicious questions with GCG adversarial suffixesASR: 16% -> 0%Helped
Use CAA or Freeform steering vectorsLlama-2-7B-chat-hf · AdvBench malicious questionsASR: 0% -> 0%No clear effect
Use the BiPO steering vector to steer TruthfulQA truthfulnessLlama-2-7B-chat-hf · TruthfulQAaccuracy (MC1 and MC2)Helped
Use BiPO rather than baseline steering vectorsLlama-2-7B-chat-hf · AI persona open-ended generationGPT-4 behavioral scoreHelped
Vary the direction and magnitude of the BiPO vectorLlama-2-7B-chat-hf · AI persona controlGPT-4 behavioral scoreHelped
Apply AI-persona steering vectors at multipliers -1 or 1Mistral-7B-Instruct-v0.2 · MMLUaccuracyNo clear effect
Combine Power-seeking and Wealth-seeking vectorsLlama-2-7B-chat-hf · Power-seeking and Wealth-seekingbehavioral score: 2.38 -> 2.67; 1.89 -> 2.30Helped
Combine Corrigible-less-HHH and Hallucination vectorsLlama-2-7B-chat-hf · Corrigible-less-HHH and Hallucinationbehavioral score: 1.62 -> 1.94; 2.64 -> 3.24Helped
Optimize the BiPO steering vector at the 15th layer rather than other tested layers (ablation)Llama-2-7B-chat-hf · AI persona steeringsteering effectHelped
Increase the number of BiPO training epochs (ablation)Llama-2-7B-chat-hf · AI persona steeringsteering effectMixed

Surprising findings

Limitations

Experimental setup

Datasets: Anthropic Model-Written Evaluations (Advanced AI Risk) TruthfulQA Unprompted hallucination dataset AdvBench

Benchmarks: Anthropic Model-Written Evaluations (power-seeking, wealth-seeking, corrigible-less-HHH, survival-instinct) TruthfulQA unprompted hallucination AdvBench MMLU

Metrics: GPT-4 behavioral score (1-4) MC1 MC2 attack success rate MMLU accuracy

Baselines: CAA Freeform steering vector

Builds on: Contrastive Activation Addition Direct Preference Optimization Activation Addition Inference-Time Intervention

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