English

DAST: Difficulty-Adaptive Slow-Thinking for Large Reasoning Models

Machine Learning 2026-01-13 v3 Artificial Intelligence

Abstract

Recent advancements in slow thinking reasoning models have shown exceptional performance in complex reasoning tasks. However, these models often exhibit overthinking (generating redundant reasoning steps for simple problems), leading to excessive computational resource usage. While current mitigation strategies uniformly reduce reasoning tokens, they risk degrading performance on challenging tasks that require extended reasoning. This paper introduces Difficulty-Adaptive Slow Thinking (DAST), a novel framework that enables models to autonomously adjust the length of Chain-of-Thought (CoT) based on problem difficulty. We first propose a Token Length Budget (TLB) metric to quantify difficulty, then leverage budget-aware reward shaping and budget preference optimization to implement DAST. DAST penalizes overlong responses for simple tasks while incentivizing sufficient reasoning for complex problems. Experiments on diverse datasets and model scales demonstrate that DAST effectively mitigates overthinking (reducing token usage by over 30\% on average) while preserving reasoning accuracy on complex problems. Our codes and models are available at https://github.com/AnonymousUser0520/AnonymousRepo01.

Keywords

Cite

@article{arxiv.2503.04472,
  title  = {DAST: Difficulty-Adaptive Slow-Thinking for Large Reasoning Models},
  author = {Yi Shen and Jian Zhang and Jieyun Huang and Shuming Shi and Wenjing Zhang and Jiangze Yan and Ning Wang and Kai Wang and Zhaoxiang Liu and Shiguo Lian},
  journal= {arXiv preprint arXiv:2503.04472},
  year   = {2026}
}

Comments

EMNLP 2025 Industry Track

R2 v1 2026-06-28T22:09:16.447Z