arXiv:2505.15074cs.CLcs.AI2025-05EMNLP被引 6

针对大模型对不平衡数据的偏好偏差,提出动态调整优化策略的新方法。

DISCO Balances the Scales: Adaptive Domain- and Difficulty-Aware Reinforcement Learning on Imbalanced Data

  • 基于领域和难度双重感知,自适应重加权奖励信号。
  • 在多领域不平衡数据上提升模型泛化能力,性能超现有方法5%。
  • 适合需要公平性与鲁棒性的大模型对齐任务场景。

大型语言模型(LLM)正通过人类反馈强化学习(RLHF)不断对齐人类偏好。其中,组相对策略优化(GRPO)因结构简单且表现优异而受到关注,尤其无需训练价值函数。然而,GRPO隐含假设各领域分布均衡、语义对齐一致,这一假设在真实数据中极少成立。当应用于多领域不平衡数据时,GRPO会过度优化主导领域,忽视少数领域,导致泛化能力差与不公平。本文提出领域感知自一致性策略优化(DISCO),作为GRPO的严谨扩展,通过两项关键创新应对组间不平衡:领域感知奖励缩放根据领域出现频率重新加权优化;难度感知奖励缩放利用提示级自一致性识别不确定性高、学习价值大的提示。二者协同促进跨领域的更公平、高效策略学习。在多个大模型及偏斜训练分布下的实验表明,DISCO显著提升泛化性能,在Qwen3模型上优于现有GRPO变体5%,并在多领域对齐基准上达到新最优结果。代码与数据已开源:https://github.com/Tonyzhou98/disco_grpo。

原文摘要 · Abstract (English)

Large Language Models (LLMs) are increasingly aligned with human preferences through Reinforcement Learning from Human Feedback (RLHF). Among RLHF methods, Group Relative Policy Optimization (GRPO) has gained attention for its simplicity and strong performance, notably eliminating the need for a learned value function. However, GRPO implicitly assumes a balanced domain distribution and uniform semantic alignment across groups, assumptions that rarely hold in real-world datasets. When applied to multi-domain, imbalanced data, GRPO disproportionately optimizes for dominant domains, neglecting underrepresented ones and resulting in poor generalization and fairness. We propose Domain-Informed Self-Consistency Policy Optimization (DISCO), a principled extension to GRPO that addresses inter-group imbalance with two key innovations. Domain-aware reward scaling counteracts frequency bias by reweighting optimization based on domain prevalence. Difficulty-aware reward scaling leverages prompt-level self-consistency to identify and prioritize uncertain prompts that offer greater learning value. Together, these strategies promote more equitable and effective policy learning across domains. Extensive experiments across multiple LLMs and skewed training distributions show that DISCO improves generalization, outperforms existing GRPO variants by 5% on Qwen3 models, and sets new state-of-the-art results on multi-domain alignment benchmarks. Our code and data are available at https://github.com/Tonyzhou98/disco_grpo.

大模型对齐强化学习数据不平衡

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