通过低代价方差减少提升基于价值的推理验证器性能。
Improving Value-based Process Verifier via Low-Cost Variance Reduction
- 提出ComMCS方法,线性组合多步蒙特卡洛估计以降低方差。
- 在MATH-500上相比基线提升2.2分,优于回归优化方法2.8分。
- 无需额外大模型推理成本,适合高精度数学推理任务优化。
大语言模型在众多任务中表现卓越,但在数学等复杂领域推理能力仍受限。基于价值的过程验证器通过估计部分推理链通向正确答案的概率,是提升推理的有效方法。然而,其效果常受训练标注误差影响,主因是蒙特卡洛采样数量有限导致的高方差。本文指出该误差主要源于方差而非偏差,且现有蒙特卡洛估计器为最小方差无偏估计量(MVUE)。为此,我们提出复合蒙特卡洛采样(ComMCS)方法,通过线性组合当前与后续步骤的蒙特卡洛估计器构造无偏估计量。理论上,该方法可预测地降低方差,且不增加额外的大模型推理开销。我们在MATH-500和GSM8K数据集上进行实证实验,结果表明:在Best-of-32采样下,ComMCS在MATH-500上比非方差减少基线提升2.2分,比回归优化方法提升2.8分。
原文摘要 · Abstract (English)
Large language models (LLMs) have achieved remarkable success in a wide range of tasks. However, their reasoning capabilities, particularly in complex domains like mathematics, remain a significant challenge. Value-based process verifiers, which estimate the probability of a partial reasoning chain leading to a correct solution, are a promising approach for improving reasoning. Nevertheless, their effectiveness is often hindered by estimation error in their training annotations, a consequence of the limited number of Monte Carlo (MC) samples feasible due to the high cost of LLM inference. In this paper, we identify that the estimation error primarily arises from high variance rather than bias, and the MC estimator is a Minimum Variance Unbiased Estimator (MVUE). To address the problem, we propose the \textsc{Com}pound \textsc{M}onte \textsc{C}arlo \textsc{S}ampling (ComMCS) method, which constructs an unbiased estimator by linearly combining the MC estimators from the current and subsequent steps. Theoretically, we show that our method leads to a predictable reduction in variance, while maintaining an unbiased estimation without additional LLM inference cost. We also perform empirical experiments on the MATH-500 and GSM8K benchmarks to demonstrate the effectiveness of our method. Notably, ComMCS outperforms regression-based optimization method by 2.8 points, the non-variance-reduced baseline by 2.2 points on MATH-500 on Best-of-32 sampling experiment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。