REVIEW 2 major objections 5 minor 42 references
Improving Task Diversity in Label Efficient Supervised Finetuning of LLMs
T0 review · 2 major / 5 minor · reviewed 2026-08-06 · deepseek-v4-flash
Pith's one-line read This paper argues that a simple inverse-confidence task allocation can beat full-dataset LLM finetuning with up to 80% fewer labels.
desk verdict A simple task-level inverse-confidence selection method with a promising MMLU result, but the confidence metric is confounded with output length and the abstract overclaims consistency. 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 load-bearing object is the task-level inverse-confidence allocation $\alpha_t \propto 1/\mathrm{conf}_t$, realized through a round-robin sampler. Task confidence is defined as $\mathrm{conf}_t = \frac{1}{|X_t|}\sum_{x \in X_t} \prod_{j=1}^m g(y_j \mid y_{<j}, x)$, the average over a task's prompts of the product of token probabilities of the base model's generated response. The allocation first gives every task a small base budget, then distributes the remainder in proportion to $1/\mathrm{conf}_t$, clamped to each task's available prompts. The round-robin pass converts fractional allocations into an integer selection and enforces that small tasks are covered before the budget moves on. This machinery does two jobs at once: it guarantees task coverage, and it shifts labels toward tasks where the base model is most uncertain.
What would settle it
Run Weighted Task Diversity with confidence normalized by response length or computed as a per-token geometric mean, holding the budget, pool, and training recipe fixed; if the 3K-label MMLU advantage over full-data training disappears, the reported gain rests on the length confound rather than on task difficulty.
Extended reading notes
Core claim
The central claim is that a pretrained model's own confidence, averaged per task, is a usable signal for deciding which instruction-tuning prompts to pay humans to label. Under Weighted Task Diversity, each task first receives a small coverage floor (the paper uses 5 examples), and the remaining budget is split across tasks proportionally to $1/\mathrm{conf}_t$, where $\mathrm{conf}_t$ is the mean over prompts in task $t$ of the product of token probabilities of the base model's generated response; a round-robin procedure then selects examples uniformly within each task. The paper reports that on Dolly, this selection at a 3K budget reaches 39.74 MMLU versus 35.33 for training on the full 13.5K pool, that on FLAN V2 a 45K budget reaches the same MMLU as all 90K examples, and that the selected subsets win more often than full-data-trained models in GPT-4-judged AlpacaEval comparisons. These results are presented as evidence that task diversity plus uncertainty weighting is at least as powerful as, and much simpler than, embedding-based diversity methods.
Load-bearing premise
The method rests on treating the base model's average per-task confidence as a trustworthy signal of which tasks most need labels; if confidence mostly reflects pretraining exposure, output length, or decoding style rather than labeling value, the inverse-confidence allocation points at the wrong tasks.
Editorial extensions
If this is right
- On the Dolly pool, a 3K-label Weighted Task Diversity selection produces an MMLU score of 39.74, higher than the 35.33 from training on all 13.5K labels, an annotation saving of roughly 80% with an accuracy gain.
- On FLAN V2, the same strategy reaches full-data MMLU performance at a 45K budget, cutting the annotation load in half at matched accuracy.
- Weighted Task Diversity and plain Task Diversity match or beat all tested baselines—random, entropy, confidence, margin, k-center, facility location, DPP, and ActiveIT—on AlpacaEval win rates at matched budgets.
- The method needs no precomputed embeddings, no clustering, no ground-truth-based quality scores, and no careful kernel hyperparameters; task labels already present in instruction datasets suffice.
- Because allocation happens before any annotation is collected, the rule is immediately usable in the one-batch active learning setting where ground-truth responses are unknown.
Reading between the lines
- Inference: an unnormalized product-of-token-probabilities confidence conflates task difficulty with response length and decoding choices; a length-normalized or per-token-geometric-mean version is a natural ablation the paper does not run.
- Inference: the same allocation rule could be recycled across active-learning rounds, recomputing task confidence after each finetune; the paper studies only the one-batch setting.
- Inference: the over-4-point gain over full-data training on Dolly might be partly a curation effect, since the 3K subset drops many redundant or lower-quality examples; a deduplicated pool would separate allocation value from data cleaning.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper studies one-batch label-efficient supervised finetuning: given an unlabeled prompt pool with task labels, select k prompts to annotate. It proposes Task Diversity (round-robin equal allocation across tasks) and Weighted Task Diversity (allocation proportional to the inverse of the base model's average task-level confidence, after a base allocation of five examples per task). Confidence for a prompt is the product of token probabilities of the base model's generated response. Experiments on LLaMA-2 7B use a 90K FLAN V2 subset and Dolly, with budgets 20K/30K/45K/90K and 3K/6K/13.5K, evaluating MMLU, BBH, and AlpacaEval. The main claims are that Weighted Task Diversity beats full-data training by about 4% MMLU on Dolly at 3K labels and that it generally matches or exceeds more complex baselines while saving up to 80% of labels.
Significance. If the empirical results hold, the contribution is practically useful: a simple, transparent task-level allocation rule using readily available task labels and one forward pass over the pool, with no embedding optimization or per-example score tuning, can produce strong SFT models under small annotation budgets. The paper reports means and standard errors over three seeds, includes multiple diversity and uncertainty baselines, and provides length-controlled AlpacaEval win rates, which are strengths. The main caveat is that the uncertainty mechanism underlying the method is not identified: the confidence score conflates output length with uncertainty, so the headline gains may be explained by a length bias rather than by a calibrated notion of task difficulty. With an added length-normalized ablation and a qualified abstract, the practical claim would be credible.
major comments (2)
- [Section 4.3] The definition of conf(x) as the product of token probabilities makes the measure exponentially sensitive to output length: for a response of length m with geometric mean per-token probability p, conf(x) = p^m. Task-level averages therefore rank tasks primarily by how long the base model's generations are, rather than by calibrated uncertainty or task difficulty. This is visible in the allocations: in Figure 4, Dolly's long-generation tasks receive 2288 (open QA), 1400 (brainstorming), and 1275 (general QA) samples, while classification, closed QA, information extraction, and summarization receive 196, 75, 86, and 60; Figure 6 shows the same pattern for FLAN. Consequently, Table 2's headline result (Weighted Task Diversity at k=3K, MMLU 39.74 vs. full-pool 35.33) does not separate 'prioritize low-confidence tasks' from 'prioritize long-output tasks.' The Limitations paragraph acknowledges that confidence may reflect pretraining exposure, but it does not address the length dependence that is built into the product definition, and no length-normalized confidence ablation is reported. Please report results with a per-token average log-probability or another length-normalized uncertainty score, or otherwise explicitly control for output length.
- [Abstract / Section 5.2] The abstract's claim that the algorithm 'consistently performs at or above the level of the best existing methods' is contradicted by the BBH columns of Table 1. At k=20K, Weighted Task Diversity scores 39.96±0.52, below Min Margin (40.44±0.48) and DPP (40.68±0.50); at k=45K, FL(γ=0.002) reaches 41.68±0.34 and Task Diversity 41.10±0.53, both above Weighted Task Diversity's 40.86±0.15. Section 5.2 itself concedes that on BBH 'no single method consistently dominates.' The abstract should be qualified to the datasets and metrics where the claim actually holds (e.g., MMLU and AlpacaEval, or 'on most budgets'), rather than stated as a universal property.
minor comments (5)
- [Section 4.1 / Tables 2-3] The text describes Dolly as a 15K dataset but the experiments use a 13.5K pool; please explain how the 13.5K subset was constructed and why it differs from the full dataset.
- [Section 4.3] The clamping notation in the allocation formula is typeset ambiguously, with the bounds appearing as |X_t| and 5 without a clear order. Please define [·]_a^b explicitly and state which bound is lower and which is upper.
- [Figures 3 and 5] The bar labels 'T ask' and 'WTD T ask Div' contain spacing artifacts; please correct these to 'Task' and 'WTD Task Div.'
- [Section 5.2] The sentence about the 45K budget 'saving 50% annotation budget when compared to random sampling' is imprecise; the saving is relative to the 90K full-data model, not to random sampling per se.
- [Reproducibility] The paper does not mention a code or data-release plan; releasing the selected prompt indices and the finetuning code would substantially aid reproducibility and comparison.
Circularity Check
No circularity: the inverse-confidence task allocation is computed directly from base-model confidences and evaluated on external benchmarks; self-citations to Bhatt et al. (2024) are not load-bearing.
full rationale
The paper's central claim is empirical: selecting prompts by task-level inverse confidence yields better downstream MMLU scores than full-data training. The selection rule in Section 4.3, alpha_t = clamp(C / conft, 5, |X_t|), uses only pretrained-model confidence averaged over unlabeled prompts. No parameter is fit to MMLU, BBH, or AlpacaEval, and the base budget of five examples is hand-set rather than tuned on test performance. The confidence score is a standard product of token probabilities, and although it may confound output length with uncertainty, that is a correctness risk acknowledged in the Limitations, not a circular reduction. The only self-citations to Bhatt et al. (2024) are used to frame the one-batch active-learning problem and to name a standard confidence score; they do not supply a theorem or fitted value on which the reported result depends. The 4% MMLU gain in Table 2 is therefore an independent experimental outcome, not an artifact of the method's definition or of the cited prior work.
Assumptions & free parameters
free parameters (1)
- Base per-task allocation =
5 examples
assumptions (5)
- domain assumption Task labels are available and partition the prompt pool into meaningful groups.
- domain assumption Average generated-sequence confidence is a valid proxy for annotation informativeness.
- domain assumption The unnormalized product of token probabilities is comparable across tasks of different output lengths.
- domain assumption Annotating more examples from low-confidence tasks improves finetuning more than annotating high-confidence tasks.
- domain assumption The one-batch active learning setting, where annotations are unknown at selection time and become available for the selected subset, is the correct problem framing.
Cite this review
Pith. "Pith review of Improving Task Diversity in Label Efficient Supervised Finetuning of LLMs." pith.science (2026). https://pith.science/paper/VWCELGOT
@misc{pith2026250721482,
author = {Pith},
title = {Pith review of: Improving Task Diversity in Label Efficient Supervised Finetuning of LLMs},
year = {2026},
howpublished = {\url{https://pith.science/paper/VWCELGOT}},
note = {Machine review of arXiv:2507.21482}
}
read the original abstract
Large Language Models (LLMs) have demonstrated remarkable capabilities across diverse domains, but developing high-performing models for specialized applications often requires substantial human annotation -- a process that is time-consuming, labor-intensive, and expensive. In this paper, we address the label-efficient learning problem for supervised finetuning (SFT) by leveraging task-diversity as a fundamental principle for effective data selection. This is markedly different from existing methods based on the prompt-diversity. Our approach is based on two key observations: 1) task labels for different prompts are often readily available; 2) pre-trained models have significantly varying levels of confidence across tasks. We combine these facts to devise a simple yet effective sampling strategy: we select examples across tasks using an inverse confidence weighting strategy. This produces models comparable to or better than those trained with more complex sampling procedures, while being significantly easier to implement and less computationally intensive. Notably, our experimental results demonstrate that this method can achieve better accuracy than training on the complete dataset (a 4\% increase in MMLU score). Across various annotation budgets and two instruction finetuning datasets, our algorithm consistently performs at or above the level of the best existing methods, while reducing annotation costs by up to 80\%.
Figures
Figures from the paper (3 more)
Reference graph
Works this paper leans on
-
[1]
Jordan Ash, Surbhi Goel, Akshay Krishnamurthy, and Sham Kakade. 2021. Gone fishing: Neural active learning with fisher embeddings. Advances in Neural Information Processing Systems, 34:8927--8939
work page 2021
-
[2]
Jordan T Ash, Chicheng Zhang, Akshay Krishnamurthy, John Langford, and Alekh Agarwal. 2019. Deep batch active learning by diverse, uncertain gradient lower bounds. arXiv preprint arXiv:1906.03671
arXiv 2019
-
[3]
Gantavya Bhatt, Yifang Chen, Arnav M Das, Jifan Zhang, Sang T Truong, Stephen Mussmann, Yinglun Zhu, Jeffrey Bilmes, Simon S Du, Kevin Jamieson, Jordan T Ash, and Robert D Nowak. 2024. An experimental design framework for label-efficient supervised finetuning of large language models. arXiv preprint arXiv:2401.06692
arXiv 2024
-
[4]
Jeff Bilmes. 2022. Submodularity in machine learning and artificial intelligence. arXiv preprint arXiv:2202.00132
arXiv 2022
-
[5]
Alexander Bukharin and Tuo Zhao. 2023. https://arxiv.org/abs/2311.14736 Data diversity matters for robust instruction tuning . Preprint, arXiv:2311.14736
arXiv 2023
-
[6]
Hardy Chen, Haoqin Tu, Fali Wang, Hui Liu, Xianfeng Tang, Xinya Du, Yuyin Zhou, and Cihang Xie. 2025. Sft or rl? an early investigation into training r1-like reasoning large vision-language models. arXiv preprint arXiv:2504.11468
arXiv 2025
-
[7]
Lichang Chen, Shiyang Li, Jun Yan, Hai Wang, Kalpa Gunaratna, Vikas Yadav, Zheng Tang, Vijay Srinivasan, Tianyi Zhou, Heng Huang, and Hongxia Jin. 2023. https://arxiv.org/abs/2307.08701 Alpagasus: Training a better alpaca with fewer data . Preprint, arXiv:2307.08701
arXiv 2023
-
[8]
Gui Citovsky, Giulia DeSalvo, Claudio Gentile, Lazaros Karydas, Anand Rajagopalan, Afshin Rostamizadeh, and Sanjiv Kumar. 2021. Batch active learning at scale. Advances in Neural Information Processing Systems, 34:11933--11944
2021
Show all 42 references
-
[9]
Mike Conover, Matt Hayes, Ankit Mathur, Jianwei Xie, Jun Wan, Sam Shah, Ali Ghodsi, Patrick Wendell, Matei Zaharia, and Reynold Xin. 2023. Free dolly: Introducing the world’s first truly open instruction-tuned llm
2023
-
[10]
Qianlong Du, Chengqing Zong, and Jiajun Zhang. 2023. https://arxiv.org/abs/2311.15653 Mods: Model-oriented data selection for instruction tuning . Preprint, arXiv:2311.15653
2023 arXiv
-
[11]
Dan Hendrycks, Collin Burns, Steven Basart, Andy Zou, Mantas Mazeika, Dawn Song, and Jacob Steinhardt. 2020. Measuring massive multitask language understanding. arXiv preprint arXiv:2009.03300
2020 arXiv
-
[12]
Edward J Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang, Lu Wang, and Weizhu Chen. 2021. Lora: Low-rank adaptation of large language models. arXiv preprint arXiv:2106.09685
2021 arXiv
-
[13]
Smith, Hannaneh Hajishirzi, and Pradeep Dasigi
Hamish Ivison, Noah A. Smith, Hannaneh Hajishirzi, and Pradeep Dasigi. 2022. Data-efficient fine-tuning using cross-task nearest neighbors. arXiv preprint arXiv:2212.00196
2022 arXiv
-
[14]
Jan Kremer, Kim Steenstrup Pedersen, and Christian Igel. 2014. Active learning with support vector machines. Wiley Interdisciplinary Reviews: Data Mining and Knowledge Discovery, 4(4):313--326
2014
-
[15]
Alex Kulesza, Ben Taskar, and 1 others. 2012. Determinantal point processes for machine learning. Foundations and Trends in Machine Learning , 5(2--3):123--286
2012
-
[16]
Po-Nien Kung, Fan Yin, Di Wu, Kai-Wei Chang, and Nanyun Peng. 2023. Active instruction tuning: Improving cross-task generalization by training on prompt sensitive tasks. arXiv preprint arXiv:2311.00288
2023 arXiv
-
[17]
David D Lewis. 1995. A sequential algorithm for training text classifiers: Corrigendum and additional data. In Acm Sigir Forum, volume 29, pages 13--19. ACM New York, NY, USA
1995
-
[18]
Hashimoto
Xuechen Li, Tianyi Zhang, Yann Dubois, Rohan Taori, Ishaan Gulrajani, Carlos Guestrin, Percy Liang, and Tatsunori B. Hashimoto. 2023. Alpacaeval: An automatic evaluator of instruction-following models. https://github.com/tatsu-lab/alpaca_eval
2023
-
[19]
Shayne Longpre, Le Hou, Tu Vu, Albert Webson, Hyung Won Chung, Yi Tay, Denny Zhou, Quoc V Le, Barret Zoph, Jason Wei, and 1 others. 2023. The flan collection: Designing data and methods for effective instruction tuning. arXiv preprint arXiv:2301.13688
2023 arXiv
-
[20]
Pitu B Mirchandani and Richard L Francis. 1990. Discrete location theory
1990
-
[21]
Baharan Mirzasoleiman, Jeff Bilmes, and Jure Leskovec. 2020. Coresets for data-efficient training of machine learning models. In International Conference on Machine Learning, pages 6950--6960. PMLR
2020
-
[22]
Mohamad Amin Mohamadi, Wonho Bae, and Danica J Sutherland. 2022. Making look-ahead active learning strategies feasible with neural tangent kernels. Advances in Neural Information Processing Systems, 35:12542--12553
2022
-
[23]
Shyam Nuggehalli, Jifan Zhang, Lalit Jain, and Robert Nowak. 2023. Direct: Deep active learning under imbalance and label noise. arXiv preprint arXiv:2312.09196
2023 arXiv
-
[24]
Long Ouyang, Jeff Wu, Xu Jiang, Diogo Almeida, Carroll L. Wainwright, Pamela Mishkin, Chong Zhang, Sandhini Agarwal, Katarina Slama, Alex Ray, John Schulman, Jacob Hilton, Fraser Kelton, Luke Miller, Maddie Simens, Amanda Askell, Peter Welinder, Paul Christiano, Jan Leike, and...
2022 arXiv
-
[25]
Yulei Qin, Yuncheng Yang, Pengcheng Guo, Gang Li, Hang Shao, Yuchen Shi, Zihan Xu, Yun Gu, Ke Li, and Xing Sun. 2024. Unleashing the power of data tsunami: A comprehensive survey on data assessment and selection for instruction tuning of language models. arXiv preprint arXiv:2...
2024 arXiv
-
[26]
Ozan Sener and Silvio Savarese. 2018. https://openreview.net/forum?id=H1aIuk-RW Active learning for convolutional neural networks: A core-set approach . In International Conference on Learning Representations (ICLR)
2018
-
[27]
Burr Settles. 2009. Active learning literature survey
2009
-
[28]
Mirac Suzgun, Nathan Scales, Nathanael Sch \"a rli, Sebastian Gehrmann, Yi Tay, Hyung Won Chung, Aakanksha Chowdhery, Quoc V Le, Ed H Chi, Denny Zhou, and 1 others. 2022. Challenging big-bench tasks and whether chain-of-thought can solve them. arXiv preprint arXiv:2210.09261
2022 arXiv
-
[29]
Simon Tong and Daphne Koller. 2001. Support vector machine active learning with applications to text classification. Journal of machine learning research, 2(Nov):45--66
2001
-
[30]
Hugo Touvron, Louis Martin, Kevin Stone, Peter Albert, Amjad Almahairi, Yasmine Babaei, Nikolay Bashlykov, Soumya Batra, Prajjwal Bhargava, Shruti Bhosale, and 1 others. 2023. Llama 2: Open foundation and fine-tuned chat models. arXiv preprint arXiv:2307.09288
2023 arXiv
-
[31]
Peiqi Wang, Yikang Shen, Zhen Guo, Matthew Stallone, Yoon Kim, Polina Golland, and Rameswar Panda. 2024. Diversity measurement and subset selection for instruction tuning datasets. arXiv preprint arXiv:2402.02318
2024 arXiv
-
[32]
Yizhong Wang, Hamish Ivison, Pradeep Dasigi, Jack Hessel, Tushar Khot, Khyathi Raghavi Chandu, David Wadden, Kelsey MacMillan, Noah A Smith, Iz Beltagy, and 1 others. 2023. How far can camels go? exploring the state of instruction tuning on open resources. arXiv preprint arXiv...
2023 arXiv
-
[33]
Yizhong Wang, Swaroop Mishra, Pegah Alipoormolabashi, Yeganeh Kordi, Amirreza Mirzaei, Anjana Arunkumar, Arjun Ashok, Arut Selvan Dhanasekaran, Atharva Naik, David Stap, Eshaan Pathak, Giannis Karamanolakis, Haizhi Gary Lai, Ishan Purohit, Ishani Mondal, Jacob Anderson, Kirby ...
2022 arXiv
-
[34]
Zhao, Kelvin Guu, Adams Wei Yu, Brian Lester, Nan Du, Andrew M
Jason Wei, Maarten Bosma, Vincent Y. Zhao, Kelvin Guu, Adams Wei Yu, Brian Lester, Nan Du, Andrew M. Dai, and Quoc V. Le. 2022. https://arxiv.org/abs/2109.01652 Finetuned language models are zero-shot learners . Preprint, arXiv:2109.01652
2022 arXiv
-
[35]
Kai Wei, Rishabh Iyer, and Jeff Bilmes. 2015. Submodularity in data subset selection and active learning. In International conference on machine learning, pages 1954--1963. PMLR
2015
-
[36]
Tian Xie, Jifan Zhang, Haoyue Bai, and Robert Nowak. 2024. Deep active learning in the open world. arXiv preprint arXiv:2411.06353
2024 arXiv
-
[37]
Yu Yang, Siddhartha Mishra, Jeffrey Chiang, and Baharan Mirzasoleiman. 2024. 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:83465--83496
2024
-
[38]
Jifan Zhang, Gregory Canal, Yinglun Zhu, Robert D Nowak, Yifang Chen, Arnav M Das, Gantavya Bhatt, Stephen Mussmann, Jeffrey Bilmes, Simon S Du, and 1 others. 2023. Labelbench: A comprehensive framework for benchmarking adaptive label-efficient learning. arXiv preprint arXiv:2...
2023 arXiv
-
[39]
Jifan Zhang, Julian Katz-Samuels, and Robert Nowak. 2022. Galaxy: Graph-based active learning at the extreme. In International Conference on Machine Learning, pages 26223--26238. PMLR
2022
-
[40]
Kuan Lok Zhou, Jiayi Chen, Siddharth Suresh, Reuben Narad, Timothy T Rogers, Lalit K Jain, Robert D Nowak, Bob Mankoff, and Jifan Zhang. 2025. Bridging the creativity understanding gap: Small-scale human alignment enables expert-level humor ranking in llms. arXiv preprint arXi...
2025 arXiv
-
[41]
online" 'onlinestring :=
ENTRY address archivePrefix author booktitle chapter edition editor eid eprint eprinttype howpublished institution journal key month note number organization pages publisher school series title type volume year doi pubmed url lastchecked label extra.label sort.label short.list...
-
[42]
write newline
" write newline "" before.all 'output.state := FUNCTION n.dashify 't := "" t empty not t #1 #1 substring "-" = t #1 #2 substring "--" = not "--" * t #2 global.max substring 't := t #1 #1 substring "-" = "-" * t #2 global.max substring 't := while if t #1 #1 substring * t #2 gl...
Reviewed August 6, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.