Martingale Foresight Sampling: A Principled Approach to Inference-Time LLM Decoding
This addresses the challenge of finding globally optimal reasoning paths in LLMs for applications requiring complex inference, though it appears incremental as it builds on existing foresight sampling methods.
The paper tackles the problem of short-sighted autoregressive decoding in LLMs by introducing Martingale Foresight Sampling, a principled framework that reformulates decoding as identifying an optimal stochastic process, resulting in improved accuracy and computational efficiency on six reasoning benchmarks.
Standard autoregressive decoding in large language models (LLMs) is inherently short-sighted, often failing to find globally optimal reasoning paths due to its token-by-token generation process. While inference-time strategies like foresight sampling attempt to mitigate this by simulating future steps, they typically rely on ad-hoc heuristics for valuing paths and pruning the search space. This paper introduces Martingale Foresight Sampling (MFS), a principled framework that reformulates LLM decoding as a problem of identifying an optimal stochastic process. By modeling the quality of a reasoning path as a stochastic process, we leverage Martingale theory to design a theoretically-grounded algorithm. Our approach replaces heuristic mechanisms with principles from probability theory: step valuation is derived from the Doob Decomposition Theorem to measure a path's predictable advantage, path selection uses Optional Stopping Theory for principled pruning of suboptimal candidates, and an adaptive stopping rule based on the Martingale Convergence Theorem terminates exploration once a path's quality has provably converged. Experiments on six reasoning benchmarks demonstrate that MFS surpasses state-of-the-art methods in accuracy while significantly improving computational efficiency. Code will be released at https://github.com/miraclehetech/EACL2026-Martingale-Foresight-Sampling.