Batched Contextual Reinforcement: A Task-Scaling Law for Efficient Reasoning
Bangji Yang, Hongbo Ma, Jiajun Fan, Ge Liu
TLDR
Batched Contextual Reinforcement (BCR) trains LLMs to solve multiple problems concurrently, significantly reducing token usage while maintaining or improving reasoning accuracy.
Key contributions
- BCR trains LLMs to solve N problems concurrently in a shared context, rewarded by per-instance accuracy.
- Discovers a task-scaling law: increasing N reduces per-problem token usage while gracefully maintaining accuracy.
- Achieves "free lunch" efficiency, reducing token usage by 15.8-62.6% while improving or maintaining accuracy.
- Demonstrates emergent self-regulated efficiency, avoiding explicit length penalties and optimization collapse.
Why it matters
This paper introduces a novel, simple training paradigm (BCR) that significantly improves the efficiency of LLM reasoning without sacrificing accuracy. It addresses the high inference costs of Chain-of-Thought models by offering a stable, implicit method for token budget control. This approach could make advanced LLM reasoning more practical and accessible.
Original Abstract
Large Language Models employing Chain-of-Thought reasoning achieve strong performance but suffer from excessive token consumption that inflates inference costs. Existing efficiency methods such as explicit length penalties, difficulty estimators, or multi-stage curricula either degrade reasoning quality or require complex training pipelines. We introduce Batched Contextual Reinforcement, a minimalist, single-stage training paradigm that unlocks efficient reasoning through a simple structural modification: training the model to solve N problems simultaneously within a shared context window, rewarded purely by per-instance accuracy. This formulation creates an implicit token budget that yields several key findings: (1) We identify a novel task-scaling law: as the number of concurrent problems N increases during inference, per-problem token usage decreases monotonically while accuracy degrades far more gracefully than baselines, establishing N as a controllable throughput dimension. (2) BCR challenges the traditional accuracy-efficiency trade-off by demonstrating a "free lunch" phenomenon at standard single-problem inference. Across both 1.5B and 4B model families, BCR reduces token usage by 15.8% to 62.6% while consistently maintaining or improving accuracy across five major mathematical benchmarks. (3) Qualitative analyses reveal emergent self-regulated efficiency, where models autonomously eliminate redundant metacognitive loops without explicit length supervision. (4) Crucially, we empirically demonstrate that implicit budget constraints successfully circumvent the adversarial gradients and catastrophic optimization collapse inherent to explicit length penalties, offering a highly stable, constraint-based alternative for length control. These results prove BCR practical, showing simple structural incentives unlock latent high-density reasoning in LLMs.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.