Suorong Yang, Jiaxin Wang, Jiaming Su, Chentao Yue, Furao Shen, Peng Ye, Soujanya Poria
7 min
Abstract
Large language models have achieved remarkable advances in complex reasoning. However, scaling reasoning post-training with ever-larger corpora can substantially increase training cost without proportional performance gains. In this study, we therefore ask: how far can reasoning supervision be compressed without sacrificing capability? To this end, we propose an uncertainty-guided progressive contraction framework that concentrates supervision and optimization on the reasoning trajectories most valuable for learning. Specifically, rather than treating a teacher-generated chain-of-thought (CoT) as a single realized path, we construct uncertainty-profiled reasoning supervision by building Monte Carlo reasoning trees around intermediate reasoning states to expose process-level uncertainty. This uncertainty first contracts the original corpus into a compact teacher-informed pool. A second, learner-aware contraction then performs online routing during SFT, concentrating optimization according to the student's evolving learning state. Together, the two stages progressively contract both supervision and optimization, substantially reducing the student training budget. Experiments across diverse reasoning benchmarks and LLM families show that our method uses as little as 0.22% of the optimization budget of full-data SFT, over 450x less training exposure, while maintaining or improving reasoning performance. Notably, our framework generalizes well across heterogeneous student families, black-box teachers, and weak-to-strong transfer. This indicates that it is not only a way to accelerate training but can also enhance reasoning capability.
Alex: Do correct, confident solutions actually contain many of these forks? Or is this a rare edge case?
Sam: Not rare, by their measurements. Take the teacher's most confident solutions, the top five percent. About eighteen percent of those still land in the top tenth for this step-level uncertainty. Among correct solutions, it's about three in ten. Confidence and correctness, in their data, mask shaky steps.
Alex: Does this need access to the teacher's internals, like token probabilities?
Sam: No, and that's a practical point. It only uses text prefixes, sampled continuations, and whether the final answer checks out. So it works with teachers you can only reach through an API.
Alex: Okay, so that's how they score solutions. What happens with the scores?
Sam: Two contractions. First, they keep only the most uncertain fraction of verified solutions, say ten percent, or as low as one percent. That pool depends only on the teacher, so it can be reused for any student. Second, the student does one pass over that pool. Then training narrows to the solutions the student still finds hardest, measured by its own loss. Those scores update as training goes on.
Alex: So the teacher says where the interesting reasoning is, and the student says which of that it hasn't learned yet.
Sam: That's a fair summary of the design.
Alex: Now the evidence. What's the main comparison, and on what?
Sam: The teacher is Qwen3-8B, the students are Qwen3.5 at four and nine billion parameters, and the training set is DAPO, about seventeen thousand math problems. They compare against fine-tuning on the full set, across nine benchmarks in competition math, general math, and science. Full fine-tuning takes over eight hours on two GPUs. Theirs takes minutes at most, and scores match or beat it on most benchmarks.
Alex: A cheaper run that's competitive is good. But how do I know the uncertainty score matters, and not just training on less?
Sam: They test that directly. Same-size pools, picked randomly from correct solutions versus picked by uncertainty, with student routing switched off. Uncertainty wins on average by between about two and four and a half points. They also ran a matched-budget comparison against plain distillation, and gained about four points on average. And switching off either stage in the ablation hurts. Together they do best.
Alex: Does it hold beyond Qwen students learning from a Qwen teacher?
Sam: They tried Gemma, Llama, and Mistral students with the same teacher pool, and report maintained or improved performance on well under one percent of the budget. They tried two closed teachers, DeepSeek and Kimi. And a weak-to-strong case: a 1.7 billion parameter teacher improving a fourteen billion student. The four billion student gained up to fourteen points on one competition set.
Alex: The abstract says the nine billion student competes with much larger models. How seriously should I take that?
Sam: They put it against models from thirty billion up to six hundred and seventy-one billion, and it's competitive or ahead on the listed math and science benchmarks. I'd read that as strong on these tasks, not as a general ranking. The comparison is against published numbers on a handful of benchmarks.
Alex: So what's the catch you mentioned up front?
Sam: The headline budget counts student-side training only. The teacher-side profiling samples several continuations at every step of every solution. That's a lot of generation, and the excerpt doesn't report its cost. Since the pool is reusable across students, it may pay off over many runs, but I'd want to see that number. It also needs answers you can check automatically, which is why everything here is math and science. And the closed-teacher tests used only a thousand problems.
Alex: And the speed story, beyond wall-clock time?
Sam: They tracked training curves. Full fine-tuning needs about fifteen hundred steps to peak. Their method passes it within about a hundred and fifty. Their reading is that the gain comes from spending the budget on better examples.
Alex: So who should sit down with the full paper, and where should they start?
Sam: Anyone working on reasoning distillation or data selection for fine-tuning. Start with the method section on uncertainty profiling, then go straight to the ablations, the random-versus-uncertainty comparison and each stage switched off. Then look for what profiling costs on the teacher side.
Alex: And for everyone else, the one line to keep?
Sam: A correct, confident answer can still rest on shaky steps. Those shaky steps are where the training pays off.
Alex: Less data, aimed better. Thanks for walking through it.