Probes flag DeepSeek-R1 reasoning failure by token 50
TL;DR
- Chain-of-thought runs that finished within budget scored 90.3% on AIME 1983-2024, while runs that ran out of tokens scored just 6.6%.
- Only 62.0% of DeepSeek-R1-Distill-Qwen-7B runs converged at all, making token-budget saturation a common failure mode rather than a rare one.
- Linear probes on hidden-state activations predicted convergence as early as token 50, with AUC 0.608 and a borderline p=0.063.
A quiet arxiv paper this month makes an unusually practical claim about reasoning models: you can tell, quite early in the generation, whether the model is actually going to finish its thought or just burn through its token budget without landing on an answer.
The setup is straightforward. The authors run DeepSeek-R1-Distill-Qwen-7B on AIME math problems from 1983 to 2024. When the chain of thought converges within its budget, accuracy is 90.3%. When it doesn't, accuracy collapses to 6.6%. Only 62.0% of runs converge at all. So the dominant failure mode here isn't the model picking a wrong answer, it's the model never getting around to committing to one before the tokens run out, and that failure is common.
The more interesting bit is what they can see inside the model while this is happening. Linear probes trained on hidden-state activations start picking up the eventual convergence outcome as early as token 50, with the sharpest signal at layer 20 around token 150. The reported probe AUC is 0.608 with a five-fold cross-validation spread of plus or minus 0.080, and a permutation test at p=0.063. That is suggestive, not conclusive, and the authors say as much.
If the effect holds up, the practical payoff is straightforward: early-exit inference and adaptive compute allocation. A router could look at the hidden state a few dozen tokens in and decide whether to keep spending on this trajectory, restart, or escalate the query to a bigger model. For anyone paying by the reasoning token, that is a real cost lever, and it fits the direction the open-weight inference stack has already been moving in.
The honest caveats are worth stating. This is one 7B distilled model on one benchmark family, with a borderline p-value and a modest AUC. What the paper doesn't tell you is how the signal generalizes to frontier-scale reasoners, to non-math domains like code or agentic tool use, or to models trained with different chain-of-thought recipes. Still, the reframing is the part inference teams should track: treating reasoning non-convergence as a detectable internal state rather than a black-box failure.
Shared on Bluesky by 2 AI experts
Originally reported by arxiv.org
Read the original article →Original headline: Token Budget Saturation and Mechanistic Early Detection of Reasoning Non-Convergence in Chain-of-Thought Models