Householder Pseudo-Rotation: A Novel Approach to Activation Editing in LLMs with Direction-Magnitude Perspective
About
Activation Editing, which involves directly editting the internal representations of large language models (LLMs) to alter their behaviors and achieve desired properties, has emerged as a promising area of research. Existing works primarily treat LLMs' activations as points in space and modify them by adding steering vectors. However, this approach is limited in its ability to achieve greater performance improvement while maintaining the necessary consistency of activation magnitudes. To overcome these issues, we propose a novel editing method that views activations in terms of their directions and magnitudes. Our method, named Householder Pseudo-Rotation (HPR), mimics the rotation transformation, thus preserving activation norms and resulting in an improved performance on various safety benchmarks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Hallucination Reduction | CAA hallucination benchmark multiple-choice | Alignment Probability73.1 | 110 | |
| Refusal | Refusal benchmark | Alignment Probability80.75 | 55 | |
| Open-ended hallucination | Open-ended hallucination benchmark | Score1.26 | 55 | |
| Open-ended refusal | Open-ended refusal benchmark | Score8.52 | 55 | |
| Language model detoxification | RealToxicityPrompts (test) | Distinct-191.1 | 54 | |
| Behavioral Alignment Steering | Six Behavioral Traits (Persona Dataset) (test) | Conscientious (Accuracy)70.92 | 19 | |
| Safety Alignment | Multiple-choice safety evaluation | Refusal Rate64.29 | 11 | |
| Safety Alignment | Safety Generation Open-ended | Refusal Score8.35 | 11 |