English

The Model Knows, the Decoder Finds: Future Value Guided Particle Power Sampling

Artificial Intelligence 2026-05-13 v2 Machine Learning

Abstract

A recurring pattern in "reasoning without training" is that base LLMs already assign non-trivial probability mass to correct multi-step solutions; the bottleneck is locating these modes efficiently at inference time. Power sampling provides a principled way to bias decoding toward such modes by targeting p_theta(x)^alpha with alpha > 1, but practical approximations must account for future-dependent correction factors that determine which prefixes remain promising. We introduce Auxiliary Particle Power Sampling (APPS), a blockwise particle algorithm for approximating the sequence-level power target with a bounded population of partial solutions. APPS propagates hypotheses in parallel using proposal-corrected power reweighting and refines their survival through future-value-guided selection at resampling boundaries. This redistributes finite compute across competing prefixes rather than committing to a single unfolding path, while providing a direct scaling knob in the particle count and predictable peak memory. We instantiate the future-value signal with short-horizon rollouts and also study an amortized variant that replaces rollouts with a lightweight learned selection head. Across reasoning benchmarks, APPS improves the accuracy-runtime trade-off of training-free decoding and suggests that part of the gap to post-trained systems can be recovered through more faithful inference-time power approximation.

Keywords

Cite

@article{arxiv.2605.02427,
  title  = {The Model Knows, the Decoder Finds: Future Value Guided Particle Power Sampling},
  author = {Tu Nguyen and Matthieu Zimmer and Rasul Tutunov and Xiaotong Ji and Haitham Bou Ammar},
  journal= {arXiv preprint arXiv:2605.02427},
  year   = {2026}
}
R2 v1 2026-07-01T12:48:17.663Z