Search papers, labs, and topics across Lattice.
This paper introduces a novel process reward model (PRM) training method that uses Monte Carlo Net Information Gain to automatically generate step-level labels for chain-of-thought reasoning, estimating the impact of each step on the final answer's likelihood. By reducing the computational complexity to O(N), this approach overcomes the scalability limitations of prior methods that relied on human annotations or computationally intensive automatic labeling. Experiments across mathematics, programming, SQL, and science benchmarks demonstrate the effectiveness of this PRM in chain-of-thought selection within best-of-K evaluation settings.
Forget expensive human annotations – this new method uses information theory to automatically score each step of an LLM's reasoning process, making chain-of-thought supervision scalable and efficient.
Multi-step reasoning improves the capabilities of large language models (LLMs) but increases the risk of errors propagating through intermediate steps. Process reward models (PRMs) mitigate this by scoring each step individually, enabling fine-grained supervision and improved reliability. Existing methods for training PRMs rely on costly human annotations or computationally intensive automatic labeling. We propose a novel approach to automatically generate step-level labels using Information Theory. Our method estimates how each reasoning step affects the likelihood of the correct answer, providing a signal of step quality. Importantly, it reduces computational complexity to $\mathcal{O}(N)$, improving over the previous $\mathcal{O}(N \log N)$ methods. We demonstrate that these labels enable effective chain-of-thought selection in best-of-$K$ evaluation settings across diverse reasoning benchmarks, including mathematics, Python programming, SQL, and scientific question answering. This work enables scalable and efficient supervision of LLM reasoning, particularly for tasks where error propagation is critical.
