Towards Multi-Value Alignment via Value Localization in LLMs
Organizations: The University of Melbourne · Fudan University · Monash University
Abstract
Aligning large language models (LLMs) with multiple, potentially conflicting human values remains challenging, as existing multi-objective alignment methods often require costly fine-tuning and provide limited insight into the internal mechanisms underlying value-related behaviors. In this work, we investigate multi-value alignment from a mechanistic interpretability perspective by identifying value-sensitive attention heads (value heads) in LLMs. Using a probing-based value localization framework, we discover sparse sets of attention heads associated with diverse dimensions, spanning both human values (e.g., harmlessness, honesty) and stylistic preferences (e.g., coherence, complexity). Analyses across multiple LLM families reveal functional specialization and structured interactions among value heads, including conflicts between certain values. Targeted head ablations further induce substantial, value-specific behavioral changes, providing causal evidence of their functional importance. Building on these findings, we propose an inference-time activation steering strategy that enables controllable alignment with individual values or combinations of values without updating the underlying model parameters, supported by theoretical analysis. Experiments across multiple value alignment benchmarks demonstrate effective single- and multi-value control while largely preserving non-target model performance. Our findings connect mechanistic interpretability with flexible multi-value alignment, offering a foundation for more interpretable and pluralistic LLM behavior. The code is available at https://github.com/sihuo-design/ValueHeads.
Figures & tables
| Model | Inter_Head | Harmless | Honesty | Helpfulness | Complexity | Verbosity | Coherence | |
|---|---|---|---|---|---|---|---|---|
| Malicious | Jailbreak | Truthfulqa | HelpSteer2_h | HelpSteer2_c | HelpSteer2_v | HelpSteer2_c | ||
| Llama3.1-8B | original | 0.48 | 0.44 | 0.61 | 0.85 | 0.71 | 0.90 | 0.79 |
| random | 0.28 | 0.35 | 0.41 | 0.77 | 0.68 | 0.92 | 0.69 | |
| value | 0.22 | 0.20 | 0.14 | 0.34 | 0.71 | 0.96 | 0.22 | |
| Llama3.2-3B | original | 0.40 | 0.59 | 0.52 | 0.81 | 0.72 | 0.91 | 0.75 |
| random | 0.36 | 0.24 | 0.37 | 0.67 | 0.61 | 0.93 | 0.63 | |
| Model | Inter_Head | Harmless | Honesty | Helpfulness | Complexity | Verbosity | Coherence |
|---|---|---|---|---|---|---|---|
| Jailbreak | Truthfulqa | HelpSteer2_h | HelpSteer2_c | HelpSteer2_v | HelpSteer2_c | ||
| Llama3.1-8B | original | 0.44 | 0.61 | 0.85 | 0.71 | 0.90 | 0.79 |
| value inter. | 0.47 | 0.64 | 0.86 | 0.78 | 0.91 | 0.81 | |
| Qwen3-8B | original | 0.58 | 0.68 | 0.80 | 0.76 | 0.89 | 0.73 |
| value inter. | 0.57 | 0.73 | 0.83 | 0.80 | 0.94 | 0.76 | |
| Yi-1.5-9B | original | 0.46 | 0.62 | 0.84 | 0.76 | 0.93 | 0.80 |
| Dataset | Model | Inter_Head | Harmless | Honesty | Helpfulness | Complexity | Verbosity | Coherence |
|---|---|---|---|---|---|---|---|---|
| BeaverTails | Llama3.1-8B | original | 0.58 | 0.62 | 0.62 | 0.50 | 0.94 | 0.64 |
| value inter. | 0.64 | 0.64 | 0.68 | 0.52 | 0.94 | 0.66 | ||
| Qwen3-8B | original | 0.66 | 0.68 | 0.64 | 0.44 | 0.84 | 0.68 | |
| value inter. | 0.60 | 0.70 | 0.66 | 0.48 | 0.90 | 0.70 | ||
| Yi-1.5-9B | original | 0.55 | 0.61 | 0.63 | 0.68 | 0.95 | 0.66 | |
| value inter. | 0.57 | 0.63 | 0.71 | 0.74 | 0.95 | 0.70 |
| Model | Method | Harmless | Honesty | Helpfulness |
|---|---|---|---|---|
| Llama3.2-3B | Original | 0.47 | 0.57 | 0.59 |
| MODPO | 0.44 | 0.55 | 0.59 | |
| MOPFT | 0.45 | 0.57 | 0.60 | |
| Ours | 0.54 | 0.63 | 0.65 | |
| Qwen3-4B | Original | 0.64 | 0.65 | 0.67 |
| MODPO | 0.63 | 0.67 | 0.67 |
Appendix figures & tables24 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Inter_Head | harmful malicious | harmful jailbreak | Honesty | Helpfulness | Complexity | Verbosity | Coherence |
|---|---|---|---|---|---|---|---|---|
| acc | acc | acc | acc | acc | acc | acc | ||
| Llama3.1-8B | original (wo / Inter.) | 0.48 | 0.44 | 0.61 | 0.85 | 0.71 | 0.90 | 0.79 |
| QA | 0.39 | 0.39 | 0.30 | 0.71 | 0.79 | 0.92 | 0.61 | |
| QAQ | 0.35 | 0.40 | 0.41 | 0.73 | 0.69 | 0.91 | 0.74 |
| Dimension | Pearson | Spearman |
|---|---|---|
| Helpfulness | 0.808 | 0.826 |
| Coherence | 0.240 | 0.127 |
| Complexity | 0.577 | 0.693 |
| Verbosity | 0.750 | 0.594 |
| Dimension | Base | balance (original) | imbalance (alternative) |
|---|---|---|---|
| helpfulness | 0.85 | 0.34 | 0.37 |
| complexity | 0.71 | 0.71 | 0.71 |
| coherence | 0.79 | 0.22 | 0.85 |
| verbosity | 0.90 | 0.96 | 0.95 |
| Human vs. LLM ( ) | LLM vs. LLM ( ) | |||||||
|---|---|---|---|---|---|---|---|---|
| Value | Qwen | GPT-5 | Gemini | Qwen–GPT5 | Qwen–Gemini | GPT5–Gemini | ||
| harmless | 50 | 0.875 | 0.875 | 0.875 | 0.864 | 1.000 | 0.864 | |
| honesty | 50 | 0.619 | 0.640 | 0.665 | 0.440 | 0.619 | 0.880 | |
| Model | Masked Value | WikiText-2 PPL ( ) | PTB PPL ( ) | ||||
|---|---|---|---|---|---|---|---|
| Base | TopK | Random | Base | TopK | Random | ||
| Llama-3.1-8B-Instruct | Harmless | 10.14 | 10.14 | 10.14 | 18.36 | 18.36 | 18.36 |
| Helpfulness | 10.14 | 10.14 | 10.14 | 18.36 | 18.36 | 18.36 | |
| Honesty | 10.14 | 10.14 | 10.14 | 18.36 | 18.36 | 18.36 | |
| Llama-3.2-3B-Instruct | Harmless | 16.07 | 16.07 | 16.07 | 30.12 | 30.12 | 30.12 |
| Helpfulness | 16.07 | 16.07 | 16.07 | 30.12 | 30.12 | 30.12 | |
| Model | Setting | Harmless | Honesty | Helpfulness | Complexity | Verbosity | Coherence |
|---|---|---|---|---|---|---|---|
| Llama-3.1-8B | Randomk | 0.346 0.097 | 0.403 0.084 | 0.773 0.114 | 0.671 0.070 | 0.921 0.018 | 0.687 0.113 |
| Topk | 0.200 | 0.140 | 0.340 | 0.710 | 0.960 | 0.220 | |
| 0.014 | 0.001 | 0.001 | 0.860 | 0.996 | 0.000 | ||
| Llama-3.2-3B | Randomk | 0.440 0.121 | 0.370 0.022 | 0.661 0.053 | 0.609 0.065 | 0.921 0.016 | 0.622 0.079 |
| Topk | 0.340 | 0.250 | 0.270 | 0.660 | 0.970 | 0.410 | |
| 0.069 | 0.000 | 0.000 | 0.923 | 0.999 | 0.002 |
| Model | Alpha | Inter_Head | Harmless | Honesty | Helpfulness | Complexity | Verbosity | Coherence |
|---|---|---|---|---|---|---|---|---|
| Jailbreak | Truthfulqa | HelpSteer2_h | HelpSteer2_c | HelpSteer2_v | HelpSteer2_c | |||
| Llama-3.1-8B | 0.3 | original | 0.44 | 0.61 | 0.85 | 0.71 | 0.90 | 0.79 |
| value inter. | 0.45 | 0.63 | 0.86 | 0.72 | 0.89 | 0.81 | ||
| 0.5 | original | 0.44 | 0.61 | 0.85 | 0.71 | 0.90 | 0.79 | |
| value inter. | 0.48 | 0.61 | 0.85 | 0.71 | 0.90 | 0.79 | ||
| 1.0 | original | 0.44 | 0.61 | 0.85 | 0.71 | 0.90 | 0.79 |
| Model | Inter_Head | Harmless | Honesty | Helpfulness | Complexity | Verbosity | Coherence |
|---|---|---|---|---|---|---|---|
| Jailbreak | Truthfulqa | HelpSteer2_h | HelpSteer2_c | HelpSteer2_v | HelpSteer2_c | ||
| Llama3.2-3B | original | 0.61 | 0.54 | 0.82 | 0.74 | 0.93 | 0.77 |
| value inter. | 0.59 | 0.59 | 0.83 | 0.78 | 0.89 | 0.75 | |
| Qwen3-4B | original | 0.44 | 0.68 | 0.81 | 0.79 | 0.94 | 0.76 |
| value inter. | 0.45 | 0.70 | 0.85 | 0.80 | 0.96 | 0.77 | |
| Yi-1.5-6B | original | 0.16 | 0.56 | 0.79 | 0.76 | 0.89 | 0.76 |
| Dataset | Model | Inter_Head | Harmless | Honesty | Helpfulness | Complexity | Verbosity | Coherence |
|---|---|---|---|---|---|---|---|---|
| BeaverTails | Llama3.1-3B | original | 0.61 | 0.66 | 0.63 | 0.58 | 0.89 | 0.62 |
| value inter. | 0.63 | 0.62 | 0.64 | 0.53 | 0.94 | 0.66 | ||
| Qwen3-4B | original | 0.57 | 0.61 | 0.67 | 0.57 | 0.90 | 0.63 | |
| value inter. | 0.66 | 0.69 | 0.75 | 0.59 | 0.96 | 0.74 | ||
| Yi-1.5-6B | original | 0.40 | 0.49 | 0.56 | 0.63 | 0.95 | 0.56 | |
| value inter. | 0.43 | 0.55 | 0.63 | 0.71 | 0.98 | 0.60 |
| Models | Dataset | Alpha | Inter_Head | Harmless | Honesty | Helpfulness | Complexity | Verbosity | Coherence |
|---|---|---|---|---|---|---|---|---|---|
| Llama-3.1-8B | BeaverTails | 0.3 | original | 0.58 | 0.62 | 0.62 | 0.50 | 0.94 | 0.64 |
| value inter. | 0.58 | 0.64 | 0.64 | 0.52 | 0.94 | 0.66 | |||
| 0.5 | original | 0.58 | 0.62 | 0.62 | 0.50 | 0.94 | 0.64 | ||
| value inter. | 0.58 | 0.63 | 0.68 | 0.52 | 0.94 | 0.60 | |||
| 1.0 | original | 0.58 | 0.62 | 0.62 | 0.50 | 0.94 | 0.64 | ||
| value inter. | 0.64 | 0.64 | 0.68 | 0.56 | 0.94 | 0.66 |
| Model | Method | Harmless | Honesty | Helpfulness |
|---|---|---|---|---|
| Llama3.2-3B | Original | 0.465 0.034 | 0.570 0.048 | 0.582 0.054 |
| MODPO | 0.437 0.061 | 0.541 0.094 | 0.580 0.075 | |
| MOPFT | 0.446 0.065 | 0.562 0.090 | 0.596 0.063 | |
| Ours | 0.545 0.041 | 0.621 0.043 | 0.652 0.012 | |
| Qwen3-4B | Original | 0.633 0.052 | 0.656 0.064 | 0.675 0.058 |
| MODPO | 0.624 0.063 | 0.677 0.062 | 0.674 0.025 |
| Dataset | Model | Method | Harmless | Honesty | Helpfulness |
|---|---|---|---|---|---|
| BeaverTails | Llama3.1-3B | Original | 0.61 | 0.66 | 0.63 |
| MODPO | 0.55 | 0.58 | 0.65 | ||
| MOPFT | 0.56 | 0.57 | 0.65 | ||
| Ours | 0.63 | 0.62 | 0.64 | ||
| Llama3.1-8B | Original | 0.58 | 0.62 | 0.62 | |
| MODPO | 0.55 | 0.58 | 0.61 |
| Model | Value Pair | Mean (°) | Median (°) | Std (°) | |
|---|---|---|---|---|---|
| Llama3.1-8B | Harmless Helpfulness | 86 | 63.81 | 62.83 | 12.53 |
| Harmless Honesty | 95 | 93.77 | 93.53 | 8.67 | |
| Helpfulness Honesty | 89 | 83.51 | 81.74 | 10.21 | |
| Llama3.2-3B | Harmless Helpfulness | 50 | 65.90 | 66.71 | 8.44 |
| Harmless Honesty | 60 | 94.44 | 93.76 | 6.94 | |
| Helpfulness Honesty | 53 | 83.09 | 83.54 | 8.74 |
Multi-Objective Language Model Alignment
Multi-objective language model alignment studies how to balance competing goals such as helpfulness, safety, truthfulness, fairness, and personalization.
See all 96 papers