paper-with-me

홈 › Papers

Heimdall: test-time scaling on the generative verification

2025-04-14 · Wenlei Shi, Xing Jin

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.

📄 PDF Abstract BibTeX arXiv:2504.10337

Code (0)

등록된 구현이 없습니다.

Tasks

Math

Similar Papers 제목 키워드 기반

Budget-aware Test-time Scaling via Discriminative Verification

2025-10-16 · Kyle Montgomery, Sijun Tan, Yuqi Chen, Siyuan Zhuang 외 arxiv

Test-time scaling is a powerful strategy for boosting the performance of large language models on complex reasoning tasks. While state-of-the-art approaches often employ generative verifiers to select the best solution f…

MaxProof: Scaling Mathematical Proof with Generative-Verifier RL and Population-Level Test-Time Scaling

2026-06-11 · Jiacheng Chen, Xinyu Zhang, Shunkai Zhang, Yanmohan Wang 외 arxiv

We present MaxProof, a population-level test-time scaling framework for competition-level mathematical proof in the MiniMax-M3 series. M3 first trains three proof-oriented capabilities -- proof generation, proof verifica…

Putting the Value Back in RL: Better Test-Time Scaling by Unifying LLM Reasoners With Verifiers

2025-05-07 · Kusha Sareen, Morgane M Moss, Alessandro Sordoni, Rishabh Agarwal 외

Prevalent reinforcement learning~(RL) methods for fine-tuning LLM reasoners, such as GRPO or Leave-one-out PPO, abandon the learned value function in favor of empirically estimated returns. This hinders test-time compute…

MathReinforcement Learning (RL)

Trust but Verify! A Survey on Verification Design for Test-time Scaling

2025-08-20 · V Venktesh, Mandeep Rathee, Avishek Anand arxiv

Test-time scaling (TTS) has emerged as a new frontier for scaling the performance of Large Language Models. In test-time scaling, by using more computational resources during inference, LLMs can improve their reasoning p…

Generative Universal Verifier as Multimodal Meta-Reasoner

2025-10-15 · Xinchen Zhang, Xiaoying Zhang, Youbin Wu, Yanbin Cao 외 arxiv

We introduce Generative Universal Verifier, a novel concept and plugin designed for next-generation multimodal reasoning in vision-language models and unified multimodal models, providing the fundamental capability of re…

Multimodal ReasoningImage Generation