English

Reasoning as Compression: Unifying Budget Forcing via the Conditional Information Bottleneck

Machine Learning 2026-05-19 v2

Abstract

\ac{CoT} prompting improves LLM accuracy on complex tasks but often increases token usage and inference cost. Existing ``Budget Forcing'' methods reduce cost via fine-tuning with heuristic length penalties, suppressing both essential reasoning and redundant filler. We recast efficient reasoning as a lossy compression problem under the \ac{IB} principle, and identify a key theoretical gap when applying naive \ac{IB} to transformers: attention violates the Markov property between prompt, reasoning trace, and response. To resolve this issue, we model \ac{CoT} generation under the \ac{CIB} principle, where the reasoning trace ZZ acts as a computational bridge that contains only the information about the response YY that is not directly accessible from the prompt XX. This yields a general Reinforcement Learning objective: maximize task reward while compressing completions under a prior over reasoning traces, subsuming common heuristics (e.g., length penalties) as special cases (e.g., uniform priors). In contrast to naive token-counting approaches, we introduce a semantic prior that measures token cost by surprisal under a language model. Crucially, the prior is queried only for token-level log-probabilities, adding negligible overhead to the training loop. Empirically, our \ac{CIB} objective prunes reasoning redundancy while preserving fluency and logic, improving accuracy at moderate compression and enabling aggressive compression with minimal accuracy drop. These gains generalize across model families and task domains, confirming \ac{CIB} as a domain-agnostic CoT compression framework.

Keywords

Cite

@article{arxiv.2603.08462,
  title  = {Reasoning as Compression: Unifying Budget Forcing via the Conditional Information Bottleneck},
  author = {Fabio Valerio Massoli and Andrey Kuzmin and Arash Behboodi},
  journal= {arXiv preprint arXiv:2603.08462},
  year   = {2026}
}
R2 v1 2026-07-01T11:10:27.933Z