English

Taming the Thinker: Conditional Entropy Shaping for Adaptive LLM Reasoning

Computation and Language 2026-05-20 v1

Abstract

Entropy-based deep reasoning has emerged as a promising direction for improving the reasoning capabilities of Large Language Models (LLMs), but existing methods often either increase response length indiscriminately or shorten responses at the cost of accuracy. To better balance this trade-off, we introduce Conditional Entropy Shaping (CES), a framework that dynamically controls token-level response entropy, enabling LLMs to produce concise solutions on simple problems while encouraging deeper exploration on hard ones. Built on DAPO, CES uses token-level entropy as an uncertainty signal and applies a conditional bidirectional policy: it penalizes high-entropy "forking point" tokens on correct reasoning paths to improve conciseness, and rewards them on incorrect paths to encourage exploration and error correction. We implement CES on DeepSeek-R1-Distill-7B and evaluate it on 12 mathematical benchmarks. CES consistently improves average accuracy while reducing response length relative to DAPO, and supplementary experiments show similar trends on a smaller 1.5B backbone and on out-of-domain benchmarks.

Keywords

Cite

@article{arxiv.2605.19358,
  title  = {Taming the Thinker: Conditional Entropy Shaping for Adaptive LLM Reasoning},
  author = {Shuyu Wei and Jian Sun and Delai Qiu and Yining Wang and Shengping Liu and Jiaen Liang and Ying Fu and Wei Huang and Jitao Sang},
  journal= {arXiv preprint arXiv:2605.19358},
  year   = {2026}
}