通过感知困惑度的强化学习,纠正大模型在数学推理中偏好低困惑度解法的偏差。
Rectify Evaluation Preference: Improving LLMs' Critique on Math Reasoning via Perplexity-aware Reinforcement Learning
- 构建一题多解基准,发现模型倾向认为困惑度低的解法正确
- 提出困惑度感知强化学习,让模型学会判断高困惑度解法为正确
- 显著提升模型对数学推理错误的检出能力,适合评估类研究者
为提升大语言模型在多步数学推理(MsMR)中的表现,关键在于从语料库中自动获取可扩展的监督信号,即对推理过程进行错误批判并给出最终解题判定。现有方法多依赖高质量微调示范数据来增强批判能力,却忽视了模型批判性能差的深层原因。本文从一个全新角度量化分析潜在原因——评价偏好失衡,并开展统计偏好分析。基于分析结果,提出一种新颖的困惑度感知强化学习算法以修正评价偏好,从而提升批判能力。具体而言,构建了一个一题多解(OPS)基准,用于量化模型在评判自身与他人生成解法时的行为差异;进一步针对困惑度开展统计分析,发现一个有趣现象:'模型倾向于将低困惑度解法判为正确',称为‘不平衡评价偏好’。为此,将困惑度作为组相对策略优化中的引导信号,促使模型探索将低困惑度判错、高困惑度判对的路径。在自建OPS及现有批判基准上的大量实验验证了该方法的有效性。
原文摘要 · Abstract (English)
To improve Multi-step Mathematical Reasoning (MsMR) of Large Language Models (LLMs), it is crucial to obtain scalable supervision from the corpus by automatically critiquing mistakes in the reasoning process of MsMR and rendering a final verdict of the problem-solution. Most existing methods rely on crafting high-quality supervised fine-tuning demonstrations for critiquing capability enhancement and pay little attention to delving into the underlying reason for the poor critiquing performance of LLMs. In this paper, we orthogonally quantify and investigate the potential reason -- imbalanced evaluation preference, and conduct a statistical preference analysis. Motivated by the analysis of the reason, a novel perplexity-aware reinforcement learning algorithm is proposed to rectify the evaluation preference, elevating the critiquing capability. Specifically, to probe into LLMs' critiquing characteristics, a One-to-many Problem-Solution (OPS) benchmark is meticulously constructed to quantify the behavior difference of LLMs when evaluating the problem solutions generated by itself and others. Then, to investigate the behavior difference in depth, we conduct a statistical preference analysis oriented on perplexity and find an intriguing phenomenon -- ``LLMs incline to judge solutions with lower perplexity as correct'', which is dubbed as \textit{imbalanced evaluation preference}. To rectify this preference, we regard perplexity as the baton in the algorithm of Group Relative Policy Optimization, supporting the LLMs to explore trajectories that judge lower perplexity as wrong and higher perplexity as correct. Extensive experimental results on our built OPS and existing available critic benchmarks demonstrate the validity of our method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。