In Their Own Words: Reasoning Traces Tailored for Small Models Make Them Better Reasoners
cs.CL
Submitted: 2025-09-26
Updated: 2026-09-26
Code: https://github.com/jaeh8nkim/equigranular
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Stronger learning signals do not necessarily produce a stronger small reasoning model.
Terminology
Abstract
Stronger learning signals do not necessarily produce a stronger small reasoning model. Off-policy supervised fine-tuning (SFT) supplies stronger traces that may be incompatible with the student. On-policy post-training instead remains constrained by the quality of the student's own trajectories: reinforcement learning (RL) provides little learning signal because a small student rarely produces a correct rollout. Consequently, effective post-training must supply stronger reasoning in a form the student can learn from. We hypothesize that a small number of tokens with very low probability under the student can make an otherwise strong reasoning trace difficult to learn. To test this hypothesis, we introduce Interleaved-Policy Distillation (IPD), which generates SFT data balancing teacher guidance and student compatibility. During generation, the student replaces teacher proposals that have low probability under its own policy, allowing the teacher to continue reasoning from a prefix the student can support. IPD improves reasoning across model families, datasets, and varying degrees of student intervention; for example, it raises the seven-benchmark average from 24.55 to 29.72 on Qwen3-0.6B/s1K. It outperforms baselines with comparable training budgets and broadens problem-solving coverage, while ablations show that filtering traces or masking losses cannot match the gains from revising trajectories during generation. Analyses link transfer failures to the rare tokens the student finds least likely and show that the source of the tokens also matters. Small models reason better when they learn strong solutions in a form they can reach in their own words.
Sources
- Phi-4-Mini Technical Report: Compact yet Powerful Multimodal Language Models via Mixture-of-LoRAs
- SplitReason: Learning To Offload Reasoning
- Role of obstacle softness in the diffusive behavior of active Particles
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- Gemma 3 Technical Report
- The First Few Tokens Are All You Need: An Efficient and Effective Unsupervised Prefix Fine-Tuning Method for Reasoning Models
- Stationary flows for viscous heat-conductive fluid in a perturbed half-space
- Stabilizing Knowledge, Promoting Reasoning: Dual-Token Constraints for RLVR
- Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning
- Effect of Dielectric Wakefields in a Capillary Discharge for Plasma Wakefield Acceleration
- Phi-4-Mini-Reasoning: Exploring the Limits of Small Reasoning Language Models in Math
- Qwen2.5 Technical Report
- Speculative Thinking: Enhancing Small-Model Reasoning with Large Model Guidance at Inference Time
- Scaling Relationship on Learning Mathematical Reasoning with Large Language Models
- Transformers Utilization in Chart Understanding: A Review of Recent Advances & Future Trends
- Short-time Fourier Transform-based Signal Recovery for Modulo Analog-to-Digital Converters
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering