Search papers, labs, and topics across Lattice.
This paper addresses the issue of overthinking in reasoning language models, which often leads to unproductive self-reflection and token consumption without improving answers. By leveraging intermediate answer commitments as a proxy for productive reflection, the authors introduce DASH (Drift Aware advantage SHaping), a method that assigns segment-level credit based on the correctness of reasoning segments. The results show that DASH outperforms existing methods on competition-level math benchmarks, achieving higher accuracy while simultaneously reducing overthinking behaviors.
Overthinking in language models can be mitigated by segment-level credit assignment, leading to a significant boost in accuracy on challenging tasks.
Reasoning language models frequently overthink: generating extended chains of behaviors such as hedging, approach abandonment, and self contradiction that consume tokens without improving answers. We show that these behaviors are not merely a consequence of length; even when controlling for response length, incorrect traces exhibit higher rates of unproductive self-reflection than correct ones. Addressing this requires identifying where self-reflection helps vs hurts, but obtaining these step-level annotations is costly. We observe that intermediate answer commitments within reasoning traces can provide a cheap proxy: by comparing each final answer candidate in the trace to the ground truth, we can determine whether subsequent reflection is productive without any additional supervision. Building on this insight, we propose DASH (Drift Aware advantage SHaping), which assigns segment-level credit based on whether each reasoning segment leads toward or away from correctness. On competition-level math benchmarks, DASH achieves the highest accuracy where overthinking is prevalent (Average Accuracy: 59.45% vs. 58.1% Dr.GRPO vs. 56.95% GRPO) while reducing overthinking behaviors and achieving more productive self-correction than baselines.