arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.04497cs.LGcs.AIcs.CL
收藏

一个模型解决所有问题:多目标可控语言模型

One Model for All: Multi-Objective Controllable Language Models

Ruhr University Bochum(波鸿鲁尔大学);Eindhoven University of Technology(埃因霍温理工大学);Mohamed bin Zayed University of Artificial Intelligence(穆罕默德·本·扎耶德人工智能大学);University of Liverpool(利物浦大学)

Qiang He, Yucheng Yang, Tianyi Zhou, Meng Fang, Mykola Pechenizkiy, Setareh Maghsudi

AI总结:

本文提出多目标控制(MOC),通过引入多目标优化原理训练单个语言模型,使其能根据用户偏好在帕累托前沿生成个性化输出,提升模型可控性、输出质量和泛化能力。

摘要:

对齐大型语言模型(LLMs)与人类偏好对于增强LLMs的安全性、有用性、幽默感、忠实度等至关重要。当前基于人类反馈的强化学习(RLHF)主要关注固定奖励,可能削弱适应性和可控性。然而,创建个性化LLMs需要对齐LLMs与个体人类偏好,这因用户数据稀缺和多目标权衡的多样性而具有挑战性。本文提出多目标控制(MOC),通过引入多目标优化原理训练单个LLM作为偏好条件政策网络,提升计算效率,使7B参数模型在单块A6000 GPU上微调。实验表明,MOC在可控性、输出质量和泛化能力方面优于基线方法。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