Activation Addition: Steering Language Models Without Optimization
Turner, Alexander Matt, Thiergart, Lisa, Udell, David, Leech, Gavin, Mini, Ulisse, MacDiarmid, Monte
–arXiv.org Artificial Intelligence
Reliably controlling the behavior of large language models is a pressing open problem. Existing methods include supervised finetuning, reinforcement learning from human feedback, prompt engineering and guided decoding. We instead investigate activation engineering: modifying activations at inference-time to predictably alter model behavior. We bias the forward pass with a 'steering vector' implicitly specified through natural language. Past work learned these steering vectors; our Activation Addition (ActAdd) method instead computes them by taking the activation differences which result from pairs of prompts. We demonstrate ActAdd on GPT-2 on OpenWebText and ConceptNet, and replicate the effect on Llama-13B and GPT-J-6B. Our approach yields inference-time control over high-level properties of output & preserves performance on off-target topics. The method requires far less compute and implementation effort than finetuning and RLHF, allows for natural language specification by users, and its overhead scales naturally with model size.
arXiv.org Artificial Intelligence
Nov-13-2023
- Country:
- Asia > Indonesia (0.04)
- North America
- United States
- New York (0.04)
- New Jersey (0.04)
- Louisiana > Orleans Parish
- New Orleans (0.04)
- Illinois > Cook County
- Chicago (0.04)
- Hawaii > Honolulu County
- Honolulu (0.04)
- Canada > British Columbia
- United States
- Europe
- Holy See (0.04)
- Ireland > Leinster
- County Dublin > Dublin (0.04)
- France > Île-de-France
- Denmark > Capital Region
- Copenhagen (0.04)
- Genre:
- Research Report (1.00)
- Industry:
- Technology: