用多智能体对比偏好优化,让弱模型教强模型对齐人类价值观。
MACPO: Weak-to-Strong Alignment via Multi-Agent Contrastive Preference Optimization
- 设计双向学习机制,让弱教师与强学生互相学正向行为。
- 在两个数据集上同时提升强学生和弱教师的对齐效果。
- 适合需要低成本标注、多弱监督源的对齐场景。
随着大语言模型在特定任务上接近甚至超越人类表现,将其对齐于人类价值观变得愈发紧迫。当模型能力超过人类时,面临‘弱到强’对齐难题——需用弱教师生成的弱监督信号来对齐强学生模型。现有方法主要针对强到弱或自对齐场景,难以适用于更难的弱到强设置。为此,我们提出多智能体对比偏好优化(MACPO)框架。MACPO通过迭代强化不熟悉的行为正向性,同时惩罚熟悉的行为负向性,实现弱教师与强学生之间的相互学习。我们设计了互为正向行为增强策略,促进双方从彼此的正面行为中学习,并在下一迭代中提供更高质量的正向行为;同时提出硬负向行为构建策略,通过在负向行为数据上微调,引导两者生成熟悉的负向行为。在HH-RLHF和PKU-SafeRLHF数据集上的实验结果表明,无论使用自动指标还是人工评估,MACPO均能同时提升强学生与弱教师的对齐性能。且随着弱教师数量增加,通过更多迭代优化轮次,弱到强对齐性能进一步提升。
原文摘要 · Abstract (English)
As large language models (LLMs) are rapidly advancing and achieving near-human capabilities on specific tasks, aligning them with human values is becoming more urgent. In scenarios where LLMs outperform humans, we face a weak-to-strong alignment problem where we need to effectively align strong student LLMs through weak supervision generated by weak teachers. Existing alignment methods mainly focus on strong-to-weak alignment and self-alignment settings, and it is impractical to adapt them to the much harder weak-to-strong alignment setting. To fill this gap, we propose a multi-agent contrastive preference optimization (MACPO) framework. MACPO facilitates weak teachers and strong students to learn from each other by iteratively reinforcing unfamiliar positive behaviors while penalizing familiar negative ones. To get this, we devise a mutual positive behavior augmentation strategy to encourage weak teachers and strong students to learn from each other's positive behavior and further provide higher quality positive behavior for the next iteration. Additionally, we propose a hard negative behavior construction strategy to induce weak teachers and strong students to generate familiar negative behavior by fine-tuning on negative behavioral data. Experimental results on the HH-RLHF and PKU-SafeRLHF datasets, evaluated using both automatic metrics and human judgments, demonstrate that MACPO simultaneously improves the alignment performance of strong students and weak teachers. Moreover, as the number of weak teachers increases, MACPO achieves better weak-to-strong alignment performance through more iteration optimization rounds.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。