Anti-Length Shift: Dynamic Outlier Truncation for Training Efficient Reasoning Models
By: Wei Wu , Liyi Chen , Congxi Xiao and more
Potential Business Impact:
Makes AI answer questions faster and smarter.
Large reasoning models enhanced by reinforcement learning with verifiable rewards have achieved significant performance gains by extending their chain-of-thought. However, this paradigm incurs substantial deployment costs as models often exhibit excessive verbosity on simple queries. Existing efficient reasoning methods relying on explicit length penalties often introduce optimization conflicts and leave the generative mechanisms driving overthinking largely unexamined. In this paper, we identify a phenomenon termed length shift where models increasingly generate unnecessary reasoning on trivial inputs during training. To address this, we introduce Dynamic Outlier Truncation (DOT), a training-time intervention that selectively suppresses redundant tokens. This method targets only the extreme tail of response lengths within fully correct rollout groups while preserving long-horizon reasoning capabilities for complex problems. To complement this intervention and ensure stable convergence, we further incorporate auxiliary KL regularization and predictive dynamic sampling. Experimental results across multiple model scales demonstrate that our approach significantly pushes the efficiency-performance Pareto frontier outward. Notably, on the AIME-24, our method reduces inference token usage by 78% while simultaneously increasing accuracy compared to the initial policy and surpassing state-of-the-art efficient reasoning methods.
Similar Papers
Correct, Concise and Complete: Multi-stage Training For Adaptive Reasoning
Computation and Language
Makes AI think less to solve problems faster.
From Long to Short: LLMs Excel at Trimming Own Reasoning Chains
Artificial Intelligence
Makes smart computers solve problems faster, simpler.
Leash: Adaptive Length Penalty and Reward Shaping for Efficient Large Reasoning Model
Artificial Intelligence
Makes AI write shorter answers without losing quality.