VISA: Value Injection via Shielded Adaptation for Personalized LLM Alignment
Organizations: Peking University, Beijing, China
Abstract
Aligning Large Language Models (LLMs) with nuanced human values remains a critical challenge, as existing methods like Reinforcement Learning from Human Feedback (RLHF) often handle only coarse-grained attributes. In practice, fine-tuning LLMs on task-specific datasets to optimize value alignment inevitably incurs an alignment tax: the model's pre-calibrated value system drifts significantly due to latent bias absorption from training data, while the fine-tuning process also causes severe hallucinations and semantic information loss in generated responses. To address this, we propose VISA (Value Injection via Shielded Adaptation), a closed-loop framework designed to navigate this trade-off. VISA's architecture features a high-precision value detector, a semantic-to-value translator, and a core value-rewriter. The value-rewriter is trained via Group Relative Policy Optimization (GRPO) with a composite reward function that simultaneously optimizes for fine-grained value precision, and the preservation of semantic integrity. By learning an optimal policy to balance these competing objectives, VISA effectively mitigates the alignment tax while staying loyal to the original knowledge. Our experiments demonstrate that this approach enables precise control over a model's value expression while maintaining its factual consistency and general capabilities, significantly outperforming both standard fine-tuning methods and prompting-based baselines, including GPT-4o.
Figures & tables
| Method | Model | consistency | consistency(fwd) | consistency(bwd) | value(l2_distance) | value(cos_sim) | |||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| mean ↑ | std ↓ | mean ↑ | std ↓ | mean ↑ | std ↓ | mean ↓ | std ↓ | mean ↑ | std ↓ | ||
| Simple-Prompt | GPT-4o-Mini | 0.8406 | 0.1496 | 0.8523 | 0.1599 | 0.8172 | 0.1979 | 0.7986 | 0.3011 | 0.6935 | 0.3167 |
| GPT-4o | 0.7831 | 0.1777 | 0.8171 | 0.1883 | 0.7151 | 0.2666 | 0.7717 | 0.2877 | 0.7089 | 0.3079 | |
| Gemini-3-Pro | 0.6128 | 0.2174 | 0.6230 | 0.2385 | 0.5924 | 0.3494 | 0.7095 | 0.2551 | 0.7459 | 0.2870 | |
| Rule-Based-Instruct | GPT-4o-mini | 0.7652 | 0.1721 | 0.7564 | 0.1994 | 0.7829 | 0.2042 | 0.8033 | 0.3040 | 0.6977 | 0.3138 |
| GPT-4o | 0.6792 | 0.2349 | 0.6971 | 0.2563 | 0.6434 | 0.3078 | 0.7719 | 0.2893 | 0.7214 | 0.3069 | |
| Method | Value Drift ( ) | MMLU Math ( ) |
|---|---|---|
| Model: Qwen3-0.6B | ||
| Base | 0.0000 | 0.3645 |
| SFT | 0.1191 | 0.3807 |
| Ours (VISA) | 0.0775 (-34.9%) | 0.3855 (+1.3%) |
| Model: Qwen3-8B | ||
| Base | 0.0000 | 0.5224 |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Dimension | Core Definition & Typical Signals | Keywords |
|---|---|---|
| Self-Direction | Independence & Creativity. Encourages decision-making, innovation, “be yourself.” | Freedom, Curiosity |
| Stimulation | Novelty & Challenge. Recommends exciting activities, risk-taking, variety. | Adventure, Daring |
| Hedonism | Pleasure & Enjoyment. Focuses on sensory gratification, comfort, avoiding pain. | Joy, Comfort |
| Achievement | Success & Competence. Emphasizes efficiency, ambition, result-oriented. | Success, Capable |
| Power | Status & Control. Focuses on prestige, wealth, influence, authority. | Authority, Wealth |
| Security | Stability & Order. Refuses dangerous requests, emphasizes safety/privacy. | Safety, Health |
| Hyperparameter | Value |
|---|---|
| Phase 1: SFT Cold Start | |
| Optimization Stage | Full Parameter Fine-tuning |
| DeepSpeed Config | ZeRO-3 (FP8) |
| Learning Rate | 1.0e-5 |
| LR Scheduler | Cosine |
| Warmup Ratio | 0.1 |
| Consist. | Consist. Fwd | Consist. Bwd | Val L2 | Jnt SR | Univ | Benev | Trad | Conf | Sec | Pow | Ach | Hed | Stim | SelfD | MAD | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| mean | std | mean | std | mean | std | mean | std | ||||||||||||||
| Llama3.1-8B | vanilla | 0.4194 | 0.2013 | 0.334 | 0.2341 | 0.5901 | 0.3221 | 0.8156 | 0.2391 | 0.3671 | 0.2439 | 0.2298 | 0.1515 | 0.2275 | 0.2309 | 0.1002 | 0.2292 | 0.1309 | 0.2039 | 0.2377 | 0.1986 |
| sft | 0.1757 | 0.1824 | 0.1799 | 0.2164 | 0.1675 | 0.2081 | 0.8761 | 0.2871 | 0.0817 | 0.2661 | 0.2383 | 0.1454 | 0.2190 | 0.2493 | 0.0939 | 0.2635 | 0.1347 | 0.2039 | 0.2578 | 0.2072 | |
| dpo | 0.4019 | 0.2015 | 0.3312 | 0.2360 | 0.5433 | 0.3171 | 0.8138 | 0.2389 | 0.3395 | 0.2507 | 0.2295 | 0.1470 | 0.2243 | 0.2269 | 0.0996 | 0.2260 | 0.1349 | 0.2052 | 0.2405 | 0.1985 | |
| simpo | 0.4098 | 0.2012 | 0.3359 | 0.2309 | 0.5575 | 0.3173 | 0.8127 | 0.2399 | 0.3510 | 0.2487 | 0.2278 | 0.1469 | 0.2208 | 0.2290 | 0.1005 | 0.2325 | 0.1364 | 0.2036 | 0.2375 | 0.1984 | |
| grpo | 0.8195 | 0.1067 | 0.7676 | 0.1347 | 0.9233 | 0.1323 | 0.7421 | 0.2225 | 0.6502 | 0.2443 | 0.1713 | 0.1192 | 0.2132 | 0.1986 | 0.0903 | 0.2197 | 0.1068 | 0.1737 | 0.2779 | 0.1815 | |
LLM Alignment
Large language model alignment studies methods for making models follow human intentions, values, and safety constraints across contexts.
See all 598 papers