无需额外奖励模型,通过自训练提升大模型对齐效果。
TSO: Self-Training with Scaled Preference Optimization
- 用模型自身生成数据构建偏好样本,避免依赖外部奖励模型
- 通过人类与AI反馈修正模型偏好错误,提升响应多样性
- 采用双截断策略迭代优化,实现更稳定高效的对齐训练
提升大语言模型(LLMs)对人类偏好的符合度仍是持续的研究挑战。近期,无需与奖励模型交互的离线方法如直接偏好优化(DPO)因其简单、高效且稳定而受到关注。然而,这些方法高度依赖成对偏好样本的质量。同时,大量迭代方法需额外训练奖励模型,从模型生成的响应中筛选正负样本进行偏好学习。随着大模型能力增强,仅凭模型输出难以持续构建高质量的正负样本,因缺乏多样性。为此,本文提出TSO(Self-Training with Scaled Preference Optimization),一种无需训练额外奖励模型的自训练偏好优化框架。TSO通过构建模型矩阵增强响应多样性,并结合人类与AI反馈纠正模型偏好偏差。此外,采用迭代双截断奖励策略更新参考模型及其输出,自适应调整偏好数据并平衡优化过程。实验表明,TSO在多个对齐评估基准上优于现有主流方法,为对齐领域中的偏好数据构建与训练策略提供了实用洞见。
原文摘要 · Abstract (English)
Enhancing the conformity of large language models (LLMs) to human preferences remains an ongoing research challenge. Recently, offline approaches such as Direct Preference Optimization (DPO) have gained prominence as attractive options due to offering effective improvement in simple, efficient, and stable without interactions with reward models. However, these offline preference optimization methods highly rely on the quality of pairwise preference samples. Meanwhile, numerous iterative methods require additional training of reward models to select positive and negative samples from the model's own generated responses for preference learning. Furthermore, as LLMs' capabilities advance, it is quite challenging to continuously construct high-quality positive and negative preference instances from the model's outputs due to the lack of diversity. To tackle these challenges, we propose TSO, or Self-Training with Scaled Preference Optimization, a framework for preference optimization that conducts self-training preference learning without training an additional reward model. TSO enhances the diversity of responses by constructing a model matrix and incorporating human preference responses. Furthermore, TSO introduces corrections for model preference errors through human and AI feedback. Finally, TSO adopts iterative and dual clip reward strategies to update the reference model and its responses, adaptively adjusting preference data and balancing the optimization process. Experimental results demonstrate that TSO outperforms existing mainstream methods on various alignment evaluation benchmarks, providing practical insight into preference data construction and model training strategies in the alignment domain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。