AI
New Technique Uses Selective Reprocessing to Boost LLM Reasoning Accuracy
Researchers propose a method that intelligently decides when language models should reconsider their predictions, cutting unnecessary computation while improving performance on math and coding tasks.
Large language models often struggle with complex reasoning tasks, particularly when computational resources are limited. A new approach called Think-at-Hard (TaH) offers a way to enhance reasoning capability without dramatically increasing model size—by having the model focus its extra processing power only where it's most needed.
The Problem with Over-Iteration#
Researchers found that when language models perform multiple passes to refine their predictions, they often run into what they call a "latent overthinking phenomenon." Most token predictions are actually correct after the very first pass, yet additional refinement rounds can inadvertently corrupt these correct answers. This suggests that blanket approaches—where every prediction gets re-processed equally—waste effort and sometimes backfire.
By studying an oracle strategy that skips unnecessary iterations and only refines genuinely difficult predictions, the team demonstrated potential gains of up to 7.3% in accuracy.
How Think-at-Hard Works#
TaH introduces three key innovations to address this inefficiency:
- Lightweight Neural Decider: A small neural module identifies which token predictions are likely incorrect after the initial forward pass, acting as a gatekeeper for additional iterations
- Depth-Aware LoRA Modules: During selective refinement rounds, these low-rank adaptation modules shift focus from general next-token prediction to specifically improving hard-to-predict tokens
- Duo-Causal Attention: An enhanced attention mechanism extends the traditional sequence dimension to include an iteration depth dimension, allowing information to flow between refinement passes while maintaining full parallelism
Experimental Results#
Testing across nine benchmarks—including mathematical reasoning, question-answering, and coding tasks—TaH demonstrated consistent improvements. Compared to models that always iterate, TaH achieved 3.8–4.4% better performance while skipping iterations on 93% of tokens. Against single-iteration baselines like Qwen3, gains ranged from 3.0–3.8%. When allowing a modest 3% increase in parameters through the LoRA and decider components, improvements expanded to 5.3–6.8% depending on the comparison point.
The approach essentially trades fine-grained token-level computation for smarter decision-making about when extra processing is worthwhile—helping models reason more effectively without proportionally larger parameter counts.

