Know when to stop: Segment-level credit assignment for reducing overthinking
A benchmark which assigns segmentlevel credit based on whether each reasoning segment leads toward or away from correctness.
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 (AIME25: 50.8% vs. 45.4% GRPO) while reducing overthinking behaviors and achieving more productive self-correction than baselines.
Latest publications
Decomposing the delta: what do models actually learn from preference pairs?
An investigation of two distinct notions of quality delta in preference data.
COLMEfficient reinforcement learning for long-horizon tool-use agentic tasks
A modular training system for RL in dual-control tool-use environments.
COLMRefusal tokens: a simple way to calibrate refusals
Refusal tokens enable controlling a single model's refusal rates without the need of any further fine-tuning.
COLM