用一个模型实现多种人类偏好对齐,无需额外训练。
MOSLIM:Align with diverse preferences in prompts through reward classification
- 用多头奖励模型分类问答对,生成可优化的标量奖励。
- 在多个基准上优于现有方法,且显存消耗更低。
- 适合快速部署多目标对齐,无需为每种偏好单独训练。
大型语言模型的多目标对齐对确保基础模型符合多样化人类偏好至关重要。当前研究通常依赖多个策略或奖励模型,或需为特定偏好训练监督微调(SFT)模型。本文提出一种新方法MOSLIM,仅用单一奖励模型和策略模型即可应对多种目标。MOSLIM通过提示灵活控制对齐目标,且在SFT阶段无需偏好训练,可直接使用数千个现成模型。该方法采用多头奖励模型对问答对进行分类,而非打分,并通过映射函数将分类结果转化为标量奖励以优化策略模型。我们在多个多目标基准上验证了该方法的有效性,并进行了不同奖励模型规模与策略优化方法的消融实验。MOSLIM在多数结果中超越现有方法,同时显著降低GPU计算资源需求。
原文摘要 · Abstract (English)
The multi-objective alignment of Large Language Models (LLMs) is essential for ensuring foundational models conform to diverse human preferences. Current research in this field typically involves either multiple policies or multiple reward models customized for various preferences, or the need to train a preference-specific supervised fine-tuning (SFT) model. In this work, we introduce a novel multi-objective alignment method, MOSLIM, which utilizes a single reward model and policy model to address diverse objectives. MOSLIM provides a flexible way to control these objectives through prompting and does not require preference training during SFT phase, allowing thousands of off-the-shelf models to be directly utilized within this training framework. MOSLIM leverages a multi-head reward model that classifies question-answer pairs instead of scoring them and then optimize policy model with a scalar reward derived from a mapping function that converts classification results from reward model into reward scores. We demonstrate the efficacy of our proposed method across several multi-objective benchmarks and conduct ablation studies on various reward model sizes and policy optimization methods. The MOSLIM method outperforms current multi-objective approaches in most results while requiring significantly fewer GPU computing resources compared with existing policy optimization methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。