Paper deep dive
Capacity-Dependent Effects of Data Selection for Reasoning
Cuong Dang, Hoang Anh Just, Ruoxi Jia
Intelligence
Status: succeeded | Model: Gemma-4-26B-A4B | Prompt: intel-v1 | Confidence: 91%
Last extracted: 8/17/2026, 4:11:18 AM
Summary
This paper investigates the impact of likelihood-based data selection on the reasoning capabilities of large language models (LLMs) during supervised fine-tuning. The authors challenge the assumption that high-likelihood responses are universally superior, demonstrating a capacity-dependent 'Fast-Fit / Slow-Gain' pattern. Small models (1.5B-4B parameters) benefit more from high-likelihood data, achieving faster early improvements, while larger models (7B-8B parameters) eventually outperform when trained on low-likelihood data over longer durations. The study highlights that effective data selection must account for model capacity and computing budget.
Entities (12)
Relation Signals (9)
Training Duration → influences → Effectiveness of Data Selection
confidence 95% · the value of likelihood-based data selection depends critically on model capacity and training duration.
Model Capacity → influences → Effectiveness of Data Selection
confidence 95% · the value of likelihood-based data selection depends critically on model capacity and training duration.
High-likelihood data → provides → Fast-Fit
confidence 95% · High-likelihood data provides faster and more stable early improvements, especially for smaller models
Low-likelihood data → provides → Slow-Gain
confidence 95% · low-likelihood data becomes increasingly beneficial for larger models when training is allowed to continue longer.
Small models → benefitsfrom → High-likelihood data
confidence 90% · High-likelihood data provides faster and more stable early improvements, especially for smaller models
Large Models → benefitsfrom → Low-likelihood data
confidence 90% · low-likelihood data becomes increasingly beneficial for larger models when training is allowed to continue longer.
GRAPE → uses → Likelihood-based Data Selection
confidence 90% · GRAPE (43) proposes a likelihood-based response selection strategy: for each instruction, it selects the response with the highest probability under the target model from a pool of candidate answers.
Cypher Suggestions (0)
No Cypher suggestions yet.
Abstract
Abstract:In reasoning supervised fine-tuning, candidate responses for the same instruction can differ substantially in how well they match the student's current distribution. Recent likelihood-based response selection methods suggest that responses closer to the student distribution provide more effective supervision, motivating the hypothesis that high-likelihood responses may generally be preferable for fine-tuning. In this paper, we revisit this intuition and show that the value of likelihood-based data selection depends critically on model capacity and training duration. Through controlled experiments on mathematical reasoning, using students ranging from 1.5B to 8B parameters and supervision generated by stronger teacher models, we observe a clear \emph{capacity-dependent} ``{\color{SMALLCOLOR}\textbf{Fast-Fit}} / {\color{LARGECOLOR}\textbf{Slow-Gain}}'' pattern. High-likelihood data provides faster and more stable early improvements, especially for smaller models, but low-likelihood data becomes increasingly beneficial for larger models when training is allowed to continue longer. To explain this phenomenon, we analyze learning dynamics, showing that small models often fail to absorb low-likelihood supervision and instead fall into shallow or repetitive behaviors, while larger models are better able to move toward the teacher distribution under such data. We further provide a capacity-constrained theoretical view of distillation that clarifies how data difficulty, data span, and student capacity jointly govern transfer. Overall, our findings show that effective data selection for reasoning should be aware of model capacity and computing budget rather than based on a single universal preference for high-likelihood supervision.
Tags
Links
- Source: https://arxiv.org/abs/2608.13721v1
- Canonical: https://arxiv.org/abs/2608.13721v1
Trouble viewing inline? Open PDF directly →
Full Text
115,595 characters extracted from source content.
Expand or collapse full text
Capacity-Dependent Effects of Data Selection for Reasoning Cuong Dang & Hoang Anh Just & Ruoxi Jia Affiliation: Department of Electrical and Computer Engineering Affiliation: Virginia Tech Affiliation: Blacksburg, VA, USA Email: cuongdc,just,ruoxijia@vt.edu Abstract In reasoning supervised fine-tuning, candidate responses for the same instruction can differ substantially in how well they match the student’s current distribution. Recent likelihood-based response selection methods suggest that responses closer to the student distribution provide more effective supervision, motivating the hypothesis that high-likelihood responses may generally be preferable for fine-tuning. In this paper, we revisit this intuition and show that the value of likelihood-based data selection depends critically on model capacity and training duration. Through controlled experiments on mathematical reasoning, using students ranging from 1.5B to 8B parameters and supervision generated by stronger teacher models, we observe a clear capacity-dependent “Fast-Fit / Slow-Gain” pattern. High-likelihood data provides faster and more stable early improvements, especially for smaller models, but low-likelihood data becomes increasingly beneficial for larger models when training is allowed to continue longer. To explain this phenomenon, we analyze learning dynamics, showing that small models often fail to absorb low-likelihood supervision and instead fall into shallow or repetitive behaviors, while larger models are better able to move toward the teacher distribution under such data. We further provide a capacity-constrained theoretical view of distillation that clarifies how data difficulty, data span, and student capacity jointly govern transfer. Overall, our findings show that effective data selection for reasoning should be aware of model capacity and computing budget rather than based on a single universal preference for high-likelihood supervision. Figure 1: Capacity-dependent learning dynamics under likelihood-based data selection. The illustration conceptualizes the optimization landscape faced by models with different capacities when learning reasoning tasks. A small model (left) can only climb to a nearby local peak, representing limited capability to learn from distant supervision signals. When trained on low-likelihood examples—responses far from its current policy—it struggles to escape local minima and therefore benefits from high-likelihood data, which allows it to exploit existing knowledge and achieve incremental improvements. In contrast, a large model (right) possesses sufficient representational capacity and optimization flexibility to traverse a longer path across the landscape. Although learning from low-likelihood data may require longer training, the model can ultimately reach a higher global peak, approaching the teacher distribution and acquiring stronger reasoning ability. 1 Introduction Large language models (LLMs) are trained on massive text corpora to predict the next token, enabling them to generate fluent language and perform a wide range of tasks such as reasoning, coding, and question answering. After this broad pretraining stage, a common next step is supervised fine-tuning (SFT), where the model is trained on curated instruction-response pairs to sharpen useful behaviors and elicit latent capabilities. In SFT for reasoning, supervision is highly heterogeneous: even for the same instruction, candidate responses can differ substantially in how well they match the student’s current distribution. Some responses are already close to what the model can produce, while others are much harder for it to reproduce. Choosing which response to train on for each instruction therefore directly affects optimization stability, sample efficiency, and the kind of reasoning behavior the student ultimately acquires. Recent work has begun to explore this problem of response selection explicitly. In particular, GRAPE (43) proposes a likelihood-based response selection strategy: for each instruction, it selects the response with the highest probability under the target model from a pool of candidate answers. We focus on GRAPE in particular because it is one of the earliest works to study response selection in instruction tuning, and because its simple likelihood-based rule, which incurs substantially lower overhead than alternatives (36; 40), was shown to be effective in the settings originally studied. The underlying intuition of GRAPE is straightforward: supervision should respect the learner’s current distribution, and examples that are already more likely under the student may be easier to absorb and thus more effective for fine-tuning. At the same time, a long and influential line of work across active learning and data selection advocates for the opposite philosophy—that models learn most from hard examples, i.e., those with low log-likelihood under the current model (16; 29; 28; 24; 19). From this perspective, easy, high-likelihood samples are precisely the ones to avoid, as they contribute little gradient signal and fail to push the model beyond its current capability boundary. Hence, this raises fundamental questions: ❶ Does high-likelihood data always work best? (Section 4), ❷ Why does it help in some settings but fail in others? (Section 5), ❸ What is the general mechanism behind likelihood-based data selection? (Section 6) In this work, we study these questions through the lens of model capacity and learning dynamics. Our central hypothesis is that the value of data selection depends critically on whether the student model has sufficient capacity to benefit from challenging supervision. Through controlled experiments across model scales, as shown in Figure 1, we observe a clear capacity-dependent pattern. For small models, selecting low-likelihood data is often ineffective: the model fails to move meaningfully toward the teacher distribution, exhibits unstable training dynamics, and tends to generate repetitive or shallow outputs rather than genuinely improved reasoning. In contrast, larger models can benefit substantially from low-likelihood data, especially when trained long enough, because such data pushes them beyond their current distribution and enables stronger adaptation toward the teacher’s reasoning behavior. We also find that training duration matters: high-likelihood data often provides rapid early gains, whereas low-likelihood data may yield slower but ultimately greater improvements for sufficiently capable models. These observations suggest that data selection for reasoning should not be treated as a one-size-fits-all rule. Instead, the effectiveness of high- or low-likelihood supervision is governed by an interaction between data distribution, model capacity, and computing budget. Overall, our contributions are threefold: ❶ We reveal a capacity-dependent effect of likelihood-based data selection for reasoning, showing that high-likelihood data is not universally optimal. ❷ We explain this phenomenon through optimization and learning dynamics. ❸ We develop a thereotical capacity-constrained view of distillation showing that knowledge transfer is jointly governed by initialization, knowledge gap, data space, and student capacity. 2 Related Works SFT, Distillation & Data Engineering for Reasoning. SFT has emerged as a central mechanism for eliciting reasoning in LLMs, with prior work showing that training on rationales can substantially improve multi-step reasoning. For instance, 42 demonstrate that reasoning can be bootstrapped by iteratively generating and fine-tuning on successful rationales. More recently, the success of DeepSeek (9) highlighted the effectiveness of distilling reasoning ability from powerful teacher models into smaller students. This distillation paradigm has since been adopted by a series of open-source projects, including OpenR1 (12), OpenThoughts (8), AceReason (20), and NVIDIA OpenMathReasoning (21). Yet, given a large pool of teacher-generated responses, an important question is which responses are most useful for student learning. On-Policy Distillation (1) argues that supervision closer to the student’s own generation distribution is more learnable, and GRAPE (43) applies this intuition to instruction tuning by selecting responses that best fit the target model. These works collectively suggest that the effectiveness of reasoning supervision depends not only on teacher quality, but also on its compatibility with the learner’s current distribution. In this paper, we revisit this hypothesis and show that while high-likelihood data often yields faster and more stable optimization, low-likelihood data can become increasingly beneficial over longer training horizons because it carries knowledge beyond the student’s current capabilities. Context-Dependent Data Selection. More broadly, our work connects to a line of research showing that there is no universally optimal rule for difficulty-based data selection, and that its effectiveness depends on the learning context. Curriculum and self-paced learning frameworks suggest that the usefulness of harder examples varies with the learner’s current stage, motivating adaptive or easy-to-hard training schedules rather than fixed policies (5; 14; 25; 15). This dependence also arises with respect to the data regime: 30 show that when data is scarce, retaining easy examples is preferable as they capture coarse-grained structure, whereas when data is abundant, harder examples become more valuable as easy ones grow increasingly redundant. More generally, recent studies suggest that no single data selection strategy consistently dominates across settings, with performance depending on factors such as training time budget, data composition, and representation (35; 6). Closest to our setting, prior work has also argued that the usefulness of difficult supervision depends on model capacity, but in different settings and with different notions of difficulty. 7 study this question in preference learning, where difficulty is defined over preference optimization examples rather than reasoning traces. 18 study reasoning distillation, but operationalize difficulty through proxies such as the length of reasoning traces and teacher capability, showing that small models often benefit more from shorter chain-of-thoughts or weaker teachers. In contrast, our work focuses on reasoning distillation under likelihood-based data selection, where difficulty is measured directly by the student’s likelihood of the teacher response. This provides a student-grounded notion of difficulty, rather than a student-independent proxy such as trace length. Beyond this difference in setup, we characterize when difficulty-based selection helps in this likelihood-based regime: the relative value of high- versus low-likelihood supervision depends jointly on model capacity and training duration. We further introduce a new axis, model capacity, into the theory of when difficulty-based selection helps, which prior work has mainly characterized along two axes: a data-size axis (30) and a learning-stage axis (34). 3 Background Following a common practice in modern LLM development, smaller language models are often trained on supervision generated by stronger teacher models. For instance, AceReason-Nemotron 1.1 (20) is trained on data generated by DeepSeek-R1 (9), while DASD-4B-Thinking (37) is supervised-fine-tuned from Qwen3-4B-Instruct-2507 (38) using teacher responses produced by gpt-oss-120b (23). Since the pool of teacher-generated responses can be very large, a central question is how to select the most useful supervision. GRAPE (43) addresses this by favoring responses that are more aligned with the student’s distribution. Let =xii=1NQ=\x_i\_i=1^N denote the instruction pool and let T denote the set of teacher models. For each teacher T∈T , let πθT(y∣x) _ _T(y x) denote the conditional distribution over response y given instruction x, where θT _T is the parameter of teacher T. For each instruction xi∈x_i , we collect one or more teacher-generated responses from every teacher T∈T . The resulting candidate response set is i=yi(T,j):T∈,j∈[JT(i)],[JT(i)]=1,…,JT(i),A_i= \y_i^(T,j)\;:\;T ,\;j∈[J_T(i)] \, [J_T(i)]=\1,…,J_T(i)\, (1) where JT(i)J_T(i) is the number of responses generated by the teacher T for instruction xix_i. The dataset D is obtained by selecting the answer yi⋆y_i for each question xix_i with the highest log-likelihood under the base student model distribution πθ0(y∣xi) _ _0(y x_i), where θ0 _0 is the parameters of the base student model, =(xi,yi⋆)i=1N,yi⋆=argmaxy∈iπθ0(y∣xi).D=\(x_i,y_i )\_i=1^N, y_i = _y _i _ _0(y x_i). (2) The student is then trained with standard supervised fine-tuning: minθ−∑(x,y)∈logπθ(y∣x). _θ\;- _(x,y) _θ(y x). (3) We examine this likelihood-based selection via two selection strategies: one formed from the highest-likelihood responses, highD_ high, and one from the lowest-likelihood responses, lowD_ low, high=(xi,yiℋ)i=1N, _ high=\(x_i,y_i^H)\_i=1^N, yiℋ=argmaxy∈Riπθ0(y∣xi), y_i^H= _y∈ R_i _ _0(y x_i), (4) low=(xi,yiℒ)i=1N, _ low=\(x_i,y_i^L)\_i=1^N, yiℒ=argminy∈Riπθ0(y∣xi). y_i^L= _y∈ R_i _ _0(y x_i). (5) 4 Main Empirical Results 4.1 Experimental Setup Dataset & Models. We conduct our experiments on the MATH12K dataset (11). We use two teacher models, Qwen/Qwen2.5-72B (39) and google/gemma-3-27b-it (31), to generate answers for each question in MATH12K. We mainly report results of these student models, including Qwen/Qwen2.5-1.5B, Qwen/Qwen2.5-3B-Instruct, Qwen/Qwen2.5-Math-7B (26), Qwen/Qwen3-4B, Qwen/Qwen3-8B (38). We show similar trends for other model families in Appendix C, and provide training hyperparameters in Appendix B.1. Evaluation Metrics. We evaluate correctness based on standard pass@1pass@1 (43). We set the temperature to 0.6, which is widely used in related works (41; 32). We evaluate on 10 popular math datasets: AIME24 (2), AMC (3), CHMATH (33), Gaokao (44), GPQA (27), GradeSchool (41), KAOYAN, MATH500 (11), Minerva (17), Olympiad Bench (10). 4.2 Results Table 1 summarizes the best performance achieved within the first epoch for models of different sizes trained on high-likelihood and low-likelihood data. To capture early reasoning performance dynamics, we save four checkpoints during the first epoch and evaluate each of them; the reported result is the best score across these checkpoints. We observe a clear early advantage for high-likelihood selection: it wins on almost all datasets across model sizes, especially for the 1.5B and 3B students, on 9/10 for the 4B student, on 8/10 datasets for the 7B student with two ties, and on 9/10 datasets for the 8B student. This pattern suggests that high-likelihood data provides a stronger optimization signal in the early stage of training, especially for smaller-capacity students. Table 1: Best performance across checkpoints after ONE epoch for models of different sizes trained on high-likelihood and low-likelihood data. Highlighted cells mark the better performance between training on high and low-likelihood data. 1.5B 3B 4B 7B 8B Dataset High Low High Low High Low High Low High Low AIME 3.33 0 10.00 3.33 20.00 10.00 13.33 13.33 36.67 26.67 AMC 27.50 17.50 47.50 32.50 62.00 50.00 62.50 47.50 87.50 80.00 CHMATH 6.67 3.33 10.00 3.33 26.67 13.33 30.00 20.00 56.67 50.00 GAOKAO 10.13 6.33 25.32 10.13 54.43 53.16 55.70 55.70 74.68 70.89 GPQA 28.79 16.67 34.85 18.18 36.36 26.26 39.39 27.78 78.79 75.25 GRADE SCHOOL 22.38 13.33 19.52 14.76 48.57 40.95 48.10 44.29 68.10 65.24 KAOYAN 19.10 8.04 24.12 7.04 40.20 25.13 46.23 31.16 71.86 65.33 MATH 54.60 41.80 66.40 52.20 75.20 66.00 78.80 71.80 89.20 91.60 MINERVA 24.06 13.24 32.35 25.37 40.44 33.82 43.01 33.82 58.46 57.72 OLYMPIAD BENCH 21.78 13.04 26.52 20.59 37.48 31.70 42.37 37.93 64.30 64.00 Table 2: Best performance across checkpoints after FIVE epochs for models of different sizes trained on high-likelihood and low-likelihood data. Highlighted cells mark the better performance between training on high and low-likelihood data. 1.5B 3B 4B 7B 8B Dataset High Low High Low High Low High Low High Low AIME 6.67 3.33 10.00 6.67 20.00 20.00 16.67 26.67 40.00 43.33 AMC 35.00 22.50 55.00 47.50 70.00 57.50 67.50 62.50 90.00 92.5 CHMATH 6.67 6.67 16.67 10.00 33.33 36.67 33.33 33.33 56.67 63.33 GAOKAO 11.39 8.86 25.32 24.05 56.96 58.23 55.70 62.03 74.68 67.09 GPQA 29.29 16.67 34.85 27.27 39.39 38.38 39.39 44.95 78.79 79.8 GRADE SCHOOL 22.86 13.33 22.38 19.52 53.33 47.14 50.95 55.71 70.48 72.38 KAOYAN 19.60 10.55 26.63 11.56 44.22 33.67 46.73 37.69 69.85 71.86 MATH 57.60 44.40 66.80 58.60 78.8 74.4 79.00 80.06 92.00 93.00 MINERVA 22.79 14.71 32.72 25.74 40.81 45.96 43.38 37.87 58.46 62.50 OLYMPIAD BENCH 22.52 14.96 30.52 24.30 41.63 41.19 42.81 44.3 65.19 65.48 Observation 1 (Early-stage advantage of high-likelihood data). High-likelihood data consistently yield faster early gains, especially for smaller models. Table 2 reports the best performance over the full 5-epoch training process. During training, we save four checkpoints per epoch and evaluate all of them, so each entry corresponds to the best result among 20 checkpoints. Unlike the first-epoch results, where high-likelihood data consistently provides a stronger early learning signal, the longer training horizon reveals a clear capacity-dependent reversal. For the smaller 1.5B, 3B, and 4B students, high-likelihood training remains dominant. However, for the larger 7B and 8B students, low-likelihood training surpasses high-likelihood training on most tasks. Observation 2 (Low-likelihood data helps large models eventually). For high-capacity models, high-likelihood peaks early, while low-likelihood improves later and can surpass it. Table 3: An illustrative example showing that larger models are less prone to getting lost under low-likelihood training. The 1.5B model repeats the question and then gets stuck repeating an incorrect final answer, the 3B model repeats intermediate reasoning, and the 7B model successfully completes the reasoning and returns the correct answer. Question: There is a sequence a1,a2,…,a2023a_1,a_2,…,a_2023 such that tan(2023x)=⋯ (2023x)=·s. What is a2023a_2023? Qwen2.5-1.5B Qwen2.5-3B-Instruct Qwen2.5-Math-7B There is a sequence a1,a2,…,a2023a_1,a_2,…,a_2023 such that tan(2023x)=⋯ (2023x)=·s. What is a2023a_2023? 2023202320232023⋯ [rgb]1,0,0 2023\ 2023\ 2023\ 2023\ ·s Let P(x)=tan2023xP(x)= 2023x. Then tan(2023x)=tan(2022x)+tanx1−tan(2022x)tanx (2023x)= (2022x)+ x1- (2022x) x tan(2023x)=tan(2022x)+tanx1−tan(2022x)tanx [rgb]1,0,0 (2023x)= (2022x)+ x1- (2022x) x ⋯·s Let t=tanxt= x. Then we have… We know that tan(A+B)=tanA+tanB1−tanAtanB (A+B)= A+ B1- A B ⋯·s Final answer: 1 1. We present an illustrative example to highlight how model capacity affects learning from low-likelihood data in Table 3, showing that the 1.5B model mostly repeats the question and then produces an incorrect final answer, indicating shallow imitation rather than acquisition of the teacher’s reasoning process. The 3B model can generate some preliminary reasoning steps, but then gets lost in repetition and still fails to complete the solution. By contrast, the 7B model successfully carries out the reasoning and reaches the correct answer. Observation 3 (Small model learns by rote). Small model: repetition, shallow copying, incomplete reasoning. Larger model: delayed but meaningful reasoning improvement. In brief, these results reveal a clear “Fast-Fit / Slow-Gain” pattern. In the Fast-Fit regime: the model quickly aligns to clean, learnable supervision and improves with relatively little training. The effect is especially strong for smaller students, whose limited capacity makes them more dependent on examples that match their current representation and optimization ability. By contrast, in the Slow-Gain regime, low-likelihood data often appears less useful at the beginning, but becomes increasingly valuable when the student has more capacity and the training budget allows longer optimization. This suggests that the best data-selection strategy is not universal: it should be chosen jointly according to how powerful the student is and how much training compute is available. Takeaway 1 (Computing budget and model capacity matter data selection). High-likelihood data is best for fast, compute-efficient early learning, whereas low-likelihood data becomes more valuable when model capacity and training budget are large enough. 5 Dual-view Learning Dynamics Analysis In this section, we further examine why the Slow-Gain regime appears for larger models, yet disappears for small-capacity students from teacher and original model views. Figure 2: Loss dynamics under the teacher model. We measure the Gemma log-likelihood of answers on the test set (AMC) and training set (MATH12K) across checkpoints for the 1.5B, 3B, and 7B student models trained on low- and high-likelihood data. Figure 3: Loss dynamics under the base model. Log-likelihood on AMC (test) and MATH12K (train) across checkpoints for 1.5B, 3B, and 7B students, evaluated under their own base models after training on low- and high-likelihood data. Learning Dynamics under Teacher View. Figure 2 reveals a clear capacity-dependent teacher-loss dynamics pattern. For the 1.5B student, training on high-likelihood data causes the teacher-model loss of the student’s generated answers to increase, indicating that the student is drifting farther away from the teacher distribution. When trained on low-likelihood data, the teacher loss remains roughly unchanged. Taken together, these results suggest that the 1.5B model lacks sufficient capacity to move meaningfully closer to the teacher, regardless of whether the selected data are high- or low-likelihood. For the 3B student, the pattern becomes more differentiated: under high-likelihood training, the teacher loss stays largely stable, whereas under low-likelihood training it decreases, showing that low-likelihood data helps this model move closer to the teacher distribution. For the 7B student, the teacher loss decreases under both high- and low-likelihood training, indicating that, once model capacity is sufficiently large, the student can absorb teacher knowledge effectively regardless of the likelihood-based data selection. Overall, these results suggest that the ability of likelihood-based data selection to pull the student toward the teacher depends strongly on model capacity: smaller students struggle to approach the teacher at all, medium-sized students benefit particularly from low-likelihood data, and larger students can align with the teacher under either selection strategy. Observation 4 (Capacity-Dependent Teacher Alignment). Low-likelihood data moves larger models closer to the teacher but fails to do so for small models. Learning Dynamics under Original Model View. Figure 3 presents the loss of generated answers measured under the base model, which helps reveal how far the fine-tuned student departs from its original distribution. For the 1.5B model, training on low-likelihood data leads to a broader and more divergent loss pattern under the base model, while still overlapping substantially with the initial loss range. This suggests that although fine-tuning pushes many generated answers away from the base distribution, a considerable portion of them remain close to the model’s original behavior. For the 3B model, answers produced after training on low-likelihood data exhibit noticeably higher loss under the base model, indicating a stronger shift away from the original distribution than in the 1.5B case. Moreover, the separation between the answers generated by models trained on high-likelihood versus low-likelihood data becomes larger, showing that likelihood-based data selection has a clearer and more distinct effect at this scale. For the 7B model, the loss variance of answers generated after low-likelihood training becomes smaller than for the 3B model, suggesting that the larger model adapts in a more stable and focused way rather than drifting broadly. On the test set, the difference between the 1.5B models trained on high- and low-likelihood data is relatively small, but this gap widens as model size increases. This indicates that larger models not only generalize better, but also express the effect of data selection more clearly in their output distribution. Observation 5 (Capacity-Dependent Distribution Shift and Generalization). As model size increases, fine-tuned behavior becomes more stable, generalization improves, and the distinction between high-likelihood and low-likelihood training becomes more pronounced. 6 Theoretical Mechanism of Capacity-Constrained Distillation Figure 4: Geometric intuition for distillation with a low-capacity student. The data space acts as a mirror that reveals teacher knowledge to the student, but limited capacity restricts learning to the portion that lies in the student’s feasible space, thereby shrinking the effective supervision. As a result, high-capacity students ∙ can absorb more of the exposed knowledge and move toward ∙ , whereas low-capacity students move only toward ∙ . In this section, we provide a theoretical account of the capacity-dependent effects observed in our experiments. Our goal is to clarify why the same data selection can produce markedly different outcomes for students of different capacities, and in particular, why low-likelihood data may benefit larger models while failing to help smaller ones. Setting. We study knowledge distillation for binary classification, where a student model learns from labels generated by a teacher model. Let the teacher and student be linear classifiers h∗,h:→h ,h:X , defined by h∗()=∗⊤≥0,h()=⊤≥0,h (x)=1\w_ x≥ 0\, h(x)=1\w x≥ 0\, (6) where ∗,∈ℝd∖0w_ ,w ^d \0\ are the teacher and student parameter vectors, respectively, and ⋅1\·\ denotes the indicator function. The weight vectors can also be parameterized as the product of matrices, ⊤=N−1⋯1w =W_NW_N-1·sW_1 for some N≥1N≥ 1. When n≥2n≥ 2, this parameterization is known as deep linear network, whose analysis is deferred in Appendix A.5. For a low-capacity student, we assume that the feasible student parameter space is a subspace ℱ⊆ℝdF ^d, and let Pℱ:ℝd→ℱP_F:R^d denote the orthogonal projection onto ℱF. The student is trained on a dataset =(i,yi)i=1nD=\(x_i,y_i)\_i=1^n, where each soft label yiy_i is provided by the teacher, yi=σ(∗⊤i),y_i=σ(w_ x_i), (7) with σ(⋅)σ(·) denoting the sigmoid function. Let (τ)w(τ) denote the student parameter at training time τ. The student is updated by gradient flow on the normalized cross-entropy loss L1L^1. L1()=−1n∑i=1n[yilogσ(⊤i)+(1−yi)log(1−σ(⊤i))]−L∗,L^1(w)=- 1n _i=1^n [y_i σ (w x_i )+ (1-y_i ) (1-σ (w x_i ) ) ]-L , (8) where L∗L is a normalization constant. Detailed setting description is in Appendix A.1. Theorem 6.1 (High-capacity Student Solution). Let P_X be the orthogonal projector onto span1,…,nspan\x_1,…,x_n\. Then, as τ→∞τ→∞, the student parameter, (τ)w(τ), converges to w, where ^=0+P(∗−0). w= [rgb]0.7891,0.6289,0.8633w_0+ [rgb]1,0.4531,0.4219P_X( [rgb]0.4336,0.7617,0.4609w_ -w_0). (9) Proof. The proof is deferred in Appendix A.4. ∎ Interpretation. Theorem 6.1 shows that training can only transfer teacher knowledge that is exposed by the data. The vector ∗−0w_ -w_0 represents the knowledge gap between teacher and student. The projection term P(∗−0) [rgb]1,0.4531,0.4219P_X( [rgb]0.4336,0.7617,0.4609w_ -w_0) indicates that the knowledge gap between the teacher and the student is transferred only through its component in the data span. Thus, the data act as a mirror of the teacher: if the data span aligns well with the knowledge gap, the student learns effectively; if it is orthogonal to that gap, then no transfer occurs. Moreover, directions orthogonal to the data span are never updated, so the final student remains partially constrained by its initialization 0w_0. Theorem 6.2 (Low-capacity student solution). Let ℱX_F be the projection of the data span onto ℱF, and let PℱP_X_F denote the orthogonal projection onto ℱX_F. Then, as τ→∞τ→∞, the student parameter converges to w, where ^=0+Pℱ(∗−0). w=w_0+ [rgb]1,0.5898,0.3086P_X_F(w_ -w_0). (10) Proof. The proof is deferred in Appendix A.4. ∎ Interpretation. The projection term Pℱ [rgb]1,0.5898,0.3086P_X_F in Theorem 6.2 shows that, for a low-capacity student, only the portion of the data span that lies in the feasible student space ℱF is useful for learning, as shown in Figure 4. Hence, distillation is constrained not only by what the data reveal about the teacher, but also by what the student can represent. Takeaway 2 (Capacity-aligned data is necessary for low-capacity distillation). For low-capacity students, successful distillation requires not only data that exposes teacher knowledge, but data whose exposed knowledge lies within the student’s representable subspace. Connection to Failure of Low-likelihood Data for Small LLMs. Low-likelihood data can be useful because it exposes teacher knowledge beyond the student’s current distribution. However, our theory shows that a low-capacity student can learn only the portion of this knowledge that lies in its feasible subspace. When the likelihood is too low, the exposed teacher knowledge may exceed what the student can represent or optimize, making the supervision ineffective. 7 Conclusion This work challenges the prevailing assumption that high-likelihood data is universally optimal for fine-tuning LLMs on reasoning tasks. Through controlled experiments across model scales (1.5B–8B parameters) and analysis of learning dynamics, we demonstrate a clear capacity-dependent “Fast-Fit / Slow-Gain” pattern: high-likelihood data yields faster and more stable early improvements, particularly for smaller models, while low-likelihood data becomes increasingly beneficial for larger models given sufficient training budget. Acknowledgements Ruoxi Jia and the ReDS lab acknowledge support from the National Science Foundation through grants IIS-2312794, IIS-2313130, OAC-2239622, CNS-2424127, OAC-2613761, and the Amazon-Virginia Tech Initiative for Efficient and Robust Machine Learning. We also acknowledge Advanced Research Computing at Virginia Tech for providing computational resources and technical support that have contributed to the results reported within this paper. References Agarwal et al. (2024) R. Agarwal, N. Vieillard, Y. Zhou, P. Stanczyk, S. R. Garea, M. Geist, and O. Bachem On-policy distillation of language models: learning from self-generated mistakes. In The Twelfth International Conference on Learning Representations, External Links: Link Cited by: §2. AIMO (2024a) AIMO AIME-2024. Note: https://huggingface.co/datasets/aimo/aimo-validation-aime External Links: Link Cited by: §4.1. AIMO (2024b) AIMO AMC. Note: https://huggingface.co/datasets/AI-MO/aimo-validation-amc External Links: Link Cited by: §4.1. Arora et al. (2018) S. Arora, N. Cohen, and E. Hazan On the optimization of deep networks: implicit acceleration by overparameterization. In International conference on machine learning, p. 244–253. Cited by: §A.5, Lemma A.11. Bengio et al. (2009) Y. Bengio, J. Louradour, R. Collobert, and J. Weston Curriculum learning. In Proceedings of the 26th annual international conference on machine learning, p. 41–48. Cited by: §2. Du et al. (2025) Y. Du, Y. Song, H. M. Wong, D. Ignatev, A. Gatt, and D. Nguyen Disentangling the roles of representation and selection in data pruning. In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), p. 16791–16809. Cited by: §2. Gao et al. (2025) C. Gao, H. Li, L. Liu, Z. Xie, P. Zhao, and Z. Xu Principled data selection for alignment: the hidden risks of difficult examples. arXiv preprint arXiv:2502.09650. Cited by: §2. Guha et al. (2026) E. K. Guha, R. Marten, S. Keh, N. Raoof, G. Smyrnis, H. Bansal, M. Nezhurina, J. Mercat, T. Vu, Z. R. Sprague, A. Suvarna, B. Feuer, L. L. Chen, Z. Khan, E. Frankel, S. Grover, C. Choi, N. Muennighoff, S. Su, W. Zhao, J. Yang, S. Pimpalgaonkar, K. sharma, C. C. Ji, Y. Deng, S. M. Pratt, V. Ramanujan, J. Saad-Falcon, S. Acharya, J. Li, A. Dave, A. Albalak, K. Arora, B. Wulfe, C. Hegde, G. Durrett, S. Oh, M. Bansal, S. Gabriel, A. Grover, K. Chang, V. Shankar, A. Gokaslan, M. A. Merrill, T. Hashimoto, Y. Choi, J. Jitsev, R. Heckel, M. Sathiamoorthy, A. Dimakis, and L. Schmidt OpenThoughts: data recipes for reasoning models. In The Fourteenth International Conference on Learning Representations, External Links: Link Cited by: §2. Guo et al. (2025) D. Guo, D. Yang, H. Zhang, J. Song, P. Wang, Q. Zhu, R. Xu, R. Zhang, S. Ma, X. Bi, et al. Deepseek-r1: incentivizing reasoning capability in llms via reinforcement learning. arXiv preprint arXiv:2501.12948. Cited by: §2, §3. He et al. (2024) C. He, R. Luo, Y. Bai, S. Hu, Z. Thai, J. Shen, J. Hu, X. Han, Y. Huang, Y. Zhang, et al. Olympiadbench: a challenging benchmark for promoting agi with olympiad-level bilingual multimodal scientific problems. In Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), p. 3828–3850. Cited by: §4.1. Hendrycks et al. (2021) D. Hendrycks, C. Burns, S. Kadavath, A. Arora, S. Basart, E. Tang, D. Song, and J. Steinhardt Measuring mathematical problem solving with the MATH dataset. In Thirty-fifth Conference on Neural Information Processing Systems Datasets and Benchmarks Track (Round 2), External Links: Link Cited by: §4.1, §4.1. Hugging Face (2025) Hugging Face Open r1: a fully open reproduction of deepseek-r1. External Links: Link Cited by: §2. Jiao et al. (2020) X. Jiao, Y. Yin, L. Shang, X. Jiang, X. Chen, L. Li, F. Wang, and Q. Liu Tinybert: distilling bert for natural language understanding. In Findings of the association for computational linguistics: EMNLP 2020, p. 4163–4174. Cited by: §A.1. Kumar et al. (2010) M. Kumar, B. Packer, and D. Koller Self-paced learning for latent variable models. Advances in neural information processing systems 23. Cited by: §2. Lalor and Yu (2020) J. P. Lalor and H. Yu Dynamic data selection for curriculum learning via ability estimation. In Findings of the Association for Computational Linguistics: EMNLP 2020, p. 545–555. Cited by: §2. Lewis (1995) D. D. Lewis A sequential algorithm for training text classifiers: corrigendum and additional data. In Acm sigir forum, Vol. 29, p. 13–19. Cited by: §1. Lewkowycz et al. (2022) A. Lewkowycz, A. Andreassen, D. Dohan, E. Dyer, H. Michalewski, V. Ramasesh, A. Slone, C. Anil, I. Schlag, T. Gutman-Solo, Y. Wu, B. Neyshabur, G. Gur-Ari, and V. Misra Solving quantitative reasoning problems with language models. In Advances in Neural Information Processing Systems, S. Koyejo, S. Mohamed, A. Agarwal, D. Belgrave, K. Cho, and A. Oh (Eds.), Vol. 35, p. 3843–3857. External Links: Link Cited by: §4.1. Li et al. (2025) Y. Li, X. Yue, Z. Xu, F. Jiang, L. Niu, B. Y. Lin, B. Ramasubramanian, and R. Poovendran Small models struggle to learn from strong reasoners. In Findings of the Association for Computational Linguistics: ACL 2025, p. 25366–25394. Cited by: §2. Lin et al. (2024) Z. Lin, Z. Gou, Y. Gong, X. Liu, Y. Shen, R. Xu, C. Lin, Y. Yang, J. Jiao, N. Duan, et al. Not all tokens are what you need for pretraining. Advances in Neural Information Processing Systems 37, p. 29029–29063. Cited by: §1. Liu et al. (2026) Z. Liu, Z. Yang, Y. Chen, C. Lee, M. Shoeybi, B. Catanzaro, and W. Ping AceReason-nemotron 1.1: advancing math and code reasoning through SFT and RL synergy. In The Fourteenth International Conference on Learning Representations, External Links: Link Cited by: §2, §3. Moshkov et al. (2025) I. Moshkov, D. Hanley, I. Sorokin, S. Toshniwal, C. Henkel, B. Schifferer, W. Du, and I. Gitman AIMO-2 winning solution: building state-of-the-art mathematical reasoning models with openmathreasoning dataset. arXiv preprint arXiv:2504.16891. Cited by: §2. Muralidharan et al. (2024) S. Muralidharan, S. Turuvekere Sreenivas, R. Joshi, M. Chochowski, M. Patwary, M. Shoeybi, B. Catanzaro, J. Kautz, and P. Molchanov Compact language models via pruning and knowledge distillation. Advances in Neural Information Processing Systems 37, p. 41076–41102. Cited by: §A.1. OpenAI et al. (2025) OpenAI, :, S. Agarwal, L. Ahmad, J. Ai, S. Altman, A. Applebaum, E. Arbus, R. K. Arora, Y. Bai, B. Baker, H. Bao, B. Barak, A. Bennett, T. Bertao, N. Brett, E. Brevdo, G. Brockman, S. Bubeck, C. Chang, K. Chen, M. Chen, E. Cheung, A. Clark, D. Cook, M. Dukhan, C. Dvorak, K. Fives, V. Fomenko, T. Garipov, K. Georgiev, M. Glaese, T. Gogineni, A. Goucher, L. Gross, K. G. Guzman, J. Hallman, J. Hehir, J. Heidecke, A. Helyar, H. Hu, R. Huet, J. Huh, S. Jain, Z. Johnson, C. Koch, I. Kofman, D. Kundel, J. Kwon, V. Kyrylov, E. Y. Le, G. Leclerc, J. P. Lennon, S. Lessans, M. Lezcano-Casado, Y. Li, Z. Li, J. Lin, J. Liss, Lily, Liu, J. Liu, K. Lu, C. Lu, Z. Martinovic, L. McCallum, J. McGrath, S. McKinney, A. McLaughlin, S. Mei, S. Mostovoy, T. Mu, G. Myles, A. Neitz, A. Nichol, J. Pachocki, A. Paino, D. Palmie, A. Pantuliano, G. Parascandolo, J. Park, L. Pathak, C. Paz, L. Peran, D. Pimenov, M. Pokrass, E. Proehl, H. Qiu, G. Raila, F. Raso, H. Ren, K. Richardson, D. Robinson, B. Rotsted, H. Salman, S. Sanjeev, M. Schwarzer, D. Sculley, H. Sikchi, K. Simon, K. Singhal, Y. Song, D. Stuckey, Z. Sun, P. Tillet, S. Toizer, F. Tsimpourlas, N. Vyas, E. Wallace, X. Wang, M. Wang, O. Watkins, K. Weil, A. Wendling, K. Whinnery, C. Whitney, H. Wong, L. Yang, Y. Yang, M. Yasunaga, K. Ying, W. Zaremba, W. Zhan, C. Zhang, B. Zhang, E. Zhang, and S. Zhao Gpt-oss-120b & gpt-oss-20b model card. External Links: 2508.10925, Link Cited by: §3. Paul et al. (2021) M. Paul, S. Ganguli, and G. K. Dziugaite Deep learning on a data diet: finding important examples early in training. Advances in neural information processing systems 34, p. 20596–20607. Cited by: §1. Platanios et al. (2019) E. A. Platanios, O. Stretcu, G. Neubig, B. Poczos, and T. Mitchell Competence-based curriculum learning for neural machine translation. In Proceedings of the 2019 conference of the North American chapter of the association for computational linguistics: human language technologies, volume 1 (long and short papers), p. 1162–1172. Cited by: §2. Qwen et al. (2025) Qwen, :, A. Yang, B. Yang, B. Zhang, B. Hui, B. Zheng, B. Yu, C. Li, D. Liu, F. Huang, H. Wei, H. Lin, J. Yang, J. Tu, J. Zhang, J. Yang, J. Yang, J. Zhou, J. Lin, K. Dang, K. Lu, K. Bao, K. Yang, L. Yu, M. Li, M. Xue, P. Zhang, Q. Zhu, R. Men, R. Lin, T. Li, T. Tang, T. Xia, X. Ren, X. Ren, Y. Fan, Y. Su, Y. Zhang, Y. Wan, Y. Liu, Z. Cui, Z. Zhang, and Z. Qiu Qwen2.5 technical report. External Links: 2412.15115, Link Cited by: §4.1. Rein et al. (2024) D. Rein, B. L. Hou, A. C. Stickland, J. Petty, R. Y. Pang, J. Dirani, J. Michael, and S. R. Bowman Gpqa: a graduate-level google-proof q&a benchmark. In First conference on language modeling, Cited by: §4.1. Robinson et al. (2021) J. D. Robinson, C. Chuang, S. Sra, and S. Jegelka Contrastive learning with hard negative samples. In International Conference on Learning Representations, Cited by: §1. Shrivastava et al. (2016) A. Shrivastava, A. Gupta, and R. Girshick Training region-based object detectors with online hard example mining. In Proceedings of the IEEE conference on computer vision and pattern recognition, p. 761–769. Cited by: §1. Sorscher et al. (2022) B. Sorscher, R. Geirhos, S. Shekhar, S. Ganguli, and A. Morcos Beyond neural scaling laws: beating power law scaling via data pruning. Advances in Neural Information Processing Systems 35, p. 19523–19536. Cited by: §2, §2. Team et al. (2025) G. Team, A. Kamath, J. Ferret, S. Pathak, N. Vieillard, R. Merhej, S. Perrin, T. Matejovicova, A. Ramé, M. Rivière, L. Rouillard, T. Mesnard, G. Cideron, J. Grill, S. Ramos, E. Yvinec, M. Casbon, E. Pot, I. Penchev, G. Liu, F. Visin, K. Kenealy, L. Beyer, X. Zhai, A. Tsitsulin, R. Busa-Fekete, A. Feng, N. Sachdeva, B. Coleman, Y. Gao, B. Mustafa, I. Barr, E. Parisotto, D. Tian, M. Eyal, C. Cherry, J. Peter, D. Sinopalnikov, S. Bhupatiraju, R. Agarwal, M. Kazemi, D. Malkin, R. Kumar, D. Vilar, I. Brusilovsky, J. Luo, A. Steiner, A. Friesen, A. Sharma, A. Sharma, A. M. Gilady, A. Goedeckemeyer, A. Saade, A. Feng, A. Kolesnikov, A. Bendebury, A. Abdagic, A. Vadi, A. György, A. S. Pinto, A. Das, A. Bapna, A. Miech, A. Yang, A. Paterson, A. Shenoy, A. Chakrabarti, B. Piot, B. Wu, B. Shahriari, B. Petrini, C. Chen, C. L. Lan, C. A. Choquette-Choo, C. Carey, C. Brick, D. Deutsch, D. Eisenbud, D. Cattle, D. Cheng, D. Paparas, D. S. Sreepathihalli, D. Reid, D. Tran, D. Zelle, E. Noland, E. Huizenga, E. Kharitonov, F. Liu, G. Amirkhanyan, G. Cameron, H. Hashemi, H. Klimczak-Plucińska, H. Singh, H. Mehta, H. T. Lehri, H. Hazimeh, I. Ballantyne, I. Szpektor, I. Nardini, J. Pouget-Abadie, J. Chan, J. Stanton, J. Wieting, J. Lai, J. Orbay, J. Fernandez, J. Newlan, J. Ji, J. Singh, K. Black, K. Yu, K. Hui, K. Vodrahalli, K. Greff, L. Qiu, M. Valentine, M. Coelho, M. Ritter, M. Hoffman, M. Watson, M. Chaturvedi, M. Moynihan, M. Ma, N. Babar, N. Noy, N. Byrd, N. Roy, N. Momchev, N. Chauhan, N. Sachdeva, O. Bunyan, P. Botarda, P. Caron, P. K. Rubenstein, P. Culliton, P. Schmid, P. G. Sessa, P. Xu, P. Stanczyk, P. Tafti, R. Shivanna, R. Wu, R. Pan, R. Rokni, R. Willoughby, R. Vallu, R. Mullins, S. Jerome, S. Smoot, S. Girgin, S. Iqbal, S. Reddy, S. Sheth, S. Põder, S. Bhatnagar, S. R. Panyam, S. Eiger, S. Zhang, T. Liu, T. Yacovone, T. Liechty, U. Kalra, U. Evci, V. Misra, V. Roseberry, V. Feinberg, V. Kolesnikov, W. Han, W. Kwon, X. Chen, Y. Chow, Y. Zhu, Z. Wei, Z. Egyed, V. Cotruta, M. Giang, P. Kirk, A. Rao, K. Black, N. Babar, J. Lo, E. Moreira, L. G. Martins, O. Sanseviero, L. Gonzalez, Z. Gleicher, T. Warkentin, V. Mirrokni, E. Senter, E. Collins, J. Barral, Z. Ghahramani, R. Hadsell, Y. Matias, D. Sculley, S. Petrov, N. Fiedel, N. Shazeer, O. Vinyals, J. Dean, D. Hassabis, K. Kavukcuoglu, C. Farabet, E. Buchatskaya, J. Alayrac, R. Anil, Dmitry, Lepikhin, S. Borgeaud, O. Bachem, A. Joulin, A. Andreev, C. Hardin, R. Dadashi, and L. Hussenot Gemma 3 technical report. External Links: 2503.19786, Link Cited by: §4.1. Wang et al. (2025) Y. Wang, Q. Yang, Z. Zeng, L. Ren, L. Liu, B. Peng, H. Cheng, X. He, K. Wang, J. Gao, W. Chen, S. Wang, S. S. Du, and yelong shen Reinforcement learning for reasoning in large language models with one training example. In The Thirty-ninth Annual Conference on Neural Information Processing Systems, External Links: Link Cited by: §4.1. Wei et al. (2023) T. Wei, J. Luan, W. Liu, S. Dong, and B. Wang CMATH: can your language model pass chinese elementary school math test?. External Links: 2306.16636, Link Cited by: §4.1. Weinshall et al. (2018) D. Weinshall, G. Cohen, and D. Amir Curriculum learning by transfer learning: theory and experiments with deep networks. In International conference on machine learning, p. 5238–5246. Cited by: §2. Wu et al. (2020) X. Wu, E. Dyer, and B. Neyshabur When do curricula work?. arXiv preprint arXiv:2012.03107. Cited by: §2. Xia et al. (2024) M. Xia, S. Malladi, S. Gururangan, S. Arora, and D. Chen LESS: selecting influential data for targeted instruction tuning. In Proceedings of the 41st International Conference on Machine Learning, p. 54104–54132. Cited by: §1. Yan et al. (2026) S. Yan, K. Liu, C. Shen, B. Wang, S. Fan, J. Zhang, Y. Wu, Z. Wang, and J. Ye Distribution-aligned sequence distillation for superior long-cot reasoning. arXiv preprint arXiv:2601.09088. Cited by: §A.1, §3. Yang et al. (2025) A. Yang, A. Li, B. Yang, B. Zhang, B. Hui, B. Zheng, B. Yu, C. Gao, C. Huang, C. Lv, C. Zheng, D. Liu, F. Zhou, F. Huang, F. Hu, H. Ge, H. Wei, H. Lin, J. Tang, J. Yang, J. Tu, J. Zhang, J. Yang, J. Yang, J. Zhou, J. Zhou, J. Lin, K. Dang, K. Bao, K. Yang, L. Yu, L. Deng, M. Li, M. Xue, M. Li, P. Zhang, P. Wang, Q. Zhu, R. Men, R. Gao, S. Liu, S. Luo, T. Li, T. Tang, W. Yin, X. Ren, X. Wang, X. Zhang, X. Ren, Y. Fan, Y. Su, Y. Zhang, Y. Zhang, Y. Wan, Y. Liu, Z. Wang, Z. Cui, Z. Zhang, Z. Zhou, and Z. Qiu Qwen3 technical report. External Links: 2505.09388, Link Cited by: §3, §4.1. Yang et al. (2024a) A. Yang, B. Zhang, B. Hui, B. Gao, B. Yu, C. Li, D. Liu, J. Tu, J. Zhou, J. Lin, et al. Qwen2. 5-math technical report: toward mathematical expert model via self-improvement. arXiv preprint arXiv:2409.12122. Cited by: §4.1. Yang et al. (2024b) Y. Yang, S. Mishra, J. Chiang, and B. Mirzasoleiman Smalltolarge (s2l): scalable data selection for fine-tuning large language models by summarizing training trajectories of small models. Advances in Neural Information Processing Systems 37, p. 83465–83496. Cited by: §1. Ye et al. (2025) Y. Ye, Z. Huang, Y. Xiao, E. Chern, S. Xia, and P. Liu LIMO: less is more for reasoning. In Second Conference on Language Modeling, External Links: Link Cited by: §4.1. Zelikman et al. (2022) E. Zelikman, Y. Wu, J. Mu, and N. Goodman STar: bootstrapping reasoning with reasoning. In Advances in Neural Information Processing Systems, A. H. Oh, A. Agarwal, D. Belgrave, and K. Cho (Eds.), External Links: Link Cited by: §2. Zhang et al. (2025) D. Zhang, Q. Dai, and H. Peng The best instruction-tuning data are those that fit. In The Thirty-ninth Annual Conference on Neural Information Processing Systems, External Links: Link Cited by: §1, §2, §3, §4.1. Zhang et al. (2023) X. Zhang, C. Li, Y. Zong, Z. Ying, L. He, and X. Qiu Evaluating the performance of large language models on gaokao benchmark. arXiv preprint arXiv:2305.12474. Cited by: §4.1. Appendix A Missing Proofs A.1 Setting Teacher-Student Model Notations. We formally introduce distillation in the context of binary classification. Let X be the input space, =0,1Y= \0,1 \ be the label space, and D the probability distribution of inputs. The teacher h∗:→h :X is a fixed linear classifier, i.e. h∗()=∗⊤≥0,h (x )=1 \w_ x≥ 0 \, (11) for some ∗∈ℝd\0w_ ^d \0 \, where .1 \. \ returns 1 if the argument is true and 0 otherwise. The student is also a linear classifier, i.e. h()=⊤≥0.h (x )=1 \w x≥ 0 \. (12) Low-capacity students lie in a subspace of teacher models. We model the low-capacity student class as an s-dimensional subspace ⊂ℝdS ^d, and let P_S denote the orthogonal projection from the teacher parameter space ℝdR^d onto S. The orthogonal projection assumption is realistic for many teacher-student pairs used in practice. In model compression and distillation, the student often preserves the teacher’s architectural form while reducing capacity, for example, by using fewer transformer layers, fewer attention heads, or smaller hidden dimensions (13; 22; 37). Under this view, the student parameterization can be interpreted as a lower-dimensional subspace of the teacher parameter space, and the representable component of teacher knowledge is naturally modeled as the orthogonal projection onto this subspace. Distillation Dataset Construction. Distillation proceeds as follows. We collect a transfer set (i,yi)i=1n \ (x_i,y_i ) \_i=1^n consisting of inputs ix_i sampled i.i.d. from D and soft labels yi=σ(∗⊤i)y_i=σ (w_ x_i ) (13) provided by the teacher, where σ is the sigmoid function, σ(x)=11+exp(−x).σ (x )= 11+exp (-x )\;. (14) The soft real-valued labels can be thought of as a more informative version of the hard (0/1-valued) labels of the standard classification setting. We denote =[1,…,n]∈ℝd×nX= [x_1,…,x_n ] ^d× n (15) the data matrix. Student Model Optimization. The student is trained by minimizing the (normalized) cross-entropy loss L1() L^1(w) =−1n∑i=1nℓi(⊤i)−L∗ =- 1n _i=1^n _i(w x_i)-L (16) =−1n∑i=1n[yilogσ(⊤i)+(1−yi)log(1−σ(⊤i))]−L∗, =- 1n _i=1^n [y_i σ (w x_i )+ (1-y_i ) (1-σ (w x_i ) ) ]-L , (17) where L∗L is a normalization constant such that the minimum of L1L^1 is 0. Equation 16 is just the simplified notation. For better analysis on optimization, especially deep neural network optimization, involving many parameter components, we define L(1,…,N):=L1((N−1⋯1)⊤),L (W_1,…,W_N ):=L^1 ( (W_NW_N-1·sW_1 ) ), (18) and optimize it via gradient descent. We write i(τ)W_i (τ ) for the value of the matrix iW_i at time τ∈[0,∞)τ∈[0,∞), with i(0)W_i (0 ) denoting the initial value, and (τ)=N(τ)⋯1(τ)w (τ )=W_N (τ )·sW_1 (τ ). Then, each i(τ)W_i (τ ), for i∈1,…,Ni∈\1,…,N\, evolves according to the forllowing differential equation, ∂i(τ)∂τ=−∂L∂i(1(τ),…,N(τ)). _i (τ )∂τ=- ∂ L _i (W_1 (τ ),…,W_N (τ ) ). (19) A.2 Properties of Cross-Entropy Loss In this section, we show below the properties of cross-entropy loss shown in Equation 16. Proposition A.1 (Space of Gradient Updates). The gradient update is constrained in the data span, ∇L1()∈span()∇ L^1 (w ) (X ). Proof. The gradient with respect to the student weight vector w is ∇L1()=1n∑i=1n(σ(⊤i)−yi)⋅i.∇ L^1(w)= 1n _i=1^n (σ(w x_i)-y_i )·x_i\,. (20) Therefore, the gradient is always a linear combination of the training inputs (1,…,nx_1,…,x_n), and hence ∇L1()∈span().∇ L^1(w) (X)\,. (21) ∎ Proposition A.2 (Global Minima Condition). The global minimum of the cross-entropy loss is 0, and the set of global minimisers is ∈ℝd:⊤=⊤∗. \w ^d:X w=X w_ \\,. (22) Proof. We know that L1≥0L^1≥ 0 and L1(∗)=0L^1(w_ )=0, so 00 is the optimal objective value, and the set of global optima consists of all w such that L1()=0L^1 (w )=0. The last condition is equivalent to ∀i:ℓi()=0 _i: _i (w )=0, which in turn is equivalent to ∀i:σ(⊤i)=σ(∗⊤i) _i:σ (w x_i )=σ (w_ x_i ). By monotonicity of σ, this is further equivalent to ∀i:⊤i=∗⊤i _i:w x_i=w_ x_i, which is a restatement of Equation 22. ∎ Lemma A.3 (Bounds for Rayleigh Quotient). Let ∈ℝn×nA ^n× n be a real symmetric matrix, and let λmin() _ (A ) and λmax() _ (A ) denote its smallest and largest eigenvalues, respectively. Then λmin()⪯λmax(), _ (A ) _ (A ), (23) or equivalently, for every ∈ℝnz ^n, λmin()‖2≤⊤≤λmax()‖2, _ (A )\|z\|^2 Az≤ _ (A )\|z\|^2, (24) Proof. Since A is real symmetric, the spectral theorem gives an orthogonal matrix ∈ℝn×nQ ^n× n and a diagonal matrix Λ=diag(λ1,…,λn) =diag ( _1,…, _n ) (25) such that =Λ⊤,A=Q , (26) where λ1,…,λn _1,…, _n are the eigenvalues of A. Let λmin()=min1≤i≤nλi,λmax()=max1≤i≤nλi. _ (A )= _1≤ i≤ n\, _i, _ (A )= _1≤ i≤ n\, _i. (27) Define :=⊤.y:=Q z. (28) Because Q is orthogonal, it preserves Euclidean norm, so ‖=‖.\|y\|=\|z\|. (29) Then ⊤=⊤Λ⊤=⊤Λ=∑i=1nλi.z Az=z Q z=y = _i=1^n _iy_i^2. (30) Since λi≤λmax() _i≤ _ (A ) for all i, we have ∑i=1nλmin()i2≤∑i=1nλii2≤∑i=1nλmax()i2 _i=1^n _ (A )y_i^2≤ _i=1^n _iy_i^2≤ _i=1^n _ (A )y_i^2 (31) or λmin()∑i=1ni2≤∑i=1nλii2≤λmax()∑i=1ni2. _ (A ) _i=1^ny_i^2≤ _i=1^n _iy_i^2≤ _ (A ) _i=1^ny_i^2. (32) Hence, λmin()‖2≤⊤≤λmax()‖2. _ (A )\|y\|^2 Az≤ _ (A )\|y\|^2. (33) Because ‖=‖\|y\|=\|z\|, we obtain λmin()‖2≤⊤≤λmax()‖2. _ (A )\|z\|^2 Az≤ _ (A )\|z\|^2. (34) This proves the claim. ∎ Proposition A.4 (Restricted Strong Convexity). Assume X is full rank. For any sublevel set =:L1()≤lW= \w:L^1 (w )≤ l \, there exists μ>0μ>0 such that L1()≥L1()+∇L1()⊤(−)+μ2∥−∥2,L^1 (v )≥ L^1 (w )+∇ L^1 (w ) (v-w )+ μ2\|v-w\|^2, (35) for all ,∈w,\,v∈ W such that −∈span()v-w (X ). Proof. Consider the 2nd-Taylor explansion of L1L^1 around w, L1()=L1()+∇L1()⊤(−)+12(−)[∇2L1(¯)](−),L^1 (v )=L^1 (w )+∇ L^1 (w ) (v-w )+ 12 (v-w ) [∇^2L^1 ( w ) ] (v-w ), (36) where ∇2L1(¯)∇^2L^1 ( w ) is the Hessian of L1L^1 evaluated at ¯ w, a point lying between v and w. Hessian takes the form ∇2L1(¯)=¯⊤,∇^2L^1 ( w )=XD_ wX , (37) where ¯=diag[σ(¯⊤1)(1−σ(¯⊤1)),…,σ(¯⊤n)(1−σ(¯⊤n))].D_ w=diag [σ ( w x_1 ) (1-σ ( w x_1 ) ),…,σ ( w x_n ) (1-σ ( w x_n ) ) ]. (38) We now show that there is a constant ω>0ω>0 such that σ(¯i)(1−σ(¯i))≥ω,σ ( wx_i ) (1-σ ( wx_i ) )≥ω, (39) for all ¯∈ w and i∈1,…,ni∈ \1,…,n \, so that we can claim ¯⪰ωD_ w , consequently ∇2L1(¯)⪰ω⊤∇^2L^1 ( w ) . Let ∈w . The bound on L1()L^1 (w ) implies a bound on ℓ(⊤i) (w x_i ) for all i, ℓ(⊤i)≤nL1()≤nl. (w x_i )≤ nL^1 (w )≤ nl. (40) Because ℓi _i is convex and ℓi(u)→∞ _i (u )→∞ as u→±∞u→±∞, we know that ℓi−1((−∞,nl]) ^-1_i ((-∞,nl] ) is a bounded interval, and the finite union ∪i=1nℓi−1((−∞,nl])∪^n_i=1 ^-1_i ((-∞,nl] ) is also a bounded interval, whose size depends only on nlnl and the data. Hence, there exists K>0K>0 such that ⊤i∈[−K,K]w x_i∈ [-K,K ] for all ∈w and i∈1,…,ni∈ \1,…,n \. The existence of ω>0ω>0 satisfies Equation 39. Now, let us apply ∇2L1()⪰ω⊤∇^2L^1 (w ) a lower bound to the right-hand side in Equation 36: L1()≥L1()+∇L1()⊤(−)+ω2(−)⊤(−).L^1 (v )≥ L^1 (w )+∇ L^1 (w ) (v-w )+ ω2 (v-w ) X X (v-w ). (41) Consider two cases. ■ Case 1: If n≥dn≥ d, ⊤XX is full-rank. Applying Lemma A.3, we have ⊤⪰λminXX _ I holds, where λmin>0 _ >0 is the smallest eigenvalue of ⊤XX . Combined with Equation 41, this proves the claim for n≥dn≥ d and μ=ωλminμ=ω _ . ■ Case 2: If n<dn<d, ⊤X X is full-rank. We use the assumption −∈span()v-w (X ) to deduce ‖−‖2 \|v-w\|^2 =‖(−)‖2 =\|P_X (v-w )\|^2 (42) =(−)⊤(⊤)−1⊤(−) = (v-w ) X (X X )^-1X (v-w ) (43) ≤λmax(−)⊤⊤(−), ≤ _ (v-w ) XX (v-w ), (44) where λmax>0 _ >0 is the largest eigenvalue of (⊤)−1 (X X )^-1, according to Lemma A.3. Combined with Equation 41, this proves the claim for n<dn<d and μ=ω/λmaxμ=ω/ _ . ∎ Proposition A.5 (Restricted Polyak-Lojasiewicz). Assume X is full-rank. For any sublevel set =:L1()≤lW= \w:L^1 (w )≤ l \, there exists c>0c>0 such that cL1()≤12‖∇L1()‖,cL^1 (w )≤ 12\|∇ L^1 (w )\|\,, (45) for all ∈.w∈ W. Proof. Let ∈w∈ W. (If W is empty, the claim is trivially true.) Theorem A.4 applied to W implies that for some μ>0μ>0, L1()≥L1()+∇L1()(−)+μ2‖−‖2,L^1 (v )≥ L^1 (w )+∇ L^1 (w ) (v-w )+ μ2\|v-w\|^2, (46) for all ∈∩v where =:−∈span()V= \v:v-w (X) \. Taking min∈∩ _v on both sides, then relaxing part of the constraint on the right-hand side yields min∈∩L1() _v∈ W L^1 (v ) ≥min∈∩L1()+∇L1()⊤(−)+μ2∥−∥2 ≥ _v∈ W L^1 (w )+∇ L^1 (w ) (v-w )+ μ2\|v-w\|^2 (47) ≥min∈L1()+∇L1()⊤(−)+μ2∥−∥2. ≥ _v∈ VL^1 (w )+∇ L^1 (w ) (v-w )+ μ2\|v-w\|^2. (48) ∎ A.3 Property of Low-Capacity Distilled Learning Theorem A.6 (Projected-Data Equivalence under a Capacity Constraint). When optimization is restricted to the student space ℱF, the empirical risk depends on the training inputs only through their projections onto ℱF. Proof. We derive the loss for a student model parameter as follows L1(ℱ)=L1(Pℱ) L^1(w_F)=L^1(P_Fw) =1n∑i=1nℓi((Pℱ)⊤i) = 1n _i=1^n _i((P_Fw) x_i) (49) =1n∑i=1nℓi(⊤(Pℱ⊤i)) = 1n _i=1^n _i(w (P_F x_i) ) (50) =1n∑i=1nℓi(⊤(Pℱi)). = 1n _i=1^n _i(w (P_Fx_i) ). (51) The last equation is the loss of the teacher model on the data projected on the student model space. ∎ Interpretation. The theorem formalizes the statement that a low-capacity linear student can only exploit the component of the data lying in its parameter space ℱF. Any component of ix_i in ℱ⟂F is invisible to the student, because for every ∈ℱu , ⊤i=(Pℱ⊤)i=⊤(Pℱi).u x_i=(P_Fu )x_i=u (P_Fx_i). (52) Thus, the effective data span available to the student is span(Pℱ)=Pℱspan().span(P_FX)=P_F\,span(X). (53) In particular, when distilling from a larger model into a student whose parameter space is ℱF, the student can only learn teacher behavior through the component of the training data that survives this projection. Theorem A.7 (Space of Distilled Models). Assume the student is a directly parameterised linear classifier (N=1). Then, (τ)∈(0)+span(),w(τ) (0)+span(X)\,, (54) for τ∈[0,−∞)τ∈[0,-∞), and (0)w(0) is the initialized model parameter. Proof. From Theorem A.1, the gradient satisfies ∇L1()∈span(),∇ L^1(w) (X), (55) for every w. Under gradient flow for (N=1N=1), dτ(τ)=−∇L1((τ)), d dτw(τ)=-∇ L^1(w(τ)), (56) so the velocity vector (˙(τ) w(τ)) always lies in (span()span(X)). Define (τ):=(τ)−(0).u(τ):=w(τ)-w(0). (57) Then (0)=0,dτ(τ)=dτ(τ)=−∇L1((τ))∈span().u(0)=0, d dτu(τ)= d dτw(τ)=-∇ L^1(w(τ)) (X). (58) Since (0)=0∈span()u(0)=0 (X) and its derivative always lies in span()span(X), it follows that (τ)∈span()u(τ) (X) for all τ≥0τ≥ 0. Therefore, (τ)=(0)+(τ)∈(0)+span().w(τ)=w(0)+u(τ) (0)+span(X). (59) So the trajectory is not confined to the linear subspace span()span(X) anymore; it is confined to the affine subspace obtained by translating span()span(X) by the initial point (0)w(0). ∎ A.4 Optimal Distillation Solution of a One-Layer Linear Student Model Theorem A.8 (High-capacity One-layer Student Solution). Assume the student is a directly parameterised linear classifier (N = 1). Then, the student’s weight vector almost surely coverges to w, where ^=0+P(∗−0), w=w_0+P_X (w_ -w_0 ), (60) for τ→∞τ→∞. Proof. Because (τ)w (τ ) evolve according to gradient flow, dτ(τ)=−∇L1((τ)); d dτw(τ)=-∇ L^1(w(τ)); (61) hence, L′(τ)=dτL1((τ))=∇L1((τ))⊤dτ(τ)=−∥∇L1((τ))∥2.L (τ )= d dτL^1 (w (τ ) )=∇ L^1 (w (τ ) ) d dτw (τ )=-\|∇ L^1 (w (τ ) )\|^2. (62) The data matrix X is full-rank, we can therefore apply Proposition A.5 to =:L1()≤L1(0)W= \w:L^1 (w )≤ L^1 (0 ) \ and (τ)w (τ ) to lower-bound the gradient norm on the right-hand side of Equation 62. We obtain L′(τ)≤−cL(τ)L (τ )≤-cL (τ ) (63) for some c>0c>0 and all τ∈[0,∞)τ∈[0,∞), or equivalently, (logL(τ))′≤c. ( L (τ ) ) ≤ c. (64) Integrating over [0,t] [0,t ] yields L(t)≤L(0)⋅e−ctL (t )≤ L (0 )\,·\,e^-ct, which proves global convergence in the objective L(t)→0L (t )→ 0 as t→∞t→∞. Now invoke Proposition A.4 with W as above, =(t)v=w (t ) and =^w= w (we know that both (τ),^∈∩span()w (τ ), w (X ), partly by Theorem A.7): L(t)≥μ2‖(t)−‖2.L (t )≥ μ2\|w (t )-w\|^2. (65) Since L(t)→0L (t )→ 0 as t→∞t→∞, then the theorem follows. ∎ Interpretation. Theorem A.8 shows that training can only transfer teacher knowledge that is exposed by the data. The vector ∗−0w_ -w_0 represents the knowledge gap between teacher and student. The projection term P(∗−0)P_X(w_ -w_0) indicates that the knowledge gap between the teacher and the student is transferred only through its component in the data span. Hence, learning is fundamentally constrained to the subspace revealed by the training data. When the data span covers the entire knowledge gap, the student can fully learn the teacher knowledge; otherwise, it can recover only the component of that gap that is identifiable from the data. Thus, the data act as a mirror of the teacher: if the data span aligns well with the knowledge gap, the student learns effectively; if it is orthogonal to that gap, then no transfer occurs. Moreover, directions orthogonal to the data span are never updated, so the final student remains partially constrained by its initialization 0w_0. Theorem A.9 (Low-capacity One-layer Student Solution). Assume the low-capacity student is a directly parameterised linear classifier (N = 1). Then, the student’s weight vector converges to ^ℱ w_F, where ^ℱ=0+Pℱ(∗−0), w_F=w_0+P_X_F (w_ -w_0 ), (66) for τ→∞τ→∞. Proof. By Theorem A.6, under the capacity constraint ℱF, training the one-layer student on the original dataset is equivalent to training an unconstrained one-layer student on the projected data PℱXP_FX. In particular, the effective data span becomes ℱ:=span(Pℱ)=Pℱspan().X_F:=span(P_FX)=P_Fspan(X). Now apply Theorem A.8 to this projected problem. Since Theorem A.8 states that for a directly parameterized one-layer linear classifier, gradient flow converges to the initial point plus the orthogonal projection of the teacher–student gap onto the data span, we obtain (τ)→0+Pℱ(∗−0)as τ→∞.w(τ) _0+P_X_F(w -w_0) τ→∞. This is exactly the claim of Theorem A.9. ∎ A.5 Optimal Distillation Solution of a Deep Linear Student Model Theorem A.10 (High-capacity Deep Linear Student Solution). Let w be defined as in Theorem A.8. Assume the student is a deep linear network w, initialized such that for some ϵ>0ε>0, satisfying the conditions below, ‖(0)‖<min‖^‖,ϵN(ϵ2‖^‖−2N+‖^‖2−2N)−N2,\|w (0 )\|< \\| w\|,ε^N (ε^2\| w\|^- 2N+\| w\|^2- 2N )^- N2 \, (67) L1((0))<L1(),L^1 (w (0 ) )<L^1 (0 ), (68) j+1(0)⊤j+1(0)=j(0)j(0)⊤W_j+1 (0 ) W_j+1 (0 )=W_j (0 )W_j (0 ) (69) for j=1,…,N−1j=1,…,N-1. Then, for n≥dn≥ d, student’s weight vector converges ‖(t)−^‖≤ϵ,\|w (t )- w\|≤ε, (70) for all t large enough. Proof. For the proof, we need the result by (4), characterizing the induced flow on (τ)w (τ ) when running gradient descent on the component matrices iW_i. Lemma A.11 (Claim 2 in (4)). If the ballancedness condition Equation 69 holds, then ∂(τ)∂τ=−∥(τ)∥2(N−1)N((∇L1(τ))+(N−1)⋅(τ)∇L1((τ))). (τ )∂τ=-\|w (τ )\| 2 (N-1 )N ( (∇ L^1 (τ ) )+ (N-1 )·P_w (τ )∇ L^1 (w (τ ) ) ). (71) We start by looking at the time-derivative of L, L′(τ) L (τ ) =∇L1((τ))⊤(∂(τ)∂τ) =∇ L^1 (w (τ ) ) ( (τ )∂τ ) (72) =−∥(τ)∥2(N−1)N(∥∇L1((τ))∥2+(N−1)⋅∥(τ)∇L1((τ))∥2) =-\|w (τ )\| 2 (N-1 )N (\|∇ L^1 (w (τ ) )\|^2+ (N-1 )·\|P_w (τ )∇ L^1 (w (τ ) )\|^2 ) (73) ≤−∥(τ)∥2(N−1)N⋅∥∇L1((τ))∥2 ≤-\|w (τ )\| 2 (N-1 )N·\|∇ L^1 (w (τ ) )\|^2 (74) It is non-positive, so (τ)w (τ ) stays within the L((0))L (w (0 ) )-sublevel set thoughout optimization, (τ)∈=:L1()≤L((0)).w (τ ) = \w:L^1 (w )≤ L (w (0 ) ) \. (75) Also, W is convex and by assumption shown in Assumption 68 it does not contain 0. We can therefore take δ>0δ>0 to be the distance between W and 0, and it follows that ‖(τ)‖≥δ\|w (τ )\|≥δ for τ∈[0,∞)τ∈[0,∞). Now, noting that X is full-rank, apply Proposition A.5 to W and (τ)w (τ ) to upper-bound the right-hand side of Inequality 74, L′(τ)≤−cδ2(N−1)NL(τ)L (τ )≤-cδ 2 (N-1 )NL (τ ) (76) Letting c~=cδ2(N−1)N c=cδ 2 (N-1 )N, we get (logL(τ))′≤−c~ ( L (τ ) ) ≤- c and consequently L(t)≤L(0)⋅e−c~tL (t )≤ L (0 )· e^- ct. This proves convergence in the objective, L(t)→0L (t )→ 0 as t→∞t→∞. To prove convergence in parameters, we decompose the current-optimal gap (τ)−^w (τ )- w into orthogonal components and bound each of them separately, ‖(τ)−^‖2=‖((τ)−^)‖2+‖((τ)−^)‖2,\|w (τ )- w\|^2=\|P_X (w (τ )- w )\|^2+\|P_Q (w (τ )- w )\|^2, (77) where the columns of ∈ℝd×(d−n)Q ^d× (d-n ) orthogonally complement those of X. If n≥dn≥ d, we simply bound the first term and disregard the second one. To bound the first term, invoke the Proposition A.4 with W, =(τ)v=P_Xw (τ ) and =^w=P_X w. One can check that L1()=L1()L^1 (P_Xu )=L^1 (u ) for all ∈ℝdu ^d, so (τ)∈P_Xw (τ ) and our use of the theorem is legal. We obtain L(τ)≥μ2‖((τ)−^)‖2.L (τ )≥ μ2\|P_X (w (τ )- w )\|^2. (78) Since L(τ)→0L(τ)→ 0, it follows that ‖((τ)−^)‖2→0\|P_X (w (τ )- w )\|^2→ 0 (79) as τ→∞τ→∞. For the second term, notice that ^∈span() w (X ), so (^)P_Q ( w ) vanishes and we are left with ‖(τ)‖2\|P_Qw (τ )\|^2. Denote this quanity q(τ)q (τ ). Its time derivative is q′(τ) q (τ ) =2((τ))⊤(∂(τ)∂τ) =2 (P_Qw (τ ) ) ( (τ )∂τ ) (80) =−2∥(τ)∥2(N−1)N((τ)τ∇L1((τ)) =-2\|w(τ)\| 2(N-1)N (w(τ)^τP_Q∇ L^1(w(τ)) (81) +N−1‖(τ)‖2⋅(τ)⊤(τ)⋅(τ)⊤∇L1((τ))) + N-1\|w(τ)\|^2·w(τ) P_Qw(τ)·w(τ) ∇ L^1(w(τ)) ) (82) =−2q(τ)(N−1)∥(τ)∥−2N(τ)⊤∇L1((τ)), =-2q(τ)(N-1)\|w(τ)\| -2Nw(τ) ∇ L^1(w(τ)), (83) where we have used the fact that ∇L1((τ))∈span()∇ L^1(w(τ)) (X) stated in Theorem A.1 and Q is orthogonal to X. Rearranging, we obtain dτ(logq(τ)2(N−1))=−∥(τ)∥−2N⋅(τ)⊤∇L1((τ)). d dτ ( q(τ)2(N-1) )=-\|w(τ)\|^- 2N·w(τ) ∇ L^1(w(τ)). (84) It turns out that the right-hand side expression is integrable in yet another way, namely dτ(12Nlog∥(τ)∥2)=−∥(τ)∥−2N⋅(τ)⊤∇L1((τ)). d dτ ( 12N \|w(τ)\|^2 )=-\|w(τ)\|^- 2N·w(τ) ∇ L^1(w(τ)). (85) Equating the two and integrating over [0,t][0,t] yields logq(t)q(0)=N−1N⋅log‖(t)‖2‖(0)‖2, q(t)q(0)= N-1N· \|w(t)\|^2\|w(0)\|^2, (86) which implies q(t)‖(t)‖2≤(‖(0)‖(t)‖)2N, q(t)\|w(t)\|^2≤ ( \|w(0)\|\|w(t)\| ) 2N, (87) because q(0)≤‖(0)‖2q(0)≤\|w(0)\|^2. We now bound the norm of (t)w(t). Starting from an orthogonal decompositon similar to Equation 77 and applying convergence on X space in Equation 79 and bound for parameter on X-orthogonal space in Equation 87, we get ‖(t)‖2=‖(t)‖2+‖(t)‖2\|w(t)\|^2=\|P_Xw(t)\|^2+\|P_Qw(t)\|^2 (88) lim supt→∞‖(t)‖2≤‖^‖2+|(0)|lim supt→∞2N‖(t)‖2−2N _t→∞\|w(t)\|^2≤\| w\|^2+\|w(0)\| 2N _t→∞\|w(t)\|^2- 2N (89) Denote ν:=lim supt→∞‖(t)‖ν:= _t→∞\|w(t)\|. By the same orthogonal decomposition, we also know that ν2≥lim supt→∞‖(t)‖2=‖^‖2ν^2≥ _t→∞\|P_ Xw(t)\|^2=\| w\|^2, so we can divide both sides above by ν2ν^2, 1≤‖^‖2ν2+‖(0)‖2Nν2N=:f(ν).1≤ \| w\|^2ν^2+ \|w(0)\| 2Nν 2N=:f(ν). (90) On the right-hand side, we now have a decreasing function of ν that goes to zero as ν→∞ν→∞. However,evaluated at our specific ν, it is lower-bounded by 1, implying an implicit upper bound for ν. How do we find this bound? Suppose we find some constant K such that f(K)≤1f(K)≤ 1. Then , because f is decreasing, it must be the case that ν≤Kν≤ K. One such candidate for K is K=‖^‖⋅(1−‖(0)2N‖^‖2N)−N2(N−1).K=\| w\|· (1- \|w(0) 2N\|\| w\| 2N )^- N2(N-1). (91) To check that indeed f(K)≤1f(K)≤ 1, start from the inequality (‖^‖K)2(N−1)N+(‖(0)‖^‖)2N=1≤(1−(‖(0)‖^‖)2N)−1N−1=(‖^‖K)−2N. ( \| w\|K ) 2(N-1)N+ ( \|w(0)\|\| w\| ) 2N=1≤ (1- ( \|w(0)\|\| w\| ) 2N )^- 1N-1= ( \| w\|K )^- 2N. (92) Taking the leftmost and rightmost expression and multiply by (‖^‖/K)2/N(\| w\|/K)^2/N yields f(K)=‖^‖2K2+‖(0)‖2NK2N≤1.f(K)= \| w\|^2K^2+ \|w(0)\| 2NK 2N≤ 1. (93) Hence, lim supt→∞‖(t)‖≤‖^‖⋅(1−(‖(0)‖^‖)2N)−N2(N−1). _t→∞\|w(t)\|≤\| w\|· (1- ( \|w(0)\|\| w\| ) 2N )^- N2(N-1). (94) Finally, let us turn back to our original goal of bounding ‖(τ)−^‖2\|w(τ)- w\|^2. With Equation 77, Equation 79, Equation 87, Equation 94, we now know that lim supt→∞‖(τ)−^‖2 _t→∞\|w(τ)- w\|^2 ≤‖(0)‖2N‖^‖2(N−1)N(1−((0)^2N))−1 ≤\|w(0)\| 2N\| w\| 2(N-1)N (1- ( w(0) w 2N ) )^-1 (95) =‖^‖2+2N‖^‖2N−‖(0)‖2N−‖^‖2. = \| w\|^2+ 2N\| w\| 2N-\|w(0)\| 2N-\| w\|^2. (96) Hence, if we initialize close enough to zero, as specified by condition in Inequality 67, we ensure that lim supt→∞‖(τ)−^‖2<ϵ2. _t→∞\|w(τ)- w\|^2<ε^2. (97) ∎ Theorem A.12 (Low-capacity Deep Linear Student Solution). Let ^ℱ=0+PXℱ(∗−0) w_F=w_0+P_X_F(w -w_0) be defined as in Theorem A.9. Under the same assumptions as in Theorem A.10, suppose the student is a low-capacity deep linear network with predictor (τ)w(τ). Then, for every ϵ>0ε>0, there exists τϵ>0 _ε>0 such that ‖(τ)−^ℱ‖≤ϵ,∀τ≥τϵ.\|w(τ)- w_F\|≤ε, ∀τ≥ _ε. Proof. By Theorem A.8, under the capacity constraint ℱF, training the low-capacity deep linear student on the original dataset is equivalent to training an unconstrained deep linear student on the projected data PℱP_FX. In particular, the effective data span becomes ℱ:=span(Pℱ)=Pℱspan().X_F:=span(P_FX)=P_Fspan(X). Now apply Theorem A.10 to this projected problem. Theorem A.10 states that, under the same initialization assumptions, the predictor of a deep linear network converges arbitrarily close to the optimal solution associated with the effective data span. Therefore, for every ϵ>0ε>0, there exists τϵ>0 _ε>0 such that ‖(τ)−^ℱ‖≤ϵ,∀τ≥τϵ,\|w(τ)- w_F\|≤ε, ∀τ≥ _ε, where ^ℱ=0+PXℱ(∗−0). w_F=w_0+P_X_F(w -w_0). This is exactly the claim of Theorem A.12. ∎ Appendix B Experiment Setup B.1 Training We fine-tune each student model using full-parameter supervised fine-tuning (SFT) on the selected training subset, such as the high-likelihood or low-likelihood data. Concretely, training starts from the corresponding pretrained checkpoint (e.g., Qwen/Qwen2.5-1.5B) and uses full finetuning rather than parameter-efficient adaptation. All experiments are run on 2 NVIDIA A100 GPUs. To support long-context training efficiently, we use DeepSpeed ZeRO-3, FlashAttention-2, gradient checkpointing, and bfloat16 precision. The maximum sequence length is set to 8192 tokens, and data are formatted with the Qwen template. For optimization, we use a per-device batch size of 1, gradient accumulation steps of 1, a learning rate of 1.25×10−61.25× 10^-6, cosine learning-rate decay, no warmup, and train for 5 epochs. We log training statistics at every step and save checkpoints every 1110 steps while storing model weights only. This training configuration is fixed across data-selection settings so that any performance differences can be attributed to the choice of training data rather than changes in optimization or hardware setup. Appendix C Extended Results C.1 Small Qwen models show Fast-Fit without eventual crossover Table 4 and Table 5 show Fast-Fit without eventual crossover for small Qwen models. The 1.5B result is the cleanest example of the capacity bottleneck. For Qwen2.5-1.5B, high-likelihood training jumps from an Avg of 0.58 at initialization to 21.26 after only 0.25 epoch, then reaches 24.16 by epoch 1. Low-likelihood training reaches only 12.13 at 0.25 and 14.77 at epoch 1. More importantly, another four epochs do not close the gap: its best is only 16.50 at epoch 2.25. Qwen2.5-3B is slightly different. Low likelihood exhibits a real delayed gain: its early best Avg is 19.93, whereas its eventual best reaches 24.63, a +4.70 point improvement. But high likelihood remains far ahead at 33.38. C.2 Qwen2.5-7B is the actual transition regime Table 6 shows Qwen2.5-7B appears to occupy a transitional capacity regime between small models that cannot effectively exploit low-likelihood supervision and larger models that benefit from it consistently. During the early stage of training, high-likelihood data has a clear advantage, achieving a best average score of 44.09 within the first epoch compared with 37.05 for low-likelihood data. However, the low-likelihood trajectory continues to improve substantially with additional optimization, rising to 43.07 at epoch 1.25, 45.02 at epoch 2.25, and eventually reaching 46.19 at epoch 4.75, exceeding the best high-likelihood result of 45.41 and achieving higher peak performance on six of the ten benchmarks. Importantly, this crossover is not persistent across all checkpoints: low-likelihood training outperforms high-likelihood training in average score at epochs 2.25, 2.5, 4.0, 4.5, and 4.75 but falls below it again at several intermediate checkpoints. This non-monotonic behavior suggests that the 7B model has sufficient representational capacity to extract useful knowledge from more difficult, low-likelihood supervision, yet optimization remains challenging enough that these gains are unstable and checkpoint-sensitive. Rather than simply demonstrating that larger models benefit from low-likelihood data, the 7B results therefore reveal an intermediate regime characterized by intermittent access to difficult knowledge, bridging the failure of low-capacity models and the more sustained Slow-Gain behaviour observed in higher-capacity models. C.3 Qwen3-8B gives the strongest evidence for Slow-Gain Table 7 illustrates that Qwen3-8B provides the clearest evidence for the Slow-Gain regime. High-likelihood training begins with a substantial early advantage, reaching an average score of 66.16 at epoch 0.25 compared with 57.72 for low-likelihood training. However, the low-likelihood trajectory steadily closes this gap, overtakes the high-likelihood trajectory at epoch 2.25, and remains superior at every subsequent checkpoint through epoch 5 in terms of simple average score. The contrast between early and eventual performance is particularly revealing: the best high-likelihood average improves only marginally from 66.16 in the early stage to 66.64 overall, a gain of just 0.48 points, whereas the low-likelihood average rises from an early best of 62.62 to 70.69, corresponding to an 8.07-point improvement. This pattern closely matches the proposed Fast-Fit / Slow-Gain dynamics: high-likelihood supervision enables rapid early acquisition but leaves relatively little room for further improvement, while low-likelihood supervision starts from a weaker region yet provides substantially greater long-term gains. One subtlety is that the conclusion depends somewhat on the aggregation metric: at epoch 0.5, the simple average favors high-likelihood training (64.79 vs. 62.62), whereas the weighted average favors low-likelihood training (74.65 vs. 70.17). For this reason, the Fast-Fit interpretation is most convincingly supported by the simple average together with the per-benchmark comparison, which shows that high-likelihood training dominates most individual tasks early, while low-likelihood training becomes consistently stronger after sufficient optimization. C.4 Generalization to Llama family The Llama results in Table 8 and Table 9 provide important cross-family evidence that the observed capacity-dependent effect is not specific to Qwen models. Llama3.2-3B closely follows the behavior of the smaller Qwen students: high-likelihood training achieves the better maximum performance on 9 of the 10 benchmarks, while the only nominal advantage for low-likelihood training occurs on Minerva and is extremely small, 20.82 versus 20.77. Moreover, low-likelihood training never surpasses high-likelihood training at the same checkpoint in either the simple or weighted average, providing strong evidence that low-capacity models consistently benefit more from supervision that is closer to their current distribution. In contrast, Llama3.1-8B exhibits the opposite pattern: low-likelihood training achieves higher peak performance on 6 of the 10 benchmarks, reaches a best average score of 20.93 compared with 20.10 for high-likelihood training, and outperforms high-likelihood training in simple average at 16 of the 20 checkpoints. Together, these results reproduce the same capacity-dependent trend across a different model family: high-likelihood supervision is more effective for 3B-scale students, whereas low-likelihood supervision becomes increasingly competitive and can ultimately outperform it at 8B scale. However, Llama3.1-8B also introduces an important qualification to the Fast-Fit / Slow-Gain interpretation, since high-likelihood performance does not necessarily peak during the earliest stage and continues to improve from 15.45 at epoch 1 to 20.10 at epoch 3.75. Therefore, a more precise characterization is that high-likelihood supervision establishes an early learning advantage, whereas low-likelihood supervision exhibits larger delayed gains as model capacity increases and may eventually overtake high-likelihood training. C.5 Best-Checkpoint vs. Trajectory-Level Comparison of 7B and 8B via Peak Accessibility and Persistent Superiority Metrics An important nuance is that Table 2 reports a best-checkpoint envelope: each benchmark score is selected independently as the best result among 20 checkpoints, and therefore the ten reported scores within a column do not necessarily correspond to the same model checkpoint. This distinction is particularly relevant for Qwen2.5-7B. Although low-likelihood training achieves higher benchmark-wise peak performance on 6 out of 10 datasets, suggesting a clear reversal relative to high-likelihood training, a checkpoint-aligned comparison reveals a more nuanced pattern. In terms of the simple average, low-likelihood training outperforms high-likelihood training at only 5 out of 20 checkpoints, while their average performance after the first epoch is approximately comparable and the low-likelihood trajectory exhibits greater temporal variability. These results suggest two distinct notions of improvement, which can be formalized as the following conditions: Peak Accessibility:maxtPlow(t)>maxtPhigh(t)Peak Accessibility: _tP_low(t)> _tP_high(t) Persistent Superiority:Plow(t)>Phigh(t),∀t∈latePersistent Superiority: P_low(t)>P_high(t), ∀ t _late where T denotes the set of all evaluated training checkpoints, while tearlyt_early denotes the boundary of the early-training regime. The first epoch (t≤1t≤ 1) is treated as the early stage, so lateT_late contains all checkpoints after the first epoch. Thus, persistent superiority is a stronger condition than merely achieving a higher peak: it requires low-likelihood training to remain better than high-likelihood training throughout the entire post-early-training region, rather than outperforming it at only one or a few isolated checkpoints. Qwen2.5-7B strongly satisfies the former criterion but only weakly satisfies the latter, suggesting that it lies in a transitional capacity regime where difficult supervision can unlock stronger solutions but remains optimization-sensitive. In contrast, Qwen3-8B satisfies both criteria: after the crossover, low-likelihood training consistently exceeds high-likelihood training across subsequent checkpoints, providing stronger evidence of a sustained Slow-Gain regime rather than merely a higher best-checkpoint outcome. C.6 From Fast-Fit to Sustained Slow-Gain: A Capacity-Dependent Transition Overall, the extended results in Tables 4–9 suggest that likelihood-based data selection induces not merely two, but three distinct learning regimes as model capacity increases. In the low-capacity regime, represented by smaller models such as Qwen2.5-1.5B, Qwen2.5-3B, and Llama3.2-3B, high-likelihood supervision exhibits a clear Fast-Fit / No Catch-up pattern: the model benefits rapidly from examples close to its current distribution, whereas low-likelihood supervision remains difficult to absorb and fails to close the performance gap even with additional training. In the intermediate-capacity regime, exemplified most clearly by Qwen2.5-7B, the dynamics shift to Fast-Fit / Unstable Slow-Gain. High-likelihood data still provides the stronger early optimization signal, but low-likelihood training eventually reaches higher-performing solutions at selected checkpoints; however, this advantage is intermittent rather than persistent, indicating that the model has sufficient capacity to exploit difficult supervision but remains sensitive to optimization dynamics. Finally, in the high-capacity regime, as observed most prominently for Qwen3-8B and also for Llama3.1-8B, training exhibits a Fast-Fit / Sustained Slow-Gain Reversal: high-likelihood supervision retains an early advantage, but low-likelihood supervision produces substantially larger delayed gains and can eventually overtake high-likelihood training across a broad range of later checkpoints. These three regimes suggest a gradual capacity-dependent transition from failure to absorb off-distribution supervision, through unstable access to its additional knowledge, to stable and sustained exploitation of that knowledge, providing a more fine-grained characterization of the interaction between model capacity, data likelihood, and training duration. Table 4: Qwen-2.5-1.5B results across training checkpoints under high- and low-likelihood data selection. Cyan marks the best value within the early stage (epochs 0–1), yellow marks the best value across all checkpoints within each setting, and green marks cells that satisfy both. Dashed rules separate the earliest stage from later training. Setting Epoch AIME24 AMC CNM GK GPQA GSM KY MATH MNV OB W. Avg Avg High 0 0 0 0 0 3.03 0 0 1.4 1.47 0.44 0.88 0.58 0.25 0 22.5 6.67 2.53 27.27 16.67 15.58 52.6 19.49 20.59 26.85 21.26 0.5 0 27.5 6.67 10.13 26.26 22.38 19.1 52.2 22.06 19.7 27.9 23.27 0.75 3.33 25 6.67 7.59 28.79 21.43 16.58 54.6 18.01 21.78 28.52 23.62 1 3.33 27.5 6.67 10.13 22.73 20.48 17.59 53.4 24.63 19.26 27.99 24.16 1.25 6.67 25 0 11.39 23.74 22.86 18.09 54 22.43 18.07 27.6 22.93 1.5 6.67 22.5 0 6.33 26.77 22.38 16.58 55.8 23.53 18.67 28.21 22.84 1.75 0 27.5 0 5.06 21.21 21.9 19.6 54.6 20.96 17.33 26.85 21.47 2 0 30 0 1.27 25.25 20.48 14.57 57.6 22.43 19.56 28.34 23.01 2.25 3.33 27.5 0 3.8 26.77 20.00 15.58 56.2 20.59 20.74 28.34 22.96 2.5 3.33 25 0 7.59 25.25 22.86 17.09 55.8 22.79 21.93 29.35 24.15 2.75 3.33 22.5 6.67 2.53 25.25 17.62 15.08 55.4 19.49 21.04 27.81 22.99 3 3.33 27.5 3.33 2.53 24.75 14.76 13.57 55.4 22.06 19.7 27.51 23.54 3.25 6.67 25 0 3.8 29.29 13.33 16.08 55 21.32 20.89 27.99 23.22 3.5 3.33 35 0 3.8 23.74 14.29 13.57 57.2 22.43 20.15 28.08 24.14 3.75 6.67 22.5 0 3.8 22.73 15.24 14.57 56.8 20.59 20.74 27.99 23.78 4 0 22.5 3.33 2.53 24.24 12.86 13.57 54.8 22.79 21.04 27.68 23.42 4.25 0 22.5 3.33 2.53 25.76 12.38 14.07 55.4 22.06 20.3 27.55 23.12 4.5 0 22.5 3.33 2.53 23.74 13.81 13.57 56 21.32 22.52 28.12 23.03 4.75 0 22.5 3.33 2.53 24.24 14.29 18.09 54.2 23.53 21.19 28.03 23.26 5 0 17.5 3.33 2.53 22.73 12.86 12.06 52.8 22.79 20.00 26.41 21.69 Low 0 0 0 0 0 3.03 0 0 1.4 1.47 0.44 0.88 0.58 0.25 0 12.5 0 1.27 16.67 10.48 5.03 37 13.24 11.26 16.69 12.13 0.5 0 17.5 0 3.8 11.11 8.57 5.03 41.2 9.93 10.96 16.95 13.46 0.75 0 12.5 0 6.33 7.07 11.43 2.51 40.4 12.5 10.96 16.73 12.88 1 0 17.5 3.33 3.8 11.11 13.33 8.04 41.8 12.5 13.04 18.7 14.77 1.25 3.33 17.5 0 6.33 14.65 10.48 5.53 41.2 12.87 13.19 18.57 14.82 1.5 3.33 12.5 3.33 6.33 10.1 8.1 4.52 41.8 13.6 11.85 17.61 13.77 1.75 3.33 15 0 7.59 13.13 8.1 6.03 41.6 12.5 14.96 18.83 14.57 2 3.33 20.00 0 5.06 9.09 8.57 9.05 40.2 14.71 12.89 18.22 14.99 2.25 3.33 22.5 0 8.86 11.11 11.43 8.04 39.8 9.93 14.52 18.79 16.5 2.5 0 17.5 0 5.06 11.11 10.95 10.55 41.6 10.29 12.44 18.44 15.23 2.75 0 15 0 7.59 13.13 9.05 5.03 43.2 12.5 13.19 18.7 14.61 3 0 22.5 0 7.59 10.1 10.95 4.52 43.8 14.34 12.15 18.83 15.63 3.25 0 10 6.67 3.8 13.64 10.48 7.04 44.4 13.97 13.33 19.54 15.57 3.5 0 17.5 0 3.8 12.12 10.48 7.04 39.8 10.66 12 17.74 15.04 3.75 0 12.5 0 5.06 10.61 12.38 7.54 41.8 10.29 13.63 18.57 14.71 4 0 22.5 0 5.06 11.11 9.05 6.53 43.8 12.13 13.33 19.1 16.14 4.25 0 20.00 0 7.59 12.63 10.48 5.03 43.2 11.4 12.89 18.83 15.75 4.5 0 15 0 6.33 11.62 12.86 6.03 42.8 13.24 12.89 19.01 15.34 4.75 0 15 0 6.33 12.63 13.33 7.04 43 12.5 12.89 19.27 15.88 5 3.33 15 6.67 3.8 9.09 11.43 4.52 44.4 13.6 12.74 18.97 15.87 Table 5: Qwen-2.5-3B-Instruct results across training checkpoints under high- and low-likelihood data selection. Cyan marks the best value within the early stage (epochs 0.25–1), yellow marks the best value across epochs 0.25–5 within each setting, and green marks cells that satisfy both. Dashed rules separate the earliest stage from later training. Setting Epoch AIME24 AMC CNM GK GPQA GSM KY MATH MNV OB W. Avg Avg High 0.25 10 47.5 10 25.32 27.27 19.52 24.12 64.6 29.78 26.37 34.87 30.59 0.5 10 32.5 0 12.66 27.78 17.62 23.12 63.2 32.35 25.04 33.46 27.11 0.75 6.67 25 0 13.92 28.79 14.29 16.58 66.4 28.68 26.52 33.46 26.44 1 3.33 40 10 12.66 34.85 16.67 24.12 62 29.78 26.52 34.25 29.08 1.25 6.67 45 16.67 17.72 30.81 15.71 20.6 65 30.51 29.63 35.79 31.67 1.5 3.33 37.5 6.67 12.66 30.3 11.9 19.1 63 29.41 28.74 33.9 28.24 1.75 6.67 30 10 25.32 29.29 22.38 26.63 67.2 31.25 30.52 37.58 31.93 2 6.67 27.5 6.67 13.92 30.3 13.81 20.6 64 32.35 29.48 34.95 28.66 2.25 6.67 40 13.33 21.52 33.84 17.62 17.59 64.6 29.78 28.15 35.39 31.37 2.5 6.67 45 13.33 18.99 31.31 17.14 17.09 65 29.04 29.19 35.52 31.89 2.75 6.67 45 13.33 12.66 29.8 16.19 24.62 65.4 29.04 29.33 35.87 31.82 3 10 45 10 16.46 30.3 17.14 22.61 63.2 31.99 29.48 35.92 32.38 3.25 3.33 50 10 15.19 29.29 15.24 15.08 63.2 29.04 28.74 34.47 31.19 3.5 6.67 37.5 16.67 11.39 26.77 10 22.11 63.2 29.04 26.37 33.42 30.16 3.75 3.33 35 13.33 10.13 28.79 14.76 14.57 64.4 29.78 28.44 34.25 29.87 4 10 47.5 13.33 13.92 27.27 12.86 20.1 66.8 31.25 29.19 35.83 32.75 4.25 6.67 40 6.67 8.86 28.28 11.43 22.11 65 29.04 29.33 34.91 30.49 4.5 6.67 45 16.67 7.59 27.27 13.33 21.11 65.4 32.72 29.33 35.61 32.1 4.75 10 55 10 13.92 31.31 12.38 20.1 64.6 31.62 28.3 35.57 33.38 5 6.67 42.5 10 6.33 28.28 12.86 20.6 66.4 32.35 28.15 35.3 31.29 Low 0.25 0 27.5 0 7.59 13.13 11.9 7.04 48.8 22.79 18.81 23.57 18.51 0.5 3.33 32.5 3.33 5.06 18.18 10.95 6.53 50 24.26 19.11 24.49 19.93 0.75 0 27.5 3.33 10.13 12.63 14.76 6.03 50.4 18.75 20.15 24 18.7 1 0 30 0 8.86 15.66 12.38 3.02 52.2 25.37 20.59 25.19 19.64 1.25 0 27.5 3.33 13.92 17.17 14.76 8.04 55 25 21.48 26.85 20.75 1.5 0 25 0 8.86 19.7 15.71 7.04 52.4 25 21.63 26.37 20.12 1.75 3.33 25 3.33 12.66 15.15 13.81 5.03 55.2 23.9 22.81 26.76 20.93 2 0 30 3.33 13.92 18.69 14.76 8.54 54.2 22.06 21.93 26.81 21.4 2.25 0 25 0 15.19 15.66 15.71 8.04 51.8 22.06 22.07 26.19 21.05 2.5 0 22.5 0 16.46 16.16 13.33 8.04 55.6 23.16 22.96 27.07 20.57 2.75 3.33 27.5 0 24.05 16.16 14.76 10.05 56.2 25.74 23.11 28.47 23.54 3 3.33 30 10 12.66 17.17 19.52 7.54 56 25.37 24 28.6 23.42 3.25 0 27.5 3.33 16.46 18.18 15.24 8.04 55.8 20.96 22.81 27.51 22.39 3.5 3.33 27.5 0 16.46 23.74 17.62 10.55 56.4 23.53 21.33 28.43 23.5 3.75 3.33 25 6.67 18.99 22.73 17.14 11.56 56.8 24.63 22.07 29.04 24.63 4 0 27.5 6.67 15.19 27.27 18.1 10.05 56.6 22.43 23.11 29.3 24.63 4.25 3.33 20 3.33 13.92 22.73 16.19 8.04 58 23.53 23.26 28.82 23.12 4.5 6.67 27.5 0 11.39 24.24 15.71 10.55 58.6 22.43 22.96 29.04 23.64 4.75 3.33 25 0 13.92 25.76 17.14 8.04 57.8 20.59 22.52 28.56 23.1 5 3.33 27.5 3.33 15.19 21.21 18.1 8.54 56.4 18.75 24.3 28.43 23.33 Table 6: Qwen-2.5-7B results across training checkpoints under high- and low-likelihood data selection. Cyan marks the best value within the early stage (epochs 0.25–1), yellow marks the best value across epochs 0.25–5 within each setting, and green marks cells that satisfy both. Dashed rules separate the earliest stage from later training. Setting Epoch AIME AMC CNM GK GPQA GSM KY MATH MNV OB W. Avg Avg High 0.25 10.00 65.00 26.67 49.37 37.37 45.71 46.23 77.80 43.01 39.70 49.80 44.09 0.5 13.33 62.50 30.00 55.70 33.84 48.10 35.68 78.00 42.65 39.70 49.04 43.95 0.75 13.33 57.50 13.33 51.90 39.39 42.86 39.70 75.80 42.65 39.11 48.28 41.56 1 10.00 62.50 20.00 49.37 33.84 47.14 38.69 78.80 43.01 42.37 49.84 42.57 1.25 16.67 62.50 23.33 49.37 32.83 45.24 39.70 77.80 41.54 41.93 49.26 43.09 1.5 13.33 52.50 16.67 50.63 31.31 47.62 39.20 75.20 40.07 40.15 47.74 40.67 1.75 10.00 62.50 23.33 54.43 26.77 47.62 40.20 78.60 41.54 41.78 49.22 42.68 2 13.33 55.00 26.67 50.63 30.30 47.14 41.71 77.40 43.01 42.22 49.48 42.74 2.25 20.00 50.00 26.67 51.90 32.83 50.95 41.21 78.60 44.12 41.04 50.11 43.73 2.5 10.00 57.50 20.00 50.63 35.35 46.19 42.21 78.60 43.38 42.81 50.29 42.67 2.75 10.00 67.50 33.33 50.63 38.38 47.14 41.21 77.40 43.01 41.48 50.20 45.01 3 13.33 65.00 36.67 54.43 34.85 44.76 42.21 78.60 42.65 41.63 50.20 45.41 3.25 13.33 55.00 30.00 54.43 33.33 42.86 42.71 78.60 41.54 41.04 49.35 43.28 3.5 13.33 57.50 30.00 54.43 36.36 43.33 46.73 78.00 43.01 43.26 50.78 44.60 3.75 16.67 57.50 30.00 51.90 36.36 48.10 44.72 77.80 41.18 42.37 50.47 44.66 4 10.00 52.50 33.33 53.16 34.85 44.29 44.22 78.00 42.65 42.07 49.98 43.51 4.25 13.33 50.00 33.33 54.43 32.83 44.76 45.23 78.20 42.65 41.63 49.89 43.64 4.5 16.67 47.50 30.00 53.16 35.86 45.24 43.72 77.00 42.65 42.22 49.89 43.40 4.75 16.67 47.50 30.00 55.70 36.87 46.19 40.70 78.40 43.38 41.78 50.16 43.72 5 13.33 50.00 23.33 53.16 38.89 47.14 46.23 79.00 41.54 42.52 50.87 43.52 Low 0.25 13.33 45.00 16.67 45.57 23.74 40.95 30.65 68.00 30.51 34.22 40.80 34.86 0.5 13.33 45.00 6.67 55.70 23.74 42.86 31.16 71.40 33.82 37.48 43.39 36.12 0.75 6.67 47.50 16.67 56.96 27.78 41.90 27.14 69.60 31.25 37.19 42.63 36.26 1 10.00 47.50 20.00 56.96 23.74 44.29 28.14 71.80 30.15 37.93 43.26 37.05 1.25 16.67 52.50 30.00 62.03 33.33 46.67 37.69 74.40 37.13 40.30 47.83 43.07 1.5 13.33 45.00 13.33 56.96 29.29 45.24 34.17 72.00 30.88 36.74 44.07 37.70 1.75 16.67 62.50 26.67 56.96 25.25 49.05 34.17 74.80 33.82 40.00 46.57 41.99 2 20.00 50.00 20.00 58.23 38.89 49.05 32.16 78.60 34.93 44.00 49.57 42.59 2.25 20.00 62.50 23.33 62.03 38.38 51.90 33.17 78.20 37.87 42.81 50.20 45.02 2.5 20.00 62.50 23.33 55.70 35.86 55.71 32.66 78.80 36.03 42.96 50.02 44.36 2.75 13.33 60.00 20.00 55.70 35.35 49.52 32.66 77.60 33.82 40.44 47.92 41.84 3 20.00 50.00 20.00 50.63 40.91 54.29 35.18 77.80 34.19 40.74 48.99 42.37 3.25 16.67 57.50 23.33 50.63 35.86 53.81 33.67 79.60 36.03 42.52 49.66 42.96 3.5 20.00 55.00 30.00 58.23 38.89 52.86 33.67 78.40 35.66 42.07 49.75 44.48 3.75 20.00 47.50 30.00 54.43 39.90 55.24 34.67 80.60 36.40 43.11 50.78 44.18 4 20.00 62.50 33.33 58.23 37.37 51.90 33.67 79.20 35.29 43.26 50.20 45.48 4.25 16.67 60.00 20.00 56.96 39.90 50.48 34.17 79.20 34.56 44.15 50.20 43.61 4.5 26.67 60.00 30.00 53.16 44.95 52.38 32.66 79.20 34.19 43.11 50.47 45.63 4.75 23.33 65.00 26.67 55.70 43.43 53.81 36.18 80.00 33.46 44.30 51.32 46.19 5 13.33 55.00 23.33 51.90 42.42 50.95 36.68 79.40 34.93 44.00 50.47 43.20 Table 7: Qwen3-8B results across training epochs under high- and low-likelihood data selection. Cyan marks the best value within the early stage (epochs 0.25–1), yellow marks the best value across epochs 0.25–5 within each setting, and green marks cells that satisfy both. Dashed rules separate the earliest stage from later training. Setting Epoch AIME24 AMC CNM GK GPQA GSM KY MATH MNV OB W. Avg Avg High 0.25 36.67 80 50 64.56 78.79 68.10 71.86 91.6 58.09 61.93 70.98 66.16 0.5 26.67 87.5 46.67 68.35 72.73 66.67 67.34 90.0 58.46 63.56 70.17 64.79 0.75 33.33 85 46.67 69.62 75.25 68.10 67.84 90.0 55.51 64.30 70.53 65.56 1 26.67 82.5 56.67 74.68 76.77 67.62 62.31 91.0 56.99 62.67 70.22 65.79 1.25 36.67 85 43.33 73.42 72.73 68.57 63.82 92.0 55.88 62.52 70.09 65.39 1.5 33.33 87.5 50.00 73.42 76.77 70.48 63.32 91.2 57.72 62.67 70.76 66.64 1.75 26.67 85 43.33 69.62 74.75 72.86 67.84 90.0 55.88 63.70 70.67 64.97 2 33.33 82.5 53.33 70.89 70.71 70.00 67.34 91.6 54.41 65.19 70.85 65.93 2.25 33.33 80 53.33 68.35 72.73 65.24 65.33 91.8 55.88 63.70 70.04 64.97 2.5 36.67 82.5 43.33 67.09 77.27 65.24 65.83 90.8 55.51 64.15 70.26 64.84 2.75 30.00 82.5 53.33 68.35 76.26 67.14 63.32 91.4 55.88 62.96 70.04 65.12 3 26.67 90.0 43.33 69.62 72.22 68.10 59.30 91.8 55.15 62.52 69.28 63.87 3.25 40.00 85.0 43.33 67.09 75.25 69.05 62.81 91.4 56.62 64.59 70.67 65.51 3.5 30.00 90.0 40.00 67.09 76.77 67.14 64.32 90.4 56.62 64.74 70.49 64.71 3.75 30.00 82.5 50.00 65.82 71.21 67.14 66.33 91.8 55.51 62.96 69.77 64.33 4 26.67 90.0 43.33 65.82 75.76 68.57 66.33 90.6 55.15 62.81 69.95 64.50 4.25 33.33 85.0 43.33 67.09 75.25 67.62 65.33 90.6 54.41 63.85 70.00 64.58 4.5 26.67 82.5 43.33 65.82 77.27 66.19 62.81 90.6 53.31 63.11 69.28 63.16 4.75 26.67 87.5 43.33 68.35 75.25 68.10 62.81 91.4 55.88 64.00 70.22 64.33 5 36.67 85.0 40.00 67.09 73.23 68.10 64.82 90.4 55.51 62.52 69.50 64.33 Low 0.25 20.00 77.5 23.33 51.9 75.25 63.81 64.82 88.20 58.19 54.67 66.65 57.72 0.5 26.67 80.00 40 62.03 71.72 65.24 65.33 90.00 58.46 57.19 74.65 62.62 0.75 13.33 60.00 50.00 67.09 71.21 63.33 57.29 87.20 58.09 56.00 71.54 58.68 1 20.00 67.5 50 63.29 73.23 64.76 62.81 89.20 57.35 58.96 73.87 61.35 1.25 26.67 77.5 50 67.09 75.76 66.67 58.29 90.20 55.15 58.96 74.96 64.02 1.5 26.67 82.5 43.33 67.09 76.77 68.10 66.83 90.60 59.93 60.59 77.14 65.49 1.75 26.67 75 40 69.62 78.28 66.19 63.32 90.60 58.09 62.52 76.05 63.71 2 30 72.5 53.33 68.35 76.26 68.57 63.32 92.20 59.93 62.96 77.14 65.69 2.25 33.33 82.5 53.33 70.89 75.76 70.48 65.33 91.20 57.35 62.81 77.68 67.85 2.5 40.00 82.5 63.33 69.62 76.77 71.43 67.34 90.4 58.82 62.37 72.82 69.31 2.75 40.00 82.5 53.33 65.82 76.26 69.52 64.82 91.8 58.46 61.78 72.16 67.32 3 43.33 85 60 68.35 76.26 71.43 61.81 90.8 60.29 60.30 71.60 68.59 3.25 36.67 82.5 53.33 69.62 79.29 69.05 65.83 91.4 59.93 64.44 73.43 68.01 3.5 40.00 87.5 60.00 67.09 77.27 72.38 64.32 92.4 58.46 62.37 78.77 70.12 3.75 43.33 90 63.33 67.09 75.25 71.43 63.32 91.8 58.82 63.85 78.15 70.69 4 36.67 82.5 53.33 68.35 79.80 70.48 65.33 91.2 58.09 61.48 78.23 68.46 4.25 26.67 82.5 60 67.09 74.24 70.00 69.85 92.2 58.46 63.56 78.23 67.82 4.5 36.67 92.5 56.67 65.82 76.77 70.95 63.82 91.0 61.40 63.56 77.76 69.27 4.75 36.67 82.5 60.00 64.56 75.25 69.05 64.82 92.8 58.82 64.59 77.76 68.21 5 30 85 63.33 67.09 78.28 68.10 67.34 93.0 59.19 62.07 78.69 69.02 Table 8: Llama-3.2-3B results across training epochs under high- and low-likelihood data selection. Cyan marks the best value within the early stage (epochs 0.25–1), yellow marks the best value across epochs 0.25–5 within each setting, and green marks cells that satisfy both. Dashed rules separate the earliest stage from later training. Setting Epoch AIME24 AMC CNM GK GPQA GSM KY MATH MNV OB W. Avg Avg High 0.25 3.75 23.12 5.00 3.48 18.75 1.43 1.57 43.95 17.10 14.67 18.95 13.28 0.5 5.83 17.50 4.17 1.90 20.08 2.86 1.38 41.52 18.38 13.85 18.41 12.75 0.75 6.25 21.25 3.75 3.32 20.20 2.86 1.07 44.35 17.97 15.43 19.57 13.64 1 6.25 18.44 2.92 3.01 23.42 2.74 3.14 44.12 17.78 14.33 19.55 13.62 1.25 7.50 26.25 4.17 3.16 20.83 2.08 1.76 46.65 18.98 16.15 20.58 14.75 1.5 6.25 24.38 3.75 6.65 25.06 4.29 3.77 46.38 18.29 16.35 21.32 15.52 1.75 8.33 29.06 2.92 5.06 22.60 4.82 2.58 46.95 20.04 16.46 21.47 15.88 2 7.08 27.19 3.75 5.38 25.69 4.94 3.64 46.17 19.81 16.72 21.70 16.04 2.25 5.00 23.75 2.92 5.85 25.06 3.63 3.71 45.85 18.93 16.54 21.20 15.12 2.5 6.67 27.81 4.58 7.28 26.26 3.45 3.33 47.30 19.26 17.19 21.99 16.31 2.75 7.92 28.12 1.25 5.54 25.82 4.29 4.65 47.73 20.54 16.15 22.00 16.20 3 8.33 25.31 5.42 5.38 24.87 3.10 3.52 47.60 19.03 16.35 21.56 15.89 3.25 4.58 27.50 5.00 3.48 21.78 1.73 2.45 46.40 19.81 17.22 21.06 14.99 3.5 5.42 27.50 5.00 3.96 24.37 3.93 2.70 47.93 20.08 16.81 21.80 15.77 3.75 7.08 26.25 2.92 3.64 24.56 1.85 3.64 46.40 19.44 17.56 21.47 15.33 4 7.92 28.75 4.17 4.75 25.69 3.63 4.08 47.60 18.89 17.17 21.98 16.26 4.25 7.92 26.25 5.00 3.48 23.74 2.50 2.95 48.80 20.27 16.93 21.88 15.78 4.5 6.25 27.19 2.50 3.16 21.53 2.50 2.83 47.70 20.77 16.61 21.34 15.10 4.75 7.92 25.31 3.33 3.16 23.86 3.04 2.89 47.35 19.94 17.30 21.64 15.41 5 6.25 25.31 5.00 3.96 22.85 2.56 2.26 47.35 19.21 17.26 21.37 15.20 Low 0.25 2.08 15.31 2.50 1.27 16.41 1.01 0.57 37.52 16.45 12.15 16.06 10.53 0.5 1.67 18.44 0.83 3.16 15.28 1.85 1.51 42.80 16.68 12.70 17.59 11.49 0.75 3.33 15.31 0.83 1.74 15.34 0.30 0.75 40.20 16.77 11.54 16.38 10.61 1 1.25 18.44 1.25 2.53 15.59 1.13 1.13 44.47 17.37 13.22 18.11 11.64 1.25 2.92 18.12 2.08 2.53 18.56 1.19 1.57 45.02 18.57 14.30 19.04 12.49 1.5 4.17 22.50 2.08 3.48 17.87 0.95 1.13 47.10 20.27 15.02 19.94 13.46 1.75 4.58 17.19 2.08 1.74 16.73 0.54 1.01 44.40 17.83 13.46 18.27 11.96 2 4.17 17.19 2.50 3.32 16.73 0.89 0.57 47.35 19.21 15.57 19.78 12.75 2.25 5.00 21.25 2.08 4.75 19.70 0.95 1.07 47.85 19.99 16.39 20.68 13.90 2.5 4.17 20.00 2.08 2.22 17.99 0.95 1.01 46.08 19.44 15.72 19.73 12.97 2.75 3.75 19.38 2.08 3.01 17.61 1.73 1.32 47.17 20.82 15.91 20.28 13.28 3 5.42 24.38 0.00 4.59 19.26 1.55 1.57 46.88 20.68 16.11 20.55 14.04 3.25 6.25 18.75 0.83 4.27 17.11 1.55 1.01 46.90 20.13 16.04 20.14 13.28 3.5 5.83 20.00 2.08 7.12 19.95 1.55 0.63 46.88 19.03 16.11 20.37 13.92 3.75 6.25 22.81 0.83 4.11 19.63 1.90 0.82 47.40 20.45 16.04 20.59 14.03 4 7.50 22.81 2.92 4.59 17.93 1.79 1.07 46.88 19.44 15.98 20.26 14.09 4.25 5.42 20.00 3.75 4.27 19.89 1.43 1.01 48.02 18.52 15.96 20.45 13.83 4.5 7.08 19.38 1.67 4.59 20.52 1.13 0.50 48.12 19.07 16.19 20.59 13.82 4.75 6.25 20.94 3.75 5.54 19.38 1.25 1.38 48.08 19.53 16.63 20.84 14.27 5 6.25 20.62 1.67 4.91 17.93 1.19 1.19 47.33 19.58 15.52 20.13 13.62 Table 9: Llama-3.1-8B results across training epochs under high- and low-likelihood data selection. Cyan marks the best value within the early stage (epochs 0.25–1), yellow marks the best value across epochs 0.25–5 within each setting, and green marks cells that satisfy both. Dashed rules separate the earliest stage from later training. Setting Epoch AIME24 AMC CNM GK GPQA GSM KY MATH MNV OB W. Avg Avg High 0.25 4.58 19.38 2.92 3.48 28.85 6.61 1.38 43.80 23.81 14.74 21.04 14.95 0.5 2.92 17.19 1.25 5.38 29.10 10.54 4.65 43.85 21.97 13.74 21.19 15.06 0.75 2.92 17.19 3.75 3.32 27.65 5.60 3.08 43.08 21.28 13.46 20.07 14.13 1 2.92 20.00 1.67 5.54 33.40 7.38 3.64 43.15 22.15 14.65 21.38 15.45 1.25 5.00 21.56 4.17 9.18 36.24 12.20 5.09 46.50 23.58 17.04 24.08 18.05 1.5 4.58 18.12 2.92 6.33 36.81 11.19 5.28 48.65 25.00 15.63 24.10 17.45 1.75 2.50 19.69 1.67 7.12 35.73 10.18 5.90 47.90 22.61 16.35 23.73 16.97 2 5.42 22.81 3.33 7.59 35.48 10.60 5.15 49.73 23.16 17.93 24.77 18.12 2.25 5.00 19.06 2.92 12.34 35.04 13.39 7.04 48.20 22.98 16.76 24.54 18.27 2.5 5.83 22.19 2.50 13.13 31.82 15.48 6.28 49.40 23.12 18.35 25.24 18.81 2.75 5.42 25.31 1.25 11.71 35.16 14.94 6.72 48.30 23.71 17.57 25.10 19.01 3 5.83 23.75 2.92 12.66 35.86 13.39 5.97 51.25 22.52 19.24 26.00 19.34 3.25 3.33 18.44 3.75 15.03 34.72 15.42 6.91 50.10 25.09 18.20 25.88 19.10 3.5 8.33 23.12 2.92 14.24 35.48 16.19 5.72 50.48 23.39 18.57 26.01 19.84 3.75 5.42 22.19 5.00 15.03 38.57 14.70 6.09 50.65 24.08 19.24 26.51 20.10 4 4.58 23.44 6.25 14.40 35.80 15.30 6.97 50.40 23.81 18.65 26.13 19.96 4.25 5.42 23.12 6.67 14.87 34.91 14.40 6.34 50.50 23.16 18.72 25.91 19.81 4.5 5.83 24.06 5.83 13.61 36.81 13.04 5.46 51.28 24.63 18.44 26.10 19.90 4.75 4.58 22.19 5.42 13.92 36.81 14.76 6.28 51.25 23.30 18.67 26.19 19.72 5 5.00 24.06 3.75 14.72 35.98 15.36 5.84 50.98 24.72 18.24 26.16 19.87 Low 0.25 2.92 14.37 0.83 6.33 26.64 8.81 3.39 42.62 23.71 14.50 20.84 14.41 0.5 1.25 11.88 1.67 6.96 26.33 9.58 4.65 41.70 22.06 13.54 20.26 13.96 0.75 2.08 14.06 2.50 7.59 30.05 9.29 4.02 43.38 22.33 14.46 21.28 14.98 1 1.67 17.19 1.67 6.49 30.43 8.63 2.51 46.80 22.89 15.33 22.22 15.36 1.25 3.75 19.38 3.75 13.92 33.96 13.04 6.22 48.62 25.92 16.33 24.71 18.49 1.5 4.17 20.94 4.17 11.71 33.08 13.45 4.96 48.98 25.41 17.81 24.99 18.47 1.75 3.33 21.88 3.33 9.34 32.32 10.24 4.27 48.08 24.40 16.41 23.72 17.36 2 3.33 21.88 1.25 6.96 30.81 10.42 3.64 50.95 25.41 17.11 24.41 17.18 2.25 2.08 21.25 4.17 11.23 35.29 13.21 3.89 50.12 25.23 18.56 25.49 18.50 2.5 5.42 21.25 1.67 13.61 36.30 15.48 6.47 51.25 25.23 19.26 26.58 19.59 2.75 3.75 23.75 2.50 13.13 36.17 14.40 4.90 51.80 24.54 17.83 25.95 19.28 3 5.83 27.50 2.50 12.50 35.98 16.19 6.16 51.42 25.51 18.54 26.53 20.21 3.25 4.58 25.00 4.17 13.61 35.23 16.85 5.53 51.32 24.63 19.04 26.49 20.00 3.5 7.50 21.88 2.92 14.56 37.18 17.02 4.77 52.52 26.15 18.59 26.94 20.31 3.75 5.42 23.75 5.00 17.41 35.23 17.50 5.21 52.95 26.19 19.52 27.36 20.82 4 5.00 23.12 4.58 14.56 34.72 17.14 5.53 52.45 26.33 18.69 26.84 20.21 4.25 2.50 22.50 2.50 19.62 35.23 17.80 4.71 51.78 26.84 19.70 27.20 20.32 4.5 4.17 27.81 6.25 13.77 37.18 17.14 5.09 52.80 25.46 19.63 27.35 20.93 4.75 4.17 25.94 3.75 17.56 33.52 17.02 5.21 52.92 25.23 19.56 27.07 20.49 5 4.17 22.19 4.58 15.19 34.28 18.15 5.34 51.30 25.97 20.15 27.02 20.13