English

Stable Adaptive Thinking via Advantage Shaping and Length-Aware Gradient Regulation

Machine Learning 2026-02-27 v1 Artificial Intelligence Computation and Language

Abstract

Large reasoning models (LRMs) achieve strong performance through extended reasoning traces, but they often exhibit overthinking behavior for low-complexity queries. Existing efforts to mitigate this issue are fundamentally limited by unstable accuracy-efficiency trade-offs and poor robustness to heterogeneous reasoning behaviors. To address these challenges, we propose a two-stage framework for stable adaptive thinking in LRMs. The framework first applies Hybrid Fine-Tuning to expose the model to both thinking and no-thinking behaviors, establishing well-conditioned initialization. It then performs adaptive reinforcement learning with Correctness-Preserving Advantage Shaping (CPAS) to avoid suppressing correct long-chain reasoning, and Length-Aware Gradient Regulation (LAGR) to stabilize optimization under severe reasoning-length heterogeneity. Extensive experiments on Qwen2.5-1.5B and 7B show consistent improvements over strong baselines, achieving up to +3.7/+3.6 accuracy points while reducing generated tokens by 40.6%/43.9%. Further analyses across varying problem difficulties and out-of-distribution tasks confirm the robustness and generalization of our approach.

Keywords

Cite

@article{arxiv.2602.22556,
  title  = {Stable Adaptive Thinking via Advantage Shaping and Length-Aware Gradient Regulation},
  author = {Zihang Xu and Haozhi Xie and Ziqi Miao and Wuxuan Gong and Chen Qian and Lijun Li},
  journal= {arXiv preprint arXiv:2602.22556},
  year   = {2026}
}

Comments

15 pages, 7 figures

R2 v1 2026-07-01T10:53:13.394Z