On the Power of (Approximate) Reward Models for Inference-Time Scaling: Sequential Monte Carlo and Beyond
Abstract
Lay Summary
Large language models can often solve harder problems by spending more computation at test time: instead of producing only one answer, they generate many possible reasoning paths, compare partial solutions along the way, and focus more effort on the most promising ones. A key component in this process is a scoring model that judges whether a partial reasoning path looks useful. In practice, however, such scoring models are never perfect. They are learned from data and inevitably make mistakes. This raises an important question: why can imperfect scoring models still help language models reason more efficiently? This paper gives a theoretical answer for a class of inference-time algorithms that repeatedly generate, score, discard, and copy candidate reasoning paths. We identify the kind of scoring error that matters most and show that if this error is small enough at each reasoning step, then the amount of computation needed can drop from growing exponentially with the reasoning length to growing only polynomially. In other words, even approximate scoring models can provide a dramatic efficiency gain, as long as their step-by-step errors are sufficiently controlled.