TRiMS: Real-Time Tracking of Minimal Sufficient Length for Efficient Reasoning via RL
Organizations: Baidu Inc. · Shenzhen University · Harbin Institute of Technology
Abstract
Large language models achieve breakthroughs in complex reasoning via long chain-of-thought sequences. However, this often leads to severe reasoning inflation, causing substantial computational redundancy. To maximize Intelligence per Token, we introduce a theoretical metric, MSL-Minimal Sufficient Length. MSL rigorously characterizes the shortest reasoning length that preserves answer correctness. We provide a recursive definition based on independently sampled sequences and prove the existence of its limit, establishing the first measurable lower bound for reasoning-chain compression. Building on an analysis of mainstream CoT compression strategies, we identify key structural factors enabling a model to approach MSL. Based on these insights, we propose TRiMS which employs the GRPO algorithm in conjunction with MSL-based estimation during training, while mitigating instabilities during the training process through dynamic batch aggregation and advantage computation using batch-level standard deviation. TRiMS achieves over 80% CoT token reduction with a minor accuracy boost across all benchmarks.
Figures & tables
| Method | AMC23 | AIME24 | MATH-500 | Minerva | Olympiad | Avg | |||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Acc | Token | Acc | Token | Acc | Token | Acc | Token | Acc | Token | Acc (%) | Token (%) | IPT | |
| DeepSeek-R1-Distill-Qwen-1.5B | |||||||||||||
| Vanilla | 61.1 | 8458.9 | 29.2 | 12303.4 | 82.3 | 4619.2 | 25.7 | 6284.1 | 41.0 | 8994.3 | – | – | 7.2 |
| LC-R1 | 56.5 | 3862.9 | 19.3 | 6737.5 | 71.2 | 2080.4 | 26.5 | 3827.3 | 37.9 | 3950.1 | -11.9 | -49.9 | 13.6 |
| Laser | 60.2 | 4714.9 | 30.4 | 5713.1 | 84.1 | 1974.1 | 33.1 | 3218.3 | 44.3 | 5645.9 | +8.4 | -48.2 | 15.8 |
| AutoThink | 57.8 | 4244.9 | 30.0 | 8984.4 | 85.6 | 2111.5 | 29.0 | 3031.5 | 44.7 | 5072.6 | +1.9 | -45.3 | 14.9 |
| Method | MMLU | ||||
|---|---|---|---|---|---|
| Acc | Token | Acc | Token | IPT | |
| DeepSeek-R1-Distill-Qwen-1.5B | |||||
| Vanilla | 38.1 | 1718.7 | – | – | 22.2 |
| TRiMS | 48.2 | 388.1 | +26.5 | -77.4 | 124.2 |
| DeepSeek-R1-Distill-Qwen-7B | |||||
| Vanilla | 63.7 | 1158.5 | – | – | 55.0 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Configuration | Temperature | top- | top- | Default |
|---|---|---|---|---|
| Low-temperature | 0.3 | 0.95 | No | |
| Moderate-temperature | 0.6 | 0.95 | Yes | |
| High-temperature | 1.0 | 0.95 | No | |
| Constrained Stochastic | 0.6 | 1.0 | 100 | No |
| Method | 1.0 | 2.0 | 3.0 | 4.0 | 5.0 |
|---|---|---|---|---|---|
| Vanilla | 0.9070 | 0.9333 | 0.8952 | 0.7969 | 0.6567 |
| TRiMS | 0.9535 | 0.9111 | 0.9429 | 0.8125 | 0.6493 |
| Method | AMC23 | AIME24 | MATH-500 | Minerva | Olympiad | Avg | |||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Acc | Token | Acc | Token | Acc | Token | Acc | Token | Acc | Token | Acc (%) | Token (%) | IPT | |
| Qwen3-4B | |||||||||||||
| Vanilla | 77.1 | 9060.0 | 57.9 | 11933.0 | 91.6 | 5021.6 | 41.5 | 6673.1 | 58.2 | 9243.0 | – | – | 8.8 |
| TRiMS | 80.7 | 1046.2 | 53.5 | 1713.6 | 91.0 | 662.2 | 41.9 | 783.0 | 57.9 | 1081.6 | -0.63 | -88.3 | 70.6 |
Efficient Language Model Reasoning
Efficient language model reasoning studies how to improve problem-solving accuracy while reducing inference, memory, latency, and training costs.
See all 370 papersCRISP: Compressing Redundancy in Chain-of-Thought via Intrinsic Saliency Pruning
[object Object] acts as an information anchor, where its attention pattern effectively demarcates essential reasoning from redundancy. Based on this finding, we design a policy that utilizes these intrinsic attention signals to guide atomic compression operations. In contrast to coarse-grained pruning strategies, CRISP strategically distills the reasoning chain to maximize information density while preserving logical coherence. Empirical results across various backbone models and mathematical datasets demonstrate that CRISP achieves a 50-60% reduction in token count without compromising accuracy, effectively mitigating the efficiency bottleneck of long-context reasoning. We open-source our implementation to facilitate further research in efficient reasoning.