Reinforcement learning (RL) has enhanced the capabilities of large language models (LLMs) through reward-driven training. Nevertheless, this process can introduce excessively long responses, inflating inference latency and computational overhead. Prior length-control approaches typically rely on fixed heuristic reward shaping, which can misalign with the task objective and require brittle tuning. In this work, we propose LACONIC, a reinforcement learning method that enforces a target token budget during training. Specifically, we update policy models using an augmented objective that combines the task reward with a length-based cost. To balance brevity and task performance, the cost scale is adaptively adjusted throughout training. This yields robust length control while preserving task reward. We provide a theoretical guarantee that support the method. Across mathematical reasoning models and datasets, LACONIC preserves or improves pass@1 while reducing output length by over 50%. It maintains out-of-domain performance on general knowledge and multilingual benchmarks with 44% fewer tokens. Moreover, LACONIC integrates into standard RL-tuning with no inference changes and minimal deployment overhead.
Figures & tables
Figure 1: The top panel skethces RL-tuning with a fixed length-aware shaping objective (blue). As the heuristically shaped objective generally differs from the true task reward R (red), optimizing it may converge to a policy πN that is suboptimal in R . The bottom panel sketches training with LACONIC . LACONIC adaptively updates the length-aware objective (green) so that it better aligns with the true task reward while achieving shorter outputs, yielding near-optimal policies.
Figure 2: Illustration of LACONIC. LACONIC alternates two steps: (1) in a primal update, the policy model is updated on an augmented objective that trades off task reward r with a length-aware cost c scaled by the dual variable λ ; (2) in a dual update, λ is adaptively updated to enforce a token budget constraint B by increasing when the average length Lˉ exceeds the budget B and decreasing otherwise. Together, these updates maximize task reward while meeting the budget on average.
Table 1: Evaluation results across four math benchmarks.
Table 2: Evaluation results across out-of-domain (OOD) benchmarks.
Figure 3: Ablation of the cost functions on DeepScaleR-1.5B with token budget B=1500 . We plot (a) accuracy reward and (b) average response length over training steps. For the linear cost, the Langrangian reward used in primal updates is computed by r(q,o)−λc(q,o) , where c(q,o)=(L(o)−B)/B . All other experiment setups and hyperparameters are identical.
Figure 4: Ablation of the token budget B on DeepScaleR-1.5B. We plot (a) accuracy reward; (b) average response length; and (c) dual variable λ over training steps with budgets B∈{1000,1500,1750,2000} . All other setups and hyperparameters are identical.
Token Budget
AIME2024
MATH
Minerva
Olympiad
Macro Average
Pass@1
# Tokens
Pass@1
# Tokens
Pass@1
# Tokens
Pass@1
# Tokens
Pass@1
# Tokens
+ LACONIC (2000)
34.79
3976
85.61
1556
31.76
1869
48.96
2448
50.28
2462
+ LACONIC (1750)
31.21
3803
84.82
1516
31.52
2059
48.88
2261
49.11
2410
+ LACONIC (1500)
29.58
3458
84.35
1353
31.48
1582
47.93
2084
48.33
2119
+ LACONIC (1000)
28.12
2665
83.75
1049
29.24
1189
47.07
1669
47.05
1643
Table 3: Evaluation results of the token-budget B ablation on DeepScaleR-1.5B across four math benchmarks.
Figure 5: Average computational resource usage of LACONIC (green) and GRPO (blue).
Figure 6: Ablation of the dual step size η on DeepScaleR-1.5B with B=1500 . We plot (a) accuracy reward; (b) average response length; and (c) dual variable λ over training steps with step size η∈{0.001,0.002,0.01} . We set Λ=0.3 for all experiments except for the run with η=0.01 and Λ=0.15 . All other setups and hyperparameters are identical.
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
Table 4: Evaluation results of the cost function ablation on DeepScaleR-1.5B across four math benchmarks.
Table 5: Evaluation results of the cost functions on Qwen-Math-1.5B across four math benchmarks.
Figure 7: Ablation of the cost functions on Qwen-Math-1.5B with a token budget B=550 : (a) reward accuracy, (b) average response length, (c) dual variable λ ,
Step size η
AIME2024
MATH
Minerva
Olympiad
Macro Average
Pass@1
# Tokens
Pass@1
# Tokens
Pass@1
# Tokens
Pass@1
# Tokens
Pass@1
# Tokens
η=0.001,Λ=0.3
27.50
3231
84.25
1356
30.92
1543
47.87
2090
47.64
2055
η=0.002,Λ=0.3
29.58
3458
84.35
1353
31.48
1582
47.93
2084
48.33
2119
η=0.01,Λ=0.3
29.79
3429
84.30
1365
31.42
1602
47.19
2097
48.18
2123
η=0.01,Λ=0.15
29.79
3477
84.20
1319
31.15
1562
47.83
2067
48.24
2106
Appendix
Table 6: Evaluation results of the token-budget B ablation across four math benchmarks.
Table 7: Evaluation results of length-control on Qwen3-8B across four math benchmarks.
Table 8: Evaluation results of fixed λ ablation across four math benchmarks.
Despite the strong performance of Large Language Models (LLMs) on complex instruction-following tasks, precise control of output length remains a persistent challenge. Existing methods primarily attempt to enforce length constraints by externally imposing length signals or optimization objectives, while largely overlooking the underlying limitation: the model's intrinsic deficit in length cognition. To address this, we propose LARFT (Length-Aware Reinforcement Fine-Tuning), a training framework that aligns the model's length cognition with its action. Specifically, LARFT integrates length-oriented reinforcement learning with a hindsight length awareness. By transforming on-policy data into hindsight self-awareness tasks where the model learns to identify the actual length of its own generation, LARFT jointly optimizes the model's internal representation of length information and refines its policy to satisfy length constraints, thereby achieving precise and reliable length instruction following. Extensive experiments across four base models demonstrate that LARFT outperforms existing baselines, achieving an average improvement of +20.92 points across three length instruction following benchmarks with only a marginal decline of -1.45 points on four general capability benchmarks.
Wei Zhang, Lintong Du, Yuanhe Zhang +4
Beijing University of Posts and Telecommunications, Beijing, China · Nanyang Technological University, Singapore, Singapore
Reinforcement learning significantly enhances LLM capabilities but suffers from a critical issue: length inflation, where models adopt verbosity or inefficient reasoning to maximize rewards. Prior approaches struggle to address this challenge in a general and lossless manner, primarily because additive penalties introduce a compensatory effect that creates optimization shortcuts, while heuristic gating strategies lack generality beyond binary feedback. To bridge this gap, we present Group Relative Reward Rescaling (GR3), which reframes length control as a multiplicative rescaling paradigm, effectively establishing a generalized, continuous, and reward-dependent gating mechanism. To further ensure lossless optimization, we incorporate group-relative regularization and advantage-aware calibration, which dynamically adapt length budgets to instance difficulty and preserve the advantage signal of high-quality trajectories. Empirically, across both RLHF and RLVR settings, GR3~maintains training dynamics and downstream performance comparable to standard GRPO while significantly mitigating length inflation, outperforming state-of-the-art length-regularized baselines.
Zichao Li, Jie Lou, Fangchen Dong +8
Chinese Information Processing Laboratory, Institute of Software, Chinese Academy of Sciences · University of Chinese Academy of Sciences · Xiaohongshu Inc.
Recent large reasoning models often develop long chain-of-thought responses during reinforcement learning (RL), resulting in high inference latency and deployment cost. Existing methods for response length control typically rely on explicit length penalties or additional control modules, which require careful tuning and may compromise reasoning quality. We propose Quadrant-weighted Sampling for Length-aware Policy Optimization (QLPO), a simple resampling-based variant of GRPO that introduces implicit length control without modifying the reward function. QLPO first over-generates candidate responses and then resamples the training group by preserving the empirical correct/incorrect ratio while favoring short correct responses and long incorrect responses. This reshapes the training distribution and implicitly encourages shorter model outputs. Across models ranging from 1.5B to 32B parameters, including both base models and strong reasoning models, QLPO consistently improves the accuracy-length trade-off. It reduces response length by 30% to 70% while preserving reasoning performance. These results suggest that structured resampling provides an effective and robust approach to efficient reasoning.
Siwei Chen, Siqi Chen, Xupeng Miao +1
School of Computer Science, Peking University · Beijing Key Laboratory of Software and Hardware Cooperative Artificial Intelligence Systems · Department of Electronic Engineering, Tsinghua University +1