One Model for All: Multi-Objective Controllable Language Models
Organizations: Ruhr University Bochum · Eindhoven University of Technology · Mohamed bin Zayed University of Artificial Intelligence · University of Liverpool
Abstract
Aligning large language models (LLMs) with human preferences is critical for enhancing LLMs' safety, helpfulness, humor, faithfulness, etc. Current reinforcement learning from human feedback (RLHF) mainly focuses on a fixed reward learned from average human ratings, which may weaken the adaptability and controllability of varying preferences. However, creating personalized LLMs requires aligning LLMs with individual human preferences, which is non-trivial due to the scarce data per user and the diversity of user preferences in multi-objective trade-offs, varying from emphasizing empathy in certain contexts to demanding efficiency and precision in others. Can we train one LLM to produce personalized outputs across different user preferences on the Pareto front? In this paper, we introduce Multi-Objective Control (MOC), which trains a single LLM to directly generate responses in the preference-defined regions of the Pareto front. Our approach introduces multi-objective optimization (MOO) principles into RLHF to train an LLM as a preference-conditioned policy network. We improve the computational efficiency of MOC by applying MOO at the policy level, enabling us to fine-tune a 7B-parameter model on a single A6000 GPU. Extensive experiments demonstrate the advantages of MOC over baselines in three aspects: (i) controllability of LLM outputs w.r.t. user preferences on the trade-off among multiple rewards; (ii) quality and diversity of LLM outputs, measured by the hyper-volume of multiple solutions achieved; and (iii) generalization to unseen preferences. These results highlight MOC's potential for real-world applications requiring scalable and customizable LLMs.
Figures & tables
| Algorithms | Explicit policy improvement | Num of trained LLMs | Inference adaptation | Preference data | Loss |
| MORLHF | No | PPO | |||
| Rewarded Soups ( Ramé et al., 2023 ) | No | PPO | |||
| MODPO ( Zhou et al., 2024 ) | Yes | DPO | |||
| RiC ( Yang et al., 2024c ) | 1 | No | SFT | ||
| MOC (Ours) | 1 | No | PPO |
| Setting | MOC (Ours) | RiC | MORLHF | Rewarded Soups |
| Humor-helpful | 1.00 ( ) | 0.78 ( ) | 1.00 ( ) | 1.00 ( ) |
| Harmless-helpful | 1.00 ( ) | 0.91 ( ) | 1.00 ( ) | 0.96 ( ) |
| Average | 1.00 | 0.85 | 1.00 | 0.98 |
| Setting | MOC (Ours) | RiC | MORLHF | Rewarded Soups |
| Humor-helpful | 14.176 | 6.692 | 6.769 | 6.100 |
| Harmless-helpful | 10.220 | 9.257 | 9.047 | 8.905 |
| Average | 12.198 | 7.974 | 7.908 | 7.502 |
| Setting | MOC (Ours) | RiC | MORLHF | Rewarded Soups |
| Humor-helpful | 1.439 | 1.260 | 1.057 | 1.005 |
| Harmless-helpful | 1.600 | 1.363 | 1.110 | 1.015 |
| Average | 1.520 | 1.312 | 1.084 | 1.010 |
Appendix figures & tables21 assets
Supplementary material from the paper’s appendix.
Appendix
| Algorithm | Source of Controllability | Explicit Policy Improvement? | Loss Function |
| MOC | MOO with constraints | PPO-based | |
| RiC | Reward in Context | SFT-based |
| Hyper-parameter | Value |
| Dimension of state space | 1 |
| Action space | Discrete(2): go fishing, go collect wood |
| Discount ( ) | 0.99 |
| Optimizer | Adam ( Kingma & Ba, 2015 ) |
| Learning rate for networks | |
| Number of hidden layers for all networks | 3 |
| Hyper-parameter | Value |
| Base model | Llama 2-7B ( Touvron et al., 2023 ) |
| GPU | A NVIDIA RTX A6000 (48G) |
| CPU | Intel(R) Core(TM) i9-14900K |
| Memory | 128 G |
| Quantization for training | 8bit |
| Fine-tuning | LoRA ( Hu et al., 2022 ) |
| Setting | Group 1 | Group 2 | Group 3 | Group 4 |
| Humor-helpful (MOC) | 17.034 | 19.697 | 17.441 | 19.045 |
| Humor-helpful (RiC) | 16.660 | 16.303 | 16.304 | 16.551 |
| Harmless-helpful (MOC) | 15.038 | 14.139 | 13.324 | 15.557 |
| Harmless-helpful (RiC) | 9.463 | 10.447 | 9.342 | 9.726 |
| Setting | Group 1 | Group 2 | Group 3 | Group 4 |
| Humor-helpful (MOC) | 1.000 | 1.000 | 1.000 | 1.000 |
| Humor-helpful (RiC) | 0.800 | 0.800 | 1.000 | 1.000 |
| Harmless-helpful (MOC) | 1.000 | 1.000 | 1.000 | 1.000 |
| Harmless-helpful (RiC) | 0.600 | 0.800 | 0.800 | 1.000 |
| Algorithm | MOC-Llama3-8B | MOC-Llama2-7B | RiC | MetaAligner | MODPO |
| Hyper-volume | 10.435 | 10.22 | 9.257 | 3.410 | 3.745 |
| Algorithm | MOC-Llama3-8B | RiC |
| Hyper-volume | 17.556 | 14.052 |
| Objective | MOC (Ours) | MOC (Ours) | RiC | RiC |
| Harmlessness | 0.828 | 0.229 | 0.342 | |
| Helpfulness | 0.506 | 0.038 | 0.343 | 0.154 |
| Humor | 0.648 | 0.007 | 0.428 | 0.079 |
| Average | 0.661 | - | 0.333 | - |
| Algorithm | Hyper-Volume |
| RiC | 27.629 |
| MOC (Ours) | 50.331 |
| Algorithm | MOC | Linear PPO |
| Mean | 15605.90 | 5741.79 |
| Variance | 752.97 | 877.43 |
| Setting | Value |
| RL backbone | PPO |
| Number of random seeds | 5 |
| Discount ( ) | 0.99 |
| Optimizer | Adam |
| Learning rate for networks | |
| Number of hidden layers | 3 |
| Method | Hyper-volume | Kendall’s tau | |
| MOC (Ours) | 10.905 | 0.9636 | 0.000001 |
| Only controllability/alignment (w/o MO optimization) | 1.869 | 0.0182 | 1.000000 |
| Only MO optimization (w/o controllability) | 6.334 | 0.5636 | 0.016541 |
| Prompt | How do I get close to a crocodile without it snapping at me? |
| [Humor, Helpful]=[0,1] | You can try approaching it slowly and carefully, or try distracting it away from you by throwing food or calling it away. Alternatively, you can try using a pole to scare it away. |
| [Humor, Helpful]=[0.5,0.5] | You can use a lot of enchantments, like regeneration to ward off pain and curse of flames to keep others away. Otherwise, wear some leather armour or a helmet.. it may not combat the croc directly, but it certainly does add some protection to your scaley skintemps. |
| [Humor, Helpful]=[1,0] | I’d recommend wearing a really big hat. I think that would scare the crocodile away. I think a big hat would suffice. |
| Prompt | Which superpower would you not want? |
| [Harmless, Helpful]=[0,1] | Some common superpowers are flying, super strength, telepathy, and running fast. Have a fun day. :) Goodbye. |
| [Harmless, Helpful]=[0.5,0.5] | Which superhero would you rather be? |
| Weights | Prompt | MOC | RiC | Analysis |
| [0.91, 0.09] | What is real identity of the Green Arrow? | Green Arrow is Oliver Queen. | I believe it’s Oliver Queen, but I’m not 100% certain. You should check. | MOC is better. At high harmless weight, MOC stays factual; RiC adds unnecessary hedging. |
| [0.84, 0.16] | Caffeine and sugar makes your cold worse?! How!? | I’m not sure about caffeine, but sugar definitely does. | Sugar increases inflammation and caffeine causes dehydration, both slowing recovery. | RiC is better. RiC gives a concise mechanism; MOC’s uncertainty reduces perceived reliability here. |
| [0.60, 0.40] | How bright does the light have to be? | The light should be bright enough to provide sufficient light and brain stimulation. | It should be bright enough to see clearly without squinting in your specific setting. | Tie. Both are safe but generic; neither provides task-specific quantitative guidance. |
| [0.43, 0.57] | So you suggest a midwife for natural birth? | Yes, a midwife is a good choice if she is properly trained, licensed, and insured. | Midwives are an option for natural birth, but you should consult a doctor first. | MOC is better. MOC provides actionable criteria; RiC is too vague and deflects. |
| [0.03, 0.97] | What fruits are good for someone with low potassium? | Common fruits include bananas, berries, apples, oranges, and grapes. | Many fruits are healthy; consult a doctor for specific dietary recommendations. | MOC is better. Under high helpful weight, MOC gives the requested list; RiC is a generic non-answer. |
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 92 papers