PreSTO Speeds Up LLM Power Sampling Up to 2.33x via Predictive Subtree Prefetching

du_yilun · x · 2026-10-12

A new arXiv paper, PreSTO, accelerates power sampling — a test-time technique that boosts LLM reasoning by sampling from a sharpened model distribution without extra training — without changing the sampler's semantics.

Problem: The PowerMH sampler relies on sequential Metropolis–Hastings updates, where each accept/reject decision sets the state for the next proposal.

Method:

Results: Across 7 datasets and 5 base LLMs, PreSTO achieves median speedups of 1.71x, 1.91x, and 2.33x on PowerMH, EntropyCut, and MultiTryMH; on the first two it makes 3.9x fewer model calls, each only 2.1x longer.

Related event: PreSTO Speeds Up LLM Power Sampling by Up to 2.33x(2 posts)→

Original post →

More from Research

Research channel →