SCAN: Self-Denoising Monte Carlo Annotation for Robust Process Reward Learning

TL;DR

SCAN: Self-Denoising Monte Carlo Annotation reduces noise in synthetic data, boosting PRM F1 from 19.9 to 59.1 with only 6% inference cost.

cs.LG 🔴 Advanced 2025-09-20 32 views
Yuyang Ding Xinyu Shi Juntao Li Xiaobo Liang Zhaopeng Tu Min Zhang
AI NLP Reinforcement Learning Noise Robustness Large-Scale Data

Key Findings

Methodology

This study analyzes the noise distribution in MC-generated synthetic data, revealing bias types—under- and over-estimation—due to model limitations. It introduces a self-confidence metric to filter high-quality samples and employs a self-correcting loss to mitigate noise effects. Lightweight models (e.g., 1.5B parameters) generate responses efficiently, enabling high-quality annotations at low cost. The framework selectively applies MC estimation to uncertain samples, maximizing data utilization. Experiments on ProcessBench show a F1 increase from 19.9 to 59.1, outperforming models trained on large human-annotated datasets like PRM800K. The approach scales well, with performance improving as synthetic data volume grows, demonstrating strong potential for scalable, cost-effective PRM training.

Key Results

  • Using lightweight models with self-confidence filtering, PRM achieves F1 of 59.1 on ProcessBench, a 39.2 point increase over baseline, with inference cost only 6% of traditional MC.
  • Training on just 101K synthetic samples surpasses PRM800K trained with human labels; performance further improves with larger synthetic datasets, confirming scalability.
  • Self-confidence metrics effectively identify and exclude noisy samples, leading to robust training and significant performance gains in complex reasoning tasks.

Significance

This work addresses the critical challenge of noisy synthetic data in PRM training, offering a scalable, low-cost solution that rivals large-scale human annotation. It enhances the robustness and interpretability of models in complex reasoning, paving the way for broader adoption in real-world applications such as automated reasoning, educational tools, and decision support systems. The framework’s ability to leverage lightweight models for high-quality annotations reduces barriers for deploying advanced AI systems across industries, fostering more accessible and reliable AI solutions.

Technical Contribution

The paper pioneers a systematic analysis of noise distribution in MC-generated data, introducing a self-confidence metric for sample filtering and a self-correcting loss function for robust learning. It demonstrates that lightweight models can generate high-quality annotations through self-denoising, significantly reducing inference costs. The combination of noise distribution insights and adaptive training strategies offers a new paradigm for noise-tolerant learning in large-scale synthetic data scenarios, expanding the frontier of scalable, cost-efficient PRM development.

Novelty

This is the first comprehensive study analyzing the noise distribution in MC-based synthetic data for process reward learning, proposing a self-confidence driven filtering and self-denoising training framework. Unlike prior works relying on external critic models, this approach leverages lightweight models and internal metrics, achieving high performance with minimal supervision. Its innovative combination of noise analysis, selective sampling, and adaptive loss design marks a significant advancement in noise-robust training for large language models.

Limitations

  • The reliance on self-confidence metrics may limit effectiveness in extremely noisy or highly complex tasks where model bias is severe.
  • The method's performance depends on hyperparameter tuning (e.g., confidence thresholds, tolerance distance), which may affect stability across different datasets.
  • Scalability to multi-modal data or tasks beyond text-based reasoning remains to be validated, and the current framework may require adaptation for such scenarios.

Future Work

Future research will explore dynamic adjustment of confidence thresholds, integration with multi-modal data, and extension to more complex reasoning tasks. Combining this framework with external critic models or reinforcement learning techniques could further enhance robustness. Additionally, developing theoretical guarantees for noise correction and expanding the approach to other domains like vision-language tasks will be key directions.

AI Executive Summary

In the rapidly evolving landscape of large language models (LLMs), process reward models (PRMs) have emerged as vital tools for enabling fine-grained, step-by-step evaluation and reasoning. Traditional training of PRMs relies heavily on costly human annotations, which limits scalability. Monte Carlo (MC) estimation offers a promising alternative by generating synthetic data efficiently, but it introduces significant noise, leading to overfitting and performance bottlenecks. Addressing this challenge, the paper introduces SCAN (Self-Denoising Monte Carlo Annotation), a novel framework that leverages noise distribution analysis and self-confidence metrics to improve data quality and robustness.

