English

Reject, Resample, Repeat: Understanding Parallel Reasoning in Language Model Inference

Machine Learning 2026-03-10 v1 Artificial Intelligence Computation and Language Statistics Theory Machine Learning Statistics Theory

Abstract

Inference-time methods that aggregate and prune multiple samples have emerged as a powerful paradigm for steering large language models, yet we lack any principled understanding of their accuracy-cost tradeoffs. In this paper, we introduce a route to rigorously study such approaches using the lens of *particle filtering* algorithms such as Sequential Monte Carlo (SMC). Given a base language model and a *process reward model* estimating expected terminal rewards, we ask: *how accurately can we sample from a target distribution given some number of process reward evaluations?* Theoretically, we identify (1) simple criteria enabling non-asymptotic guarantees for SMC; (2) algorithmic improvements to SMC; and (3) a fundamental limit faced by all particle filtering methods. Empirically, we demonstrate that our theoretical criteria effectively govern the *sampling error* of SMC, though not necessarily its final *accuracy*, suggesting that theoretical perspectives beyond sampling may be necessary.

Keywords

Cite

@article{arxiv.2603.07887,
  title  = {Reject, Resample, Repeat: Understanding Parallel Reasoning in Language Model Inference},
  author = {Noah Golowich and Fan Chen and Dhruv Rohatgi and Raghav Singhal and Carles Domingo-Enrich and Dylan J. Foster and Akshay Krishnamurthy},
  journal= {arXiv preprint arXiv:2603.07887},
  year   = {2026}
}
R2 v1 2026-07-01T11:09:32.927Z