arXiv:2504.10337cs.AI2025-04被引 27

让AI自检答案对错,准确率从62.5%提至97.5%

Heimdall: test-time scaling on the generative verification

  • 用强化学习训练模型自动判断数学解法对错
  • 重复采样后验证准确率达97.5%,解决难题泛化能力强
  • 适合需要高可信解法的竞赛数学与自动知识发现

AI系统只能在自身能验证知识的前提下持续积累和维护知识。近期长链思维(long CoT)推理研究显示大语言模型在解决竞赛级问题上潜力巨大,但其验证能力仍较弱且未被充分探索。本文提出Heimdall,一种可准确判断解法正确性的长链思维验证模型。通过纯强化学习,其验证准确率在竞赛数学题上从62.5%提升至94.5%;进一步通过重复采样缩放,准确率再升至97.5%。人类评估表明,Heimdall具备出色泛化能力,能有效识别训练中未出现的复杂数学证明中的多数错误。我们还提出悲观验证(Pessimistic Verification),利用Heimdall评估求解模型输出并基于悲观原则选择最可能正确的解。以DeepSeek-R1-Distill-Qwen-32B为求解器,在AIME2025上,16倍计算预算下准确率从54.2%提升至70.0%,更多预算下达83.3%;使用更强求解器Gemini 2.5 Pro时,得分达93.0%。最后,我们构建了一个自动知识发现原型系统:提问、求解、验证三元结构。结合NuminaMath数据合成流程,Heimdall成功识别出近一半数据存在缺陷,这一结果与NuminaMath近期消融研究高度一致。

原文摘要 · Abstract (English)

An AI system can create and maintain knowledge only to the extent that it can verify that knowledge itself. Recent work on long Chain-of-Thought reasoning has demonstrated great potential of LLMs on solving competitive problems, but their verification ability remains to be weak and not sufficiently investigated. In this paper, we propose Heimdall, the long CoT verification LLM that can accurately judge the correctness of solutions. With pure reinforcement learning, we boost the verification accuracy from 62.5% to 94.5% on competitive math problems. By scaling with repeated sampling, the accuracy further increases to 97.5%. Through human evaluation, Heimdall demonstrates impressive generalization capabilities, successfully detecting most issues in challenging math proofs, the type of which is not included during training. Furthermore, we propose Pessimistic Verification to extend the functionality of Heimdall to scaling up the problem solving. It calls Heimdall to judge the solutions from a solver model and based on the pessimistic principle, selects the most likely correct solution with the least uncertainty. Taking DeepSeek-R1-Distill-Qwen-32B as the solver model, Pessimistic Verification improves the solution accuracy on AIME2025 from 54.2% to 70.0% with 16x compute budget and to 83.3% with more compute budget. With the stronger solver Gemini 2.5 Pro, the score reaches 93.0%. Finally, we prototype an automatic knowledge discovery system, a ternary system where one poses questions, another provides solutions, and the third verifies the solutions. Using the data synthesis work NuminaMath for the first two components, Heimdall effectively identifies problematic records within the dataset and reveals that nearly half of the data is flawed, which interestingly aligns with the recent ablation studies from NuminaMath.

AI验证数学推理生成质量

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