SCAN's core innovation lies in its ability to identify and filter noisy samples through a self-confidence measure, which estimates the reliability of model-generated annotations. By selectively applying MC estimation to uncertain samples and employing a self-correcting loss function, the framework effectively reduces noise impact without external supervision. This approach enables the use of lightweight models, such as Qwen2.5-Math-1.5B, to produce high-quality annotations at a fraction of the traditional cost—only 6% inference cost—while achieving performance surpassing models trained on large human-annotated datasets.

Experimental results demonstrate that training PRMs on just 101K synthetic samples yields performance comparable to or better than models trained on 800K human-labeled data, with F1 scores rising from 19.9 to 59.1 on ProcessBench. As the synthetic dataset scales to over 200K samples, performance continues to improve, confirming the scalability and effectiveness of SCAN. The framework's ability to mitigate noise and enhance robustness opens new avenues for cost-efficient, large-scale deployment of process reward models in complex reasoning tasks, with promising applications across AI-driven education, automated reasoning, and decision support systems.

Despite these advances, limitations remain in hyperparameter sensitivity and applicability to multi-modal data. Future work will focus on adaptive thresholding, multi-modal extension, and theoretical analysis to further solidify SCAN's role in scalable AI development. Overall, this work marks a significant step toward robust, low-cost, high-performance AI systems capable of deep reasoning and complex decision-making.

Deep Analysis

Background

The evolution of large language models (LLMs) such as GPT-4 and PaLM has driven interest in enhancing their reasoning capabilities. Process reward models (PRMs) evaluate models step-by-step, improving interpretability and self-correction. Early efforts like Chain-of-Thought prompting demonstrated the importance of intermediate reasoning steps. However, high-quality data for training PRMs remains scarce due to expensive human annotation. Synthetic data via Monte Carlo (MC) estimation emerged as a cost-effective alternative, but its high noise level hampers performance. Recent approaches incorporated external critic models (e.g., Qwen-72B) for noise filtering, yet these depend on additional supervision. This paper addresses the core challenge: how to effectively denoise MC-generated data without external supervision, enabling scalable, robust PRM training.

Core Problem

The key issue is that MC estimation produces noisy labels due to the imperfect nature of the completer models, which tend to both underestimate and overestimate step correctness. This bias leads to overfitting, poor generalization, and limited scalability of PRMs trained solely on synthetic data. Existing solutions rely on external critic models or large human datasets, which are costly and less flexible. The challenge is to develop a method that can automatically identify and correct noise within synthetic data, maintaining high annotation quality while minimizing computational costs. Addressing this problem is crucial for deploying PRMs in real-world applications requiring large, diverse datasets.

Innovation

The paper introduces a comprehensive framework combining noise distribution analysis, self-confidence metrics, and self-correcting training strategies. Key innovations include: 1) analyzing the bias patterns in MC-generated data, revealing under- and over-estimation issues; 2) designing a self-confidence score to evaluate annotation reliability; 3) filtering samples based on confidence to focus training on high-quality data; 4) employing a self-denoising loss that adaptively corrects biased labels; 5) leveraging lightweight models for efficient data synthesis, drastically reducing inference costs. These contributions enable high-quality, scalable PRM training without external critic models, setting a new standard for noise-robust learning.

Methodology

  • �� Analyze noise distribution in MC synthetic data, identifying bias types (under- and over-estimation). • Develop a self-confidence metric SCθ(q) to measure annotation reliability. • Use this metric to filter high-confidence samples for positive data, reducing noise. • Collect negative samples with uncertain annotations for step-wise correctness annotation via a lightweight generator model. • Apply Monte Carlo estimation selectively to these uncertain samples, maximizing data utilization. • Introduce a self-correcting loss function that reweights labels based on confidence scores, mitigating bias. • Use lightweight models like Qwen2.5-Math-1.5B for response generation, lowering inference costs. • Tune confidence thresholds and tolerance distance d for optimal noise filtering. • Train the PRM with reweighted labels, improving robustness against noisy annotations.

Experiments

The experimental setup involves training PRMs on synthetic datasets generated by lightweight models, primarily Qwen2.5-Math-1.5B and 7B, using ProcessBench, GSM8K, and Numina-Math as benchmarks. The core evaluation metrics include F1 scores, error detection accuracy, and best-of-N response selection. The experiments compare traditional MC estimation, external critic filtering, and the proposed SCAN framework, with ablation studies on confidence thresholds, sample size, and loss functions. Hyperparameters such as response count (k=64 or 128) and tolerance distance (d=2) are optimized. Results show that models trained with SCAN outperform baselines trained on large human datasets, with significant improvements in both accuracy and robustness across multiple reasoning benchmarks.

