用离线强化学习提升大模型多步推理能力,无需成对偏好数据
Offline Reinforcement Learning for LLM Multi-Step Reasoning
- 基于软贝尔曼方程联合优化策略与价值函数,解决多步推理中的信用分配问题
- 在GSM8K、MATH等任务上超越现有离线学习方法,提升推理准确率
- 可免费利用学习到的价值函数指导树搜索,提升测试阶段性能
通过离线强化学习提升大语言模型(LLMs)的多步推理能力,对于快速适应复杂任务至关重要。尽管直接偏好优化(DPO)在对齐人类偏好方面表现良好,但其在多步推理任务中存在两大局限:一是依赖成对偏好数据,而此类数据在多步推理中难以获取;二是对所有标记一视同仁,无法有效处理稀疏奖励下的信用分配问题。本文提出OREO(Offline Reasoning Optimization),一种用于增强LLM多步推理能力的离线强化学习方法。基于最大熵强化学习的洞见,该方法通过优化软贝尔曼方程联合学习策略模型与价值函数,理论上减少了对成对数据的需求,并实现更优的信用分配。实验表明,OREO在多个多步推理基准上优于现有离线学习方法,包括数学推理任务(GSM8K、MATH)和具身智能体控制任务(ALFWorld)。当资源允许时,该方法可扩展为多轮迭代框架。此外,训练得到的价值函数可免费用于测试时的树搜索,进一步提升性能。
原文摘要 · Abstract (English)
Improving the multi-step reasoning ability of large language models (LLMs) with offline reinforcement learning (RL) is essential for quickly adapting them to complex tasks. While Direct Preference Optimization (DPO) has shown promise in aligning LLMs with human preferences, it is less suitable for multi-step reasoning tasks because (1) DPO relies on paired preference data, which is not readily available for multi-step reasoning tasks, and (2) it treats all tokens uniformly, making it ineffective for credit assignment in multi-step reasoning tasks, which often come with sparse reward. In this work, we propose OREO (Offline Reasoning Optimization), an offline RL method for enhancing LLM multi-step reasoning. Building on insights from previous works of maximum entropy reinforcement learning, it jointly learns a policy model and value function by optimizing the soft Bellman Equation. We show in principle that it reduces the need to collect pairwise data and enables better credit assignment. Empirically, OREO surpasses existing offline learning methods on multi-step reasoning benchmarks, including mathematical reasoning tasks (GSM8K, MATH) and embodied agent control (ALFWorld). The approach can be extended to a multi-iteration framework when additional resources are available. Furthermore, the learned value function can be leveraged to guide the tree search for free, which can further boost performance during test time.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。