arXiv:2411.16345cs.CL2024-11ICLR被引 41

用伪反馈提升大模型推理能力,无需人工标注。

Preference Optimization for Reasoning with Pseudo Feedback

  • 用测试用例生成伪反馈,替代人工标注。
  • 数学和编程任务性能显著提升,最高达68.6分。
  • 适合无高质量标注数据的推理模型优化场景。

偏好优化技术如直接偏好优化(DPO)常用于提升大语言模型在数学推理和编程等领域的推理能力,通常在监督微调后使用。然而,这类方法依赖于经过人工验证的高质量标签,而此类数据集有限。本文提出一种新方法,将解题过程的评分建模为对关联测试用例的评估,从而生成伪反馈。我们探索了两种基于测试用例的伪反馈形式:由前沿大模型生成的反馈,以及扩展自一致性策略以支持多测试用例的反馈。在数学推理与编程任务上,我们使用伪反馈进行偏好优化并取得显著效果。以Mathstral-7B为基线模型,在MATH数据集上得分从58.3提升至68.6,超越NuminaMath-72B和GPT-4-Turbo-1106-preview;在GSM8K和College Math上分别从85.6提升至90.3、从34.3提升至42.3。基于Deepseek-coder-7B-v1.5,在LiveCodeBench上得分从21.1提升至24.6,超过Claude-3-Haiku。

原文摘要 · Abstract (English)

Preference optimization techniques, such as Direct Preference Optimization (DPO), are frequently employed to enhance the reasoning capabilities of large language models (LLMs) in domains like mathematical reasoning and coding, typically following supervised fine-tuning. These methods rely on high-quality labels for reasoning tasks to generate preference pairs; however, the availability of reasoning datasets with human-verified labels is limited. In this study, we introduce a novel approach to generate pseudo feedback for reasoning tasks by framing the labeling of solutions to reason problems as an evaluation against associated test cases. We explore two forms of pseudo feedback based on test cases: one generated by frontier LLMs and the other by extending self-consistency to multi-test-case. We conduct experiments on both mathematical reasoning and coding tasks using pseudo feedback for preference optimization, and observe improvements across both tasks. Specifically, using Mathstral-7B as our base model, we improve MATH results from 58.3 to 68.6, surpassing both NuminaMath-72B and GPT-4-Turbo-1106-preview. In GSM8K and College Math, our scores increase from 85.6 to 90.3 and from 34.3 to 42.3, respectively. Building on Deepseek-coder-7B-v1.5, we achieve a score of 24.6 on LiveCodeBench (from 21.1), surpassing Claude-3-Haiku.

偏好优化推理增强伪反馈大模型

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