What they did
Muennighoff et al. analyzed attention patterns during long reasoning and found that intermediate tokens contribute progressively less to the model's current computation compared with the initial prompt/prefix and the most recent tokens. Based on this, they introduced Prefix Sliding: an inference-time method that keeps the fixed instruction/tool prefix plus a sliding window of the last few thousand tokens, discarding all other reasoning tokens. The authors evaluated the method both as a drop-in inference trick for existing models and as part of reinforcement learning training, comparing against baselines such as summarizing intermediate tokens and plain sliding-window attention.
Key findings
- Applying Prefix Sliding to existing models without any training makes them up to 3x faster while maintaining performance.
- RL training with Prefix Sliding allows test-time scaling to reasoning traces beyond 100,000 tokens, achieving better performance than standard RL training.
- Ablations show Prefix Sliding outperforms both summarizing intermediate tokens and vanilla sliding-window attention.
- The core empirical finding — that most intermediate tokens lose importance as reasoning progresses — directly motivates the method.
Why it matters
Capping memory during reasoning removes a major bottleneck for test-time scaling. Instead of quadratic memory growth with trace length, Prefix Sliding makes long-horizon thinking practical on existing hardware, potentially enabling frontier models to solve harder problems by reasoning much longer.
Caveats
The empirical observation that intermediate tokens lose importance may not hold universally across all tasks and model generations; the window size remains a hyperparameter that likely needs tuning. The reported gains are measured on specific benchmarks and settings, and the method's effectiveness on tasks requiring retrieval or re-examination of early reasoning steps is not fully addressed.