Results

The PRM trained with 101K synthetic samples using SCAN achieves an F1 of 59.1 on ProcessBench, a substantial increase from 19.9 baseline, surpassing models trained on 800K human-annotated data. Performance scales positively with data volume, reaching over 197K samples, with continued improvements. The approach effectively filters noisy samples via self-confidence, leading to more reliable training data. Ablation studies confirm the importance of confidence thresholds and the self-denoising loss. The lightweight models demonstrate that high-quality annotations are feasible at low inference costs, making large-scale training practical. Overall, the results validate the framework’s scalability, robustness, and efficiency.

Applications

SCAN enables cost-effective training of PRMs for complex reasoning tasks such as automated mathematics, legal reasoning, and scientific analysis. Its low-cost synthetic data generation makes it accessible for industry and academia, reducing dependence on expensive human annotations. The framework can be integrated into multi-task learning pipelines, supporting scalable AI systems in education, research, and decision-making. Long-term, it paves the way for autonomous AI agents capable of deep reasoning with minimal supervision, facilitating deployment in real-world scenarios like intelligent tutoring, legal document analysis, and scientific discovery.

Limitations & Outlook

Current reliance on self-confidence metrics may limit effectiveness in highly noisy or ambiguous scenarios. The method's sensitivity to hyperparameters like confidence thresholds and tolerance distance requires careful tuning. Scalability to multi-modal data or tasks involving vision, audio, or sensor inputs remains to be validated. The computational cost, although reduced, still involves multiple response sampling and MC estimation, which could be challenging for extremely large datasets. Future work should focus on adaptive parameter tuning, multi-modal extension, and theoretical analysis of noise correction guarantees.

Plain Language Accessible to non-experts

想象你在厨房里做饭,食材代表模型的推理步骤。传统做法是用大量新鲜食材(人工标注)来保证菜的味道,但成本很高。用MC估计就像用剩菜(合成数据)快速做菜,虽然省钱,但剩菜中可能夹杂变质的东西(噪声),影响菜的质量。SCAN就像用一种特殊的味觉检测器(自信度指标),只挑选新鲜的食材(高自信样本),并用厨师(模型)自己不断尝试调整(自我校正),最终做出既便宜又好吃的菜。这种方法让厨房效率大大提高,菜品也更稳定,既省钱又好用。

ELI14 Explained like you're 14

想象你在学校的科学实验室里做实验,老师告诉你用一种新方法可以节省时间和材料,但可能会出错。你用一种特别的“自我检测器”判断每个步骤是否正确,只用最靠谱的步骤继续实验。这样,即使有一些错误,也不会影响整体结果。慢慢地,你的实验变得又快又准,还能用很少的材料做出好成果。这就像SCAN,用聪明的“自我检测”帮模型筛除错误,让它既省钱又能表现得更棒。未来,这种方法还能帮我们在很多复杂任务中变得更聪明、更可靠,就像你在实验中变成了真正的科学家!

Abstract

Process reward models (PRMs) offer fine-grained, step-level evaluations that facilitate deeper reasoning processes in large language models (LLMs), proving effective in complex tasks like mathematical reasoning. However, developing PRMs is challenging due to the high cost and limited scalability of human-annotated data. Synthetic data from Monte Carlo (MC) estimation is a promising alternative but suffers from a high noise ratio, which can cause overfitting and hinder large-scale training. In this work, we conduct a preliminary study on the noise distribution in synthetic data from MC estimation, identifying that annotation models tend to both underestimate and overestimate step correctness due to limitations in their annotation capabilities. Building on these insights, we propose Self-Denoising Monte Carlo Annotation (SCAN), an efficient data synthesis and noise-tolerant learning framework. Our key findings indicate that: (1) Even lightweight models (e.g., 1.5B parameters) can produce high-quality annotations through a self-denoising strategy, enabling PRMs to achieve superior performance with only 6% the inference cost required by vanilla MC estimation. (2) With our robust learning strategy, PRMs can effectively learn from this weak supervision, achieving a 39.2 F1 score improvement (from 19.9 to 59.1) in ProcessBench. Despite using only a compact synthetic dataset, our models surpass strong baselines, including those trained on large-scale human-annotated datasets such as PRM800K. Furthermore, performance continues to improve as we scale up the synthetic data, highlighting the potential of SCAN for scalable, cost-efficient, and robust PRM training.

cs.LG cs.CL