arXiv:2502.01203cs.LGstat.ML2025-02NeurIPS被引 11

首次解决多参考模型强化学习对齐的理论难题,提升模型多样性与鲁棒性。

KL-Regularized RLHF with Multiple Reference Models: Exact Solutions and Sample Complexity

  • 提出反KL正则化框架下的精确解法,支持多参考模型并行对齐。
  • 给出样本复杂度保证,证明在多参考场景下仍能高效收敛。
  • 为现代大模型对齐提供可信赖的理论基础,适合研究者与工程团队参考。

当前主流的大语言模型(LLM)对齐方法主要依赖单一参考模型,限制了多样性,易导致模型过拟合,并未能充分利用丰富的预训练模型资源。引入多个参考模型有望通过拓展视角、降低偏差、融合开源模型优势来克服上述问题。然而,在人类反馈强化学习(RLHF)框架中整合多个参考模型面临重大理论挑战,精确解法长期未被解决。本文首次提出反KL正则化RLHF中多参考模型问题的精确解法,构建了完整的理论框架,包含严格的统计分析和样本复杂度保证。此外,我们将分析扩展至前向KL正则化场景,揭示了多参考情形下的样本复杂度新规律。本工作为更先进、可适应的LLM对齐技术奠定基础,推动实现理论严谨且适配现代人工智能生态系统的对齐框架。

原文摘要 · Abstract (English)

Recent methods for aligning large language models (LLMs) with human feedback predominantly rely on a single reference model, which limits diversity, model overfitting, and underutilizes the wide range of available pre-trained models. Incorporating multiple reference models has the potential to address these limitations by broadening perspectives, reducing bias, and leveraging the strengths of diverse open-source LLMs. However, integrating multiple reference models into reinforcement learning with human feedback (RLHF) frameworks poses significant theoretical challenges, where achieving exact solutions has remained an open problem. This paper presents the first \emph{exact solution} to the multiple reference model problem in reverse KL-regularized RLHF. We introduce a comprehensive theoretical framework that includes rigorous statistical analysis and provides sample complexity guarantees. Additionally, we extend our analysis to forward KL-regularized RLHF, offering new insights into sample complexity requirements in multiple reference scenarios. Our contributions lay the foundation for more advanced and adaptable LLM alignment techniques, enabling the effective use of multiple reference models. This work paves the way for developing alignment frameworks that are both theoretically sound and better suited to the challenges of modern AI ecosystems.

大模型对齐强化学习理论分析

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