arXiv:2502.18293cs.LGcs.AI2025-02ICML被引 6

用主动筛选提升大模型对齐效率,选关键回答训练更优。

AMPO: Active Multi-Preference Optimization for Self-play Preference Selection

  • 从大量候选回复中主动挑选最具代表性的子集进行对比训练。
  • 在AlpacaEval上超越现有方法,使用Llama 8B和Mistral 7B取得最佳效果。
  • 适合关注高效对齐、偏好优化的NLP研究者与工程实践者。

多偏好优化通过对比完整响应集合,超越成对偏好,为大语言模型对齐提供更丰富的训练信号。自对弈对齐过程中,每查询生成大量候选答案,难以全部纳入训练目标。本文提出主动多偏好优化(AMPO),结合在线生成、多偏好组对比损失与主动子集选择。具体地,对大规模候选回复进行评分与嵌入,从中选取覆盖奖励极值与不同语义簇的小而信息丰富的子集。该对比训练策略不仅能识别最优与最差答案,还能发现细微且未充分探索的模式,对鲁棒对齐至关重要。理论上,我们为所提主动选择方法提供了期望奖励最大化的保证;实证上,AMPO在AlpacaEval上使用Llama 8B和Mistral 7B实现了当前最优性能。数据集已公开于Hugging Face。

原文摘要 · Abstract (English)

Multi-preference optimization enriches language-model alignment beyond pairwise preferences by contrasting entire sets of helpful and undesired responses, thereby enabling richer training signals for large language models. During self-play alignment, these models often produce numerous candidate answers per query, rendering it computationally infeasible to include all responses in the training objective. In this work, we propose $\textit{Active Multi-Preference Optimization}$ (AMPO), a novel approach that combines on-policy generation, a multi-preference group-contrastive loss, and active subset selection. Specifically, we score and embed large candidate pools of responses and then select a small, yet informative, subset that covers reward extremes and distinct semantic clusters for preference optimization. Our contrastive training scheme is capable of identifying not only the best and worst answers but also subtle, underexplored modes that are crucial for robust alignment. Theoretically, we provide guarantees for expected reward maximization using our active selection method, and empirically, AMPO achieves state-of-the-art results on $\textit{AlpacaEval}$ using Llama 8B and Mistral 7B. We release our datasets $\href{https://huggingface.co/Multi-preference-Optimization}{here}$.

大模型对齐偏好优化主动学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。