Better Starts, Better Ends: Bootstrapped Iterative Self-Reasoning Distillation for Compressed Reasoning
Summary
Large reasoning models often produce lengthy chain-of-thought (CoT) traces with redundant computations, impacting efficiency. Existing on-policy self-distillation methods attempt to reduce this by matching a student model to a concise self-copy, but suffer from an initialization bottleneck due to noisy prefixes from verbose base models. To address this, BIRD (Bootstrapped Iterative Self-Reasoning Distillation) proposes a two-stage approach. First, it samples concise, answer-correct solutions from a base model under a brevity instruction, then performs a lightweight prompt-switch SFT step to embed this conciseness as default behavior. This "warm" model then undergoes on-policy reverse-KL distillation with a concise self-teacher, utilizing cleaner prefixes. BIRD achieves a stronger accuracy-efficiency trade-off on Qwen3 series models, improving Qwen3-8B's MATH-500 accuracy from 86.2% to 92.0% while reducing average response length from 3,099 to 1,115 tokens.
Key takeaway
For Machine Learning Engineers optimizing large reasoning models, addressing the initialization bottleneck in self-distillation is critical. If you are struggling with verbose chain-of-thought traces and inefficient inference, consider BIRD's two-stage approach. By first bootstrapping a "warm" model with brevity-instructed, correct traces and prompt-switch SFT, you can significantly improve accuracy and reduce response length. This method provides cleaner prefixes for subsequent on-policy distillation, leading to more efficient and accurate reasoning.
Key insights
Improving initial rollout distribution is crucial for efficient self-reasoning distillation in large language models.
Principles
- Brevity instruction can induce concise reasoning.
- Prompt-switch SFT embeds conciseness as default.
- Cleaner prefixes enhance distillation effectiveness.
Method
BIRD first samples concise, correct traces with brevity instruction, then applies prompt-switch SFT, followed by on-policy reverse-KL distillation.
In practice
- Use brevity instructions to generate concise CoT.
- Apply prompt-switch SFT for behavior transfer.
- Pre-filter noisy reasoning traces for distillation.
Topics
- Self-Reasoning Distillation
- Chain-of-Thought
- Large Language Models
- Model Compression
- Mathematical Reasoning
- LLM Efficiency
Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.