REVIEW 4 major objections 7 minor 1 cited by
LLM Data Selection and Utilization via Dynamic Bi-level Optimization
T0 review · 4 major / 7 minor · reviewed 2026-08-06 · deepseek-v4-flash
Pith's one-line read A per-batch data weighting model, learned end-to-end, improves LLM pretraining beyond what random selection alone achieves.
desk verdict Overstated headline claim obscures a plausible but thinly supported result; DWM reliably helps random/DSIR data at 370M, but QuRating still beats it and the meta-learning core is uncited. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The working object is the Data Weighting Model (DWM), a small Transformer that reads all samples in a micro-batch and outputs a weight per sample; these weights rescale the per-sample autoregressive losses before backpropagation. The carrying identity is the chained bi-level update in Eq. (6), obtained by unrolling one step of LLM training and differentiating the validation reward through the unrolled parameters. What makes it work, per the paper, is the stage-wise alternation that re-trains the weighting model on the current LLM state, so the weights reflect the model's present preferences rather than a fixed quality score.
What would settle it
Run the same 370M random-data setup but replace the LAMBADA reward with an i.i.d. held-out split of SlimPajama; if the two-shot average gain over RANDOM (1.3 points in Table 2) disappears or reverses, the reported benefit comes from the proxy task rather than from dynamic weighting per se.
Extended reading notes
Core claim
The central claim is that per-batch data weighting can be learned by treating model training as the lower level of a bi-level optimization: $\theta^*(\theta_w) = \arg\min_\theta \sum_i \omega_i \mathcal{L}_{\mathrm{train},i}(\theta)$ with weights $\omega = \theta_w(X_1,\dots,X_{bs})$ produced by the weighting model, while the upper level maximizes a validation reward $R_{\mathrm{val}}(\theta^*(\theta_w))$, implemented as accuracy on LAMBADA. Replacing the inner argmin by a single gradient step allows the weighting model to be updated by the chain rule, and repeating this alternation across five stages lets the weights track the model's changing data preferences. The paper reports that this DWM improves average two-shot accuracy from 45.1 to 46.4 on nine tasks for a 370M model trained on randomly selected data, that it also improves DSIR- and QuRating-selected data on average, and that a weighting model learned at 370M transfers to a 1.3B model with roughly 9% additional training FLOPs.
Load-bearing premise
Everything hinges on the assumption that one gradient step of LLM training, scored by LAMBADA accuracy, is a faithful guide to how batch weights will affect the final model on nine other downstream tasks.
Editorial extensions
If this is right
- Models trained on randomly selected data can match or exceed the accuracy of some curated selections when every batch is reweighted by DWM.
- The same weighting model trained on a 370M model transfers to a 1.3B model without retraining, adding roughly 9% to training FLOPs.
- Applying DWM to DSIR-selected and QuRating-selected data yields further average gains, and the gains are complementary to the selection method itself.
- Dynamic, stage-wise reweighting outperforms using either a fixed early-stage or a fixed late-stage weighting model.
- The preferred data properties shift as training progresses: later stages favor expertise- and reasoning-heavy content over polished general-purpose writing.
Reading between the lines
- If DWM works as claimed, the notion of a fixed 'high-quality' dataset is incomplete: the usefulness of a token appears to depend on training stage and batch context, so future selection systems could output stage-dependent weights rather than static rankings.
- The paper's reward is a single task (LAMBADA); a natural extension the authors do not pursue is to make the reward a combination of downstream tasks, which would allow DWM to steer pretraining toward specific capabilities but would require more validation compute and a multi-objective outer loop.
- Because Sec. 5.3 reports that the stage-2 weighting model can temporarily lower performance relative to uniform weighting, a production deployment would need a checkpoint-selection or early-stopping rule for the weighting model itself, something the paper leaves as future work.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes a Data Weighting Model (DWM) that assigns per-sample weights within each batch during LLM pre-training, learned through a dynamic bi-level optimization in which the outer objective is validation accuracy on LAMBADA after a one-step inner update of the LLM. The method is evaluated by pre-training 370M and 1.3B Llama-2-style models on 30B tokens selected from SlimPajama, comparing random, DSIR, and QuRating baselines, and by transferring the learned weighting model across selection methods and model sizes. The paper also reports ablations on the number of stages and on the validation task, and analyzes how learned weights correlate with QuRating-style data-quality dimensions over training stages.
Significance. The central idea—optimizing per-batch data weights against a validation signal during pre-training rather than relying on static, training-agnostic selection—is timely and potentially valuable. The paper's ablations (stage count, validation task, static vs. dynamic weighting) and the analysis of evolving data preferences are thoughtful and go beyond a single accuracy table. The transfer experiments to a 1.3B model and to other selection methods are a good-faith test of generality. However, the current evidence is thin: all results are single runs, average gains are small with several task-level regressions, and the headline claim in Section 1 that RANDOM+DWM outperforms carefully selected data is directly contradicted by Tables 3 and 8. The bi-level derivation also lacks the differentiable surrogate needed to backpropagate through an accuracy reward. These issues are fixable, but the manuscript as written overclaims and would need substantial revision before the contribution is convincingly supported.
major comments (4)
- [Section 1 / Tables 3 and 8] The claim in Section 1 that 'the model trained with DWM and randomly selected data outperforms both models trained with randomly-selected data and those with carefully selected data' is contradicted by the paper's own numbers. In the two-shot 370M comparison (Table 3), RANDOM+DWM averages 46.4 while QURATING averages 48.5, with QuRating winning 6 of 9 tasks (ARC-C 27.9 vs 24.7, ARC-E 56.6 vs 46.8, BoolQ 58.6 vs 56.6, HellaSwag 38.1 vs 36.5, OBQA 32.0 vs 28.2, SciQ 82.3 vs 80.5). The zero-shot results (Table 8) show the same pattern (45.0 vs 46.6). The supported claim is the more modest one in the abstract: DWM improves over randomly selected data, and it can be transferred to other selection methods with mixed task-wise outcomes. The introduction and Section 5.2 should be reworded to match the evidence, and the conclusion's 'competitive results' wording is appropriate and should be used consistently.
- [Section 3.4, Eqs. (4)-(6)] The gradient update in Eq. (6) differentiates the validation reward Rval, which is described as accuracy (Section 3.2). Accuracy is piecewise-constant, so the gradient is zero almost everywhere and undefined at jump points; no surrogate (e.g., soft accuracy, temperature-scaled logits, or a loss-based reward) is specified. This makes the core weighting-model update ill-posed as written. Moreover, Eq. (4) replaces the inner argmin in Eq. (1) with a single gradient step, an approximation that is not justified; the instability acknowledged in Section 5.3 (stage-2 weighting model causing a performance drop, visible in Table 1 at Stage 3 where RANDOM+DWM is 43.2 vs RANDOM 43.6) may stem from this approximation. Please specify the surrogate, provide the exact form of Rval used in the experiments, and report sensitivity to the unrolling step and the weighting-model learning rate η.
- [Section 4 / Tables 1-4] All reported results appear to come from single training runs with no variance estimates. The average improvements are small (e.g., +1.3 two-shot and +1.0 zero-shot at 370M for RANDOM vs RANDOM+DWM) and many individual tasks regress (Table 4 BoolQ 56.0→51.1 for RANDOM+DWM at 1.3B; Table 2 ARC-C 25.5→24.7; Table 1 Stage 3 zero-shot 43.6→43.2). Without multiple seeds or error bars, it is impossible to tell whether these differences are robust or within run-to-run noise, which is especially important for the transfer claims. Please provide at least three seeds for the key comparisons (RANDOM vs RANDOM+DWM at both scales, and QURATING vs QURATING+DWM) and report mean and standard deviation or confidence intervals.
- [Section 5.2 / Tables 4 and 9] The text claims 'consistent performance improvements' when DWM is transferred to a 1.3B model. However, Table 4 (two-shot) shows DSIR 48.7 vs DSIR+DWM 48.7, and Table 9 (zero-shot) shows DSIR 46.8 vs DSIR+DWM 46.8, i.e., no improvement; the gain over QuRating is 0.7 points (two-shot) and 1.4 points (zero-shot), within the range that could be noise given single runs. The interpretation in Section 5.2 about 'training saturation' and model capacity is speculative and not supported by additional experiments. Please either temper the transferability claim to match the data or provide additional evidence (e.g., more seeds, more downstream tasks, or a direct test of the saturation hypothesis).
minor comments (7)
- [Section 3.3] The notation for the weighting model is inconsistent (θw in Eq. (2) is used as a function producing a vector of weights, while elsewhere θw is a parameter vector; Eq. (3) and Section 3.4 sometimes use θω). Please standardize the notation and explicitly define how ω is obtained from θw and the batch (e.g., softmax over scores, with or without normalization).
- [Section 4] The description of the DWM architecture ('one attention block and two linear layers') is insufficient for reproducibility; please provide the full architecture, the input representation of a batch, and the weight normalization/constraint used in the experiments.
- [Section 5.2] 'Tale 3' should be 'Table 3'; throughout the paper 'Specially' should be 'Specifically' and 'preferred'/'unpreferred' should be used consistently.
- [Section 5.3] 'pro-collected' should be 'pre-collected', and 'prosperity' should be 'properties' in the analysis of Figure 3. The phrase 'as shown in the bottom-left corner of the Figure 3' is confusing because Figure 3 is a bar chart, not a performance curve; it likely refers to the stage-wise accuracy plots in Figure 2.
- [Tables 1-4] The table headers contain formatting artifacts (e.g., 'B OOL Q', 'L OQI QA') that should be cleaned up for readability.
- [Equation (6)] The displayed equation has a spacing issue with '∂θ ∗' and the chain rule would be clearer if the intermediate variables were named explicitly; please also check the consistent use of θw vs θω in this equation.
- [References] Reference Langley (2000) appears in the reference list but is never cited in the text; either cite it or remove it.
Circularity Check
No significant circularity: DWM's LAMBADA validation objective and the reported downstream benchmarks are disjoint, and the bi-level update is derived from first principles in Eqs. (4)-(6).
full rationale
The derivation chain is self-contained. The weighting model is trained in Eqs. (4)-(6) by one-step unrolling of the training loss (Eq. 3) and optimizing a reward Rval on LAMBADA, while the trained model is evaluated on nine downstream tasks (ARC, SciQ, LogiQA, BoolQ, OBQA, HellaSwag, PIQA, WinoGrande) that do not include LAMBADA. Thus the reported improvements are not forced by the objective by construction. No load-bearing self-citation is present: the only same-author citation in the background (Zhou et al., 2024) supports a general statement about data diversity, not the method's core claim, and no uniqueness theorem or ansatz is imported from prior work by these authors. The paper's own Table 5 ablation shows that the choice of validation task changes results, and Sec. 5.3 candidly admits that the stage-2 weighting model can cause a performance drop and leaves better weighting models to future work; these are empirical limitations rather than circular reductions. The Sec. 1 claim that RANDOM+DWM outperforms 'carefully selected data' is contradicted by Table 3 (QuRating 48.5 vs. RANDOM+DWM 46.4 at 370M two-shot), but that is an overstatement/correctness issue, not circularity, because the comparison is measured independently rather than derived from the method's definitions. Therefore the circularity score is 0.
Assumptions & free parameters
free parameters (5)
- Number of training stages T =
5
- Micro-batch size bs =
8
- Validation task for Rval =
LAMBADA
- Weighting model learning rate eta =
not reported
- Weighting model initialization schedule =
partial init from trained model at stage 2
assumptions (4)
- domain assumption LAMBADA accuracy is a representative proxy for the desired downstream generalization.
- domain assumption One-step inner-loop unrolling approximates the true bi-level argmin.
- domain assumption The learned weighting policy transfers across model sizes and data selection methods.
- domain assumption Weights assigned to one micro-batch generalize to other batches.
invented entities (1)
-
Data Weighting Model (DWM)
Cite this review
Pith. "Pith review of LLM Data Selection and Utilization via Dynamic Bi-level Optimization." pith.science (2026). https://pith.science/paper/75QJ73FW
@misc{pith2026250716178,
author = {Pith},
title = {Pith review of: LLM Data Selection and Utilization via Dynamic Bi-level Optimization},
year = {2026},
howpublished = {\url{https://pith.science/paper/75QJ73FW}},
note = {Machine review of arXiv:2507.16178}
}
read the original abstract
While large-scale training data is fundamental for developing capable large language models (LLMs), strategically selecting high-quality data has emerged as a critical approach to enhance training efficiency and reduce computational costs. Current data selection methodologies predominantly rely on static, training-agnostic criteria, failing to account for the dynamic model training and data interactions. In this paper, we propose a new Data Weighting Model (DWM) to adjust the weight of selected data within each batch to achieve a dynamic data utilization during LLM training. Specially, to better capture the dynamic data preference of the trained model, a bi-level optimization framework is implemented to update the weighting model. Our experiments demonstrate that DWM enhances the performance of models trained with randomly-selected data, and the learned weighting model can be transferred to enhance other data selection methods and models of different sizes. Moreover, we further analyze how a model's data preferences evolve throughout training, providing new insights into the data preference of the model during training.
Figures
Figures from the paper (2 more)
Forward citations
Cited by 1 Pith paper
-
Bilevel Data Curation for LLM Fine-tuning: Offline Selection and Online Self-Refining Generation
A bilevel data-curation method for LLM fine-tuning that selects validation-aligned offline data and reweights online self-refined responses via importance ratios.
Reference graph
Works this paper leans on
-
[1]
M., Longpre, S., Lambert, N., Wang, X., Muennighoff, N., Hou, B., Pan, L., Jeong, H., et al
Albalak, A., Elazar, Y ., Xie, S. M., Longpre, S., Lambert, N., Wang, X., Muennighoff, N., Hou, B., Pan, L., Jeong, H., et al. A survey on data selection for language models. arXiv preprint arXiv:2402.16827,
-
[5]
Boolq: Exploring the surprising difficulty of natural yes/no questions
Clark, C., Lee, K., Chang, M.-W., Kwiatkowski, T., Collins, M., and Toutanova, K. Boolq: Exploring the surprising difficulty of natural yes/no questions. arXiv preprint arXiv:1905.10044,
arXiv 1905
- [7]
-
[8]
Hoffmann, J., Borgeaud, S., Mensch, A., Buchatskaya, E., Cai, T., Rutherford, E., Casas, D. d. L., Hendricks, L. A., Welbl, J., Clark, A., et al. Training compute-optimal large language models. arXiv preprint arXiv:2203.15556,
-
[10]
Crafting papers on machine learning
Langley, P. Crafting papers on machine learning. In Langley, P. (ed.),Proceedings of the 17th International Conference on Machine Learning (ICML 2000), pp. 1207–1216, Stan- ford, CA,
2000
-
[11]
Logiqa: A challenge dataset for machine reading comprehension with logical reasoning
Liu, J., Cui, L., Liu, H., Huang, D., Wang, Y ., and Zhang, Y . Logiqa: A challenge dataset for machine reading comprehension with logical reasoning. arXiv preprint arXiv:2007.08124,
arXiv 2007
-
[12]
Regmix: Data mixture as regression for language model pre-training
Liu, Q., Zheng, X., Muennighoff, N., Zeng, G., Dou, L., Pang, T., Jiang, J., and Lin, M. Regmix: Data mixture as regression for language model pre-training. arXiv preprint arXiv:2407.01492,
-
[13]
Can a suit of armor conduct electricity? a new dataset for open book question answering
Mihaylov, T., Clark, P., Khot, T., and Sabharwal, A. Can a suit of armor conduct electricity? a new dataset for open book question answering. arXiv preprint arXiv:1809.02789,
Show all 19 references
-
[14]
Touvron, H., Lavril, T., Izacard, G., Martinet, X., Lachaux, M.-A., Lacroix, T., Rozi`ere, B., Goyal, N., Hambro, E., Azhar, F., et al
URL https://huggingface.co/ datasets/cerebras/SlimPajama-627B. Touvron, H., Lavril, T., Izacard, G., Martinet, X., Lachaux, M.-A., Lacroix, T., Rozi`ere, B., Goyal, N., Hambro, E., Azhar, F., et al. Llama: Open and efficient foundation lan- guage models. arXiv preprint arXiv:2...
-
[15]
Qurating: Selecting high-quality data for training language models
Wettig, A., Gupta, A., Malik, S., and Chen, D. Qurating: Selecting high-quality data for training language models. arXiv preprint arXiv:2402.09739,
-
[16]
Mates: Model-aware data selection for efficient pretraining with data influence mod- els
Yu, Z., Das, S., and Xiong, C. Mates: Model-aware data selection for efficient pretraining with data influence mod- els. arXiv preprint arXiv:2406.06046,
-
[17]
Hellaswag: Can a machine really finish your sentence? arXiv preprint arXiv:1905.07830,
Zellers, R., Holtzman, A., Bisk, Y ., Farhadi, A., and Choi, Y . Hellaswag: Can a machine really finish your sentence? arXiv preprint arXiv:1905.07830,
1905 arXiv
-
[18]
X., Zhou, K., Li, J., Tang, T., Wang, X., Hou, Y ., Min, Y ., Zhang, B., Zhang, J., Dong, Z., et al
10 LLM Data Selection and Utilization via Dynamic Bi-level Optimization Zhao, W. X., Zhou, K., Li, J., Tang, T., Wang, X., Hou, Y ., Min, Y ., Zhang, B., Zhang, J., Dong, Z., et al. A survey of large language models. arXiv preprint arXiv:2303.18223,
-
[19]
Architecture of pre-training models with 370M and 1.3B parameters. Hyperparameter 1.3B Model Value 370M Model Value V ocabulary Size 32,000 32,000 Hidden Size 2048 1024 FFN Hidden Size 5504 2812 Number of Layers 24 24 Number of Attention Heads 16 8 Number of KV Attention Heads...
-
[2019]
Think you have solved question answering? try arc, the ai2 reasoning challenge
Clark, P., Cowhey, I., Etzioni, O., Khot, T., Sabharwal, A., Schoenick, C., and Tafjord, O. Think you have solved question answering? try arc, the ai2 reasoning challenge. arXiv preprint arXiv:1803.05457,
-
[2020]
D., Dhariwal, P., Neelakantan, A., Shyam, P., Sastry, G., Askell, A., et al
Brown, T., Mann, B., Ryder, N., Subbiah, M., Kaplan, J. D., Dhariwal, P., Neelakantan, A., Shyam, P., Sastry, G., Askell, A., et al. Language models are few-shot learners. 9 LLM Data Selection and Utilization via Dynamic Bi-level Optimization Advances in neural information pro...
1901
-
[2022]
B., Chess, B., Child, R., Gray, S., Radford, A., Wu, J., and Amodei, D
Kaplan, J., McCandlish, S., Henighan, T., Brown, T. B., Chess, B., Child, R., Gray, S., Radford, A., Wu, J., and Amodei, D. Scaling laws for neural language models. arXiv preprint arXiv:2001.08361,
2001 arXiv
-
[2023]
H., Peng, J., Zhuang, X., Zhang, C., Wu, L., Qiu, J., Zhang, W., Yuan, B., et al
Bai, T., Yang, L., Wong, Z. H., Peng, J., Zhuang, X., Zhang, C., Wu, L., Qiu, J., Zhang, W., Yuan, B., et al. Multi-agent collaborative data selection for efficient llm pretraining. arXiv preprint arXiv:2410.08102,
-
[2024]
L., and Paul, M
Ankner, Z., Blakeney, C., Sreenivasan, K., Marion, M., Leavitt, M. L., and Paul, M. Perplexed by perplexity: Perplexity-based pruning with small reference models. In ICLR 2024 Workshop on Mathematical and Empirical Understanding of Foundation Models,
2024
Reviewed August 6, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.