arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.12163cs.CL

用于对齐语言模型的偏好引导反思采样

Preference-Guided Reflective Sampling for Aligning Language Models

National University of Singapore(新加坡国立大学)

Hai Ye, Hwee Tou Ng

AI总结:

本文提出偏好引导反思采样(PRS),通过基于树的生成框架和自适应自反思技术提升采样效率,并支持自然语言偏好优化,从而在AlpacaEval和Arena-Hard上显著优于重复随机采样,且适用于迭代离线RL训练。

摘要:

迭代式数据生成与模型再训练能够有效地将大语言模型(LLMs)对齐到人类偏好。数据采样过程至关重要,因为它显著影响策略改进的成败。重复随机采样是一种广泛使用的方法,它独立地多次查询模型以生成输出。在本工作中,我们提出了一种更有效的采样方法,名为偏好引导反思采样(Preference-Guided Reflective Sampling,PRS)。与随机采样不同,PRS采用基于树的生成框架来实现更高效的采样。它利用自适应自反思技术来更好地探索采样空间。通过用自然语言指定用户偏好,PRS能够根据这些偏好进一步优化响应生成。因此,PRS可以将模型对齐到多样化的用户偏好。我们的实验表明,PRS能够生成质量更高且奖励显著更高的响应。在AlpacaEval和Arena-Hard上,PRS在best-of-$N$采样中大幅优于重复随机采样。此外,PRS在应用于迭代离线RL训练时也表现出强劲的性能。

英文摘要:

Iterative data generation and model re-training can effectively align large language models(LLMs) to human preferences. The process of data sampling is crucial, as it significantly influences the success of policy improvement. Repeated random sampling is a widely used method that independently queries the model multiple times to generate outputs. In this work, we propose a more effective sampling method, named Preference-Guided Reflective Sampling (PRS). Unlike random sampling, PRS employs a tree-based generation framework to enable more efficient sampling. It leverages adaptive self-refinement techniques to better explore the sampling space. By specifying user preferences in natural language, PRS can further optimize response generation according to these preferences. As a result, PRS can align models to diverse user preferences. Our experiments demonstrate that PRS generates higher-quality responses with significantly higher rewards. On AlpacaEval and Arena-Hard, PRS substantially outperforms repeated random sampling in best-of-$N$ sampling. Moreover, PRS shows strong performance when applied in iterative offline RL training.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