Search papers, labs, and topics across Lattice.
This paper introduces REVES, a two-stage iterative framework that enhances test-time scaling for Large Language Models (LLMs) by leveraging online data augmentation and policy optimization. By transforming intermediate "near-miss" answers into decoupled revision and verification prompts, the method effectively trains the model to improve both answer transformation and error identification. The results demonstrate significant performance gains on LiveCodeBench and other constraint-satisfaction puzzles, outperforming traditional multi-turn reinforcement learning approaches while using a smaller model with fewer computational resources.
By focusing on correcting "near-miss" answers, REVES achieves a remarkable +6.5 point improvement over standard RL methods, showcasing a new way to enhance LLM reasoning without extensive computational costs.
Test-time scaling via sequential revision has emerged as a powerful paradigm for enhancing Large Language Model (LLM) reasoning. However, standard post-training methods primarily optimize single-shot objectives, creating a fundamental misalignment with multi-step inference dynamics. While recent work treats this as multi-turn reinforcement learning (RL), conventional approaches optimize over the multi-step trajectories directly, failing to further exploit the high-quality mistakes in intermediate steps that model can learn from correcting them. We propose a two-stage iterative framework that alternates between online data/prompt augmentation and policy optimization. By converting the intermediate steps (``near-miss'' answers) in the successful recovery trajectories into decoupled revision and verification prompts, our approach concentrates training on both effective answer transformation and error identification. This approach enables efficient off-policy data generation and reduces the computational overhead of long-horizon sampling compared to standard multi-turn RL. On LiveCodeBench, using publicly available test cases as feedback, we observe gains of +6.5 points over the RL baseline and +4.0 points over standard multi-turn training. Beyond coding, our approach matches the previously reported SOTA result on circle packing while using the smallest base model (4B) and far fewer rollouts than the much larger evolutionary search systems. Math results under ground-truth verification further confirm improved correction ability. It also generalizes to out-of-distribution constraint-satisfaction puzzles such as n\_queens and mini\_sudoku, where correctness is defined entirely by problem constraints. Code is available at https://github.com/yxliu02/REVES.git.
