English

Martingale Foresight Sampling: A Principled Approach to Inference-Time LLM Decoding

Machine Learning 2026-01-23 v1 Artificial Intelligence

Abstract

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.

Keywords

Cite

@article{arxiv.2601.15482,
  title  = {Martingale Foresight Sampling: A Principled Approach to Inference-Time LLM Decoding},
  author = {Huayu Li and ZhengXiao He and Siyuan Tian and Jinghao Wen and Ao Li},
  journal= {arXiv preprint arXiv:2601.15482},
  year   = {2026}
}
R2 v1 2026-07-01T09:14:57.125Z