Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Accelerated Test-Time Scaling with Model-Free Speculative Sampling

About

Language models have demonstrated remarkable capabilities in reasoning tasks through test-time scaling techniques like best-of-N sampling and tree search. However, these approaches often demand substantial computational resources, creating a critical trade-off between performance and efficiency. We introduce STAND (STochastic Adaptive N-gram Drafting), a novel model-free speculative decoding approach that exploits the inherent redundancy in reasoning trajectories to achieve significant acceleration without compromising accuracy. Our analysis shows that reasoning paths frequently reuse similar reasoning patterns, enabling efficient model-free token prediction without requiring separate draft models. By introducing stochastic drafting and preserving probabilistic information through a memory-efficient logit-based N-gram module, combined with optimized Gumbel-Top-K sampling and data-driven tree construction, STAND significantly improves token acceptance rates. Extensive evaluations across multiple models and reasoning tasks (AIME-2024, GPQA-Diamond, and LiveCodeBench) demonstrate that STAND reduces inference latency by 60-65% compared to standard autoregressive decoding while maintaining accuracy. Furthermore, STAND consistently outperforms state-of-the-art speculative decoding methods across diverse inference patterns, including single-trajectory decoding, batch decoding, and test-time tree search. As a model-free approach, STAND can be applied to any existing language model without additional training, making it a powerful plug-and-play solution for accelerating language model reasoning.

Woomin Song, Saket Dingliwal, Sai Muralidhar Jayanthi, Bhavana Ganesh, Jinwoo Shin, Aram Galstyan, Sravan Babu Bodapati• 2025

Related benchmarks

TaskDatasetResultRank
Speculative DecodingLiveCodeBench
Speedup Factor2.67
66
Speculative DecodingAIME 2024
Throughput (T)69.15
48
Speculative DecodingGPQA Diamond
Throughput91.17
48
Code GenerationLCB
Throughput66.41
16
Mathematical ReasoningAIME
Throughput61.79
16
Question AnsweringGPQA
Throughput75.39
16
Showing 6 of 6 rows

Other info

Follow for update