Step-GRPO: Internalizing Dynamic Early Exit for Efficient Reasoning

📰 ArXiv cs.AI

arXiv:2604.16890v1 Announce Type: new Abstract: Large reasoning models that use long chain-of-thought excel at problem-solving yet waste compute on redundant checks. Curbing this overthinking is hard: training-time length penalties can cripple ability, while inference-time early-exit adds system overhead. To bridge this gap, we propose Step-GRPO, a novel post-training framework that internalizes dynamic early-exit capabilities directly into the model. Step-GRPO shifts the optimization objective

Published 21 Apr 2026
Read full paper → ← Back to Reads