TPO让大模型从多级偏好树中学习,提升复杂推理能力。
TPO: Aligning Large Language Models with Multi-branch & Multi-step Preference Trees
- 直接基于偏好树的多级响应排序进行对齐,避免采样损失
- 在4个数据集上5个模型均超越DPO,数学推理准确率显著提升
- 适合需要长链推理和精细奖励调节的研究者
在数学推理等复杂推理任务中,现有方法利用思维树(ToT)生成偏好树,并通过直接偏好优化(DPO)抑制不偏好输出。但二元偏好优化无法充分学习偏好树中多层级的偏好/非偏好响应,导致偏好学习不完整。本文提出树偏好优化(TPO),不采样成对响应,而是直接从完整偏好树中学习。TPO将模型对齐建模为偏好列表排序问题,使策略能更有效利用给定提示下的响应排序信息。此外,为增强模型识别长链推理中关键步骤的能力并扩大偏好列表中的相对奖励差距,TPO引入自适应步级奖励机制,动态调整轨迹中每一步的奖励值。我们在五个公开大模型和四个数据集上进行了广泛实验,结果表明TPO在所有设置下均持续优于DPO。代码已开源。
原文摘要 · Abstract (English)
In the domain of complex reasoning tasks, such as mathematical reasoning, recent advancements have proposed the use of Direct Preference Optimization (DPO) to suppress output of dispreferred responses, thereby enhancing the long-chain reasoning capabilities of large language models (LLMs). To this end, these studies employed LLMs to generate preference trees via Tree-of-thoughts (ToT) and sample the paired preference responses required by the DPO algorithm. However, the DPO algorithm based on binary preference optimization is unable to learn multiple responses with varying degrees of preference/dispreference that provided by the preference trees, resulting in incomplete preference learning. In this work, we introduce Tree Preference Optimization (TPO), that does not sample paired preference responses from the preference tree; instead, it directly learns from the entire preference tree during the fine-tuning. Specifically, TPO formulates the language model alignment as a Preference List Ranking problem, where the policy can potentially learn more effectively from a ranked preference list of responses given the prompt. In addition, to further assist LLMs in identifying discriminative steps within long-chain reasoning and increase the relative reward margin in the preference list, TPO utilizes Adaptive Step Reward to adjust the reward values of each step in trajectory for performing fine-grained preference optimization. We carry out extensive experiments on mathematical reasoning tasks to evaluate TPO. The experimental results indicate that TPO consistently outperforms DPO across five public large language models on four datasets. Our code is publicly available at https://github.com/MrBlankness/TPO.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。