一模型适配多种用户偏好,实现可控个性化生成。
One Model for All: Multi-Objective Controllable Language Models
- 基于多目标优化,在单个模型中直接生成符合不同偏好权衡的输出。
- 在7B参数模型上用单张A6000 GPU完成训练,效率显著提升。
- 支持未见过的偏好,适合需要灵活定制的场景。
大语言模型(LLM)与人类偏好的对齐对提升安全性、帮助性、幽默感、忠实度等至关重要。现有基于人类反馈的强化学习(RLHF)主要依赖平均人类评分学习固定奖励,难以适应多样化的个体偏好。个性化需求面临每用户数据稀缺和多目标权衡差异的挑战,例如某些情境强调共情,另一些则追求效率与精准。本文提出多目标控制(MOC),训练单一模型直接生成位于帕累托前沿偏好区域的响应。MOC将多目标优化(MOO)引入RLHF,构建偏好条件化的策略网络。通过在策略层应用MOO,大幅提升计算效率,可在单张A6000 GPU上微调7B参数模型。大量实验表明,相比基线,MOC在三方面具有优势:(i)对多奖励权衡的输出可控性;(ii)输出质量与多样性,以多解集超体积衡量;(iii)对未见偏好的泛化能力。结果凸显其在可扩展、可定制化真实应用场景中的潜力。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。