AAAI 2026A STUDY IN EFFICIENT REASONING

Give every question
just enough
thought.

SABER: Switchable and Balanced Training
for Efficient LLM Reasoning

Kai Zhao1* · Yanjun Zhao1* · Jiaming Song1
Shien He1 · Lusheng Zhang1 · Qiang Zhang1,2 · Tianjiao Li1†

¹ Bilibili Inc.   ² Xi’an Jiaotong University

* Equal contribution   † Corresponding author

Some questions need a moment. Others deserve a deeper look. SABER gives you the choice—four reasoning modes in one model, learned through budget-aware reinforcement learning.

One model, four ways to thinkAn illustrative map of four paths from a question to an answer. Select a reasoning mode using the buttons below.A FIELD GUIDE TO THINKINGFIG. 01DeepThinkroom to exploreCoreThinkspace to reflectFastThinkthe essential stepsNoThink · directly?QUESTIONANSWERdifferent routes, your choice.ONE MODEL / FOUR REASONING MODES
02 / 04

Select a mode to trace its path. A conceptual illustration, not measured trajectories.

65.4%↓

fewer thinking tokens

7,866 → 2,719
+2.9pp

higher accuracy

80.6% → 83.5%
2K

training examples

1K math + 1K code

A smaller footprint.
A stronger answer.

MATH · 1.5B · FastThink
Compared with DeepSeek-R1-Distill-Qwen-1.5B.
Table 1 in the paper.

01 / THE IDEA

Thinking longer
is a choice.
Not a default.

Long chains of thought help language models solve hard problems. Applied to every question, they also bring unnecessary tokens, latency, and cost.

SABER learns when to compress and when to preserve reasoning. A system prompt lets the user choose a direct answer, essential steps, a fuller explanation, or unrestricted exploration—all from the same trained model.

One model. Four modes. A deliberate trade-off between reasoning depth and inference cost.

01

NoThink

A direct answer, without an explicit thinking trace.

02

FastThink

Keep the essential steps. Spend fewer tokens.

03

CoreThink

Leave room for intermediate steps and reflection.

04

DeepThink

Explore and self-check, without a target budget.

02 / HOW IT WORKS

Teach the model
a sense of proportion.

Start from a distilled reasoning model. Calibrate the budget to each example, then learn to switch with direct reinforcement learning.

01
PROFILE THE BASE MODEL

Observe before you constrain.

Profile the base model’s answer correctness and thinking length on 1K math and 1K code examples.

02
CALIBRATE EACH EXAMPLE

Compress with care.

Downgrade budgets for solved examples. Give unsolved ones more room. Add direct-answer versions for NoThink.

03
TRAIN WITH DIRECT RL

Learn the switch.

Use GRPO with rewards for format, correctness, budget alignment, and length-ratio safeguards. No extra SFT warmup.

THE LEARNING SIGNAL

r = format + answer + length + ratio

Correct answers matter. So does how you get there.
A closer look at budget calibration+

The base model’s thinking length is grouped around 128 / 4,096 / 16,384 tokens. Correctly solved examples receive shorter target budgets; among failed examples, half retain their original tier and half are left unrestricted.

A length-ratio reward penalizes generations outside 0.2–1.2× the base thinking length, helping prevent degenerate shortening. Budgets are prompts and training targets, not guaranteed hard cutoffs.

Original SABER framework: base-model profiling, budget calibration, and mode-conditioned RL training with a verifier.
Original framework diagram · Figure 1, AAAI 2026 paper.
03 / THE EVIDENCE

Less can be more.
Explore the trade-off.

Every point is a reported result. Switch the model size and benchmark to see where each reasoning mode lands.

↖ Higher accuracy, fewer tokens
Accuracy versus average thinking tokens
Source: paper Tables 1–2. Length = mean explicit thinking tokens; answer tokens are not included. NoThink still generates an answer.
View the exact numbers+
Model / modeAccuracy (%)Thinking tokens
Download benchmark data (JSON) ↗
↗

Beyond the training domains. The same recipe scales to 7B and transfers to unseen logical reasoning. On LiveBench-R, DeepThink reaches 38.3% accuracy versus 36.4% for the base model. Faster modes trade some accuracy for greater savings.

04 / A CLOSER LOOK

The same question.
A different depth.

A small arithmetic problem makes the difference visible: the core solution stays; the amount of reflection changes.

THE QUESTION

How many positive whole-number
divisors does 196 have?

THE ANSWER, IN EVERY MODE9✓
Core stepsReflectionSelf-check

Illustrative summaries adapted from Figure 3. The three thinking traces are condensed; NoThink is a direct-answer illustration. This is not live model inference.

05 / WHAT MATTERS

A good shortcut
still needs a foundation.

↘

Calibrate, then compress.

Aggressively shortening every example hurts accuracy. Correctness-aware budget assignment preserves room for problems the base model cannot yet solve.

∅

Train for the quiet mode.

On MATH with the 1.5B model, NoThink accuracy is 76.9%. Removing NoThink training data lowers it to 62.3%—direct answers need explicit practice.

⇄

Keep the trade-off visible.

Efficiency gains are not universal accuracy gains. The four modes expose the balance, so users can choose the reasoning depth their task deserves.

Ablation evidence: Tables 1 and 3; cross-scale trade-offs: Table 2.

06 / READ & BUILD ON IT

The whole
thought process.

SABER: Switchable and Balanced Training for Efficient LLM Reasoning

BIBTEX / arXiv version
@article{zhao2025saber,
  title   = {SABER: Switchable and Balanced
             Training for Efficient LLM Reasoning},
  author  = {Zhao, Kai and Zhao, Yanjun and
             Song, Jiaming and He, Shien and
             Zhang, Lusheng and Zhang, Qiang
             and Li, Tianjiao},
  journal = {arXiv preprint arXiv:2508.10026},
  year    = {2025},
  url     = {https://arxiv.org/abs/2508.10026}
}