LEAD: Iterative Data Selection for Efficient LLM Instruction Tuning
On the map: LLM alignment / RLHF / preference optimisation → LLM preference optimization methods
Abstract (arXiv)
Instruction tuning has emerged as a critical paradigm for improving the capabilities and alignment of large language models (LLMs). However, existing iterative model-aware data selection methods incur significant computational overhead, as they rely on repeatedly performing full-dataset model inference to estimate sample utility for subsequent training iterations, creating a fundamental efficiency bottleneck. In this paper, we propose LEAD, an efficient iterative data selection framework that accurately estimates sample utility entirely within the standard training loop, eliminating the need for costly additional model inference. At its core, LEAD introduces Instance-Level Dynamic Uncertainty (IDU), a theoretically grounded utility function combining instantaneous training loss, gradient-based approximation of loss changes, and exponential smoothing of historical loss signals. To further scale efficiently to large datasets, LEAD employs a two-stage, coarse-to-fine selection strategy, adaptively prioritizing informative clusters through a multi-armed bandit mechanism, followed by precise fine-grained selection of high-utility samples using IDU. Extensive experiments across four diverse benchmarks show that LEAD significantly outperforms state-of-the-art methods, improving average model performance by 6.1%-10.8% while using only 2.5% of the training data and reducing overall training time by 5-10x.
Key facts. Cited by 13 papers in the map (data through 14 Oct 2026). Placed in LLM preference optimization methods, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 29 of its 66 references that are in the map. First posted on arXiv 12 May 2025.
Builds on
29 of 66 references.
Closest ancestors · 2
Same direction, newest first
- 2023LIMA: Less Is More for Alignment18 May
- 2023OpenAssistant Conversations -- Democratizing Large Language Model Alignment14 Apr
Other references · 27
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023A Survey on Evaluation of Large Language Modelscited by 1,242
- 2023Enhancing Chat Language Models by Scaling High-quality Instructional Conversationscited by 839
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 629
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 532
- 2020TyDi QA: A Benchmark for Information-Seeking Question Answering in Typologically Diverse Languagescited by 488
- 2023How Far Can Camels Go? Exploring the State of Instruction Tuning on Open Resourcescited by 408
- 2022Unnatural Instructions: Tuning Language Models with (Almost) No Human Laborcited by 397
15 more
- 2017Snorkel: Rapid Training Data Creation with Weak Supervisioncited by 372
- 2024A Survey on Data Selection for Language Modelscited by 236
- 2024Superfiltering: Weak-to-Strong Data Filtering for Fast Instruction-Tuningcited by 138
- 2023DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Modelscited by 123
- 2023#InsTag: Instruction Tagging for Analyzing Supervised Fine-tuning of Large Language Modelscited by 109
- 2024What is Your Data Worth to GPT? LLM-Scale Data Valuation with Influence Functionscited by 94
- 2024The Dawn of Natural Language to SQL: Are We Fully Ready?cited by 79
- 2023Data Diversity Matters for Robust Instruction Tuningcited by 76
- 2024A Survey of Text-to-SQL in the Era of LLMs: Where are we, and where are we going?cited by 56
- 2023Self-Evolved Diverse Data Sampling for Efficient Instruction Tuningcited by 52
- 2024SmallToLarge (S2L): Scalable Data Selection for Fine-tuning Large Language Models by Summarizing Training Trajectories of Small Modelscited by 44
- 2024Diversify and Conquer: Diversity-Centric Data Selection with Iterative Refinementcited by 13
- 2024ItD: Large Language Models Can Teach Themselves Induction through Deductioncited by 8
- 2024IterSelectTune: An Iterative Training Framework for Efficient Instruction-Tuning Data Selectioncited by 6
- 2024Instruction Embedding: Latent Representations of Instructions Towards Task Identificationcited by 4
Built on by
Citing papers per year: 2025: 2; 2026: 11.
13 papers.
Sort:
By area:
- 2026AOrchestra: Automating Sub-Agent Creation for Agentic Orchestrationcited by 35
- 2025A Survey of Data Agents: Emerging Paradigm or Overstated Hype?cited by 30
- 2026AutoWebWorld: Synthesizing Infinite Verifiable Web Environments via Finite State Machinescited by 21
- 2025Concise Reasoning, Big Gains: Pruning Long Reasoning Trace with Difficulty-Aware Promptingcited by 15
- 2026InfoPO: Information-Driven Policy Optimization for User-Centric Agentscited by 12
- 2026SPICE: Submodular Penalized Information-Conflict Selection for Efficient Large Language Model Trainingcited by 7
- 2026ParisKV: Fast and Drift-Robust KV-Cache Retrieval for Long-Context LLMscited by 2
- 2026Scalable Environments Drive Generalizable Agentscited by 2
- 2026From Instance Selection to Fixed-Pool Data Recipe Search for Supervised Fine-Tuningcited by 1
- 2026TuneAhead: Predicting Fine-tuning Performance Before Full Training Beginscited by 1
- 2026A Risk Decomposition Framework for Pre-Hoc Fine-Tuning Predictioncited by 0
- 2026Demonstrating TOFFEE: A Learned System for Synthesizing Data Agent Trajectories at Scalecited by 0
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuning30 Sep 2026
- 2026Demonstrating TOFFEE: A Learned System for Synthesizing Data Agent Trajectories at Scale7 Jul 2026
- 2026TuneAhead: Predicting Fine-tuning Performance Before Full Training Begins16 Jun 2026
- 2026A Risk Decomposition Framework for Pre-Hoc Fine-Tuning Prediction16 Jun 2026
- 2026Scalable Environments Drive Generalizable Agents18 May 2026
- 2026From Instance Selection to Fixed-Pool Data Recipe Search for Supervised Fine-Tuning13 May 2026
- 2026InfoPO: Information-Driven Policy Optimization for User-Centric Agents28 Feb 2026
- 2026AutoWebWorld: Synthesizing Infinite Verifiable Web Environments via Finite State Machines15 Feb 2026
- 2026ParisKV: Fast and Drift-Robust KV-Cache Retrieval for Long-Context LLMs7 Feb 2026
- 2026AOrchestra: Automating Sub-Agent Creation for Agentic Orchestration3 Feb 2026
- 2026SPICE: Submodular Penalized Information-Conflict Selection for Efficient Large Language Model Training30 Jan 2026
- 2025A Survey of Data Agents: Emerging Paradigm or Overstated Hype?27 Oct 2025
- 2026AOrchestra: Automating Sub-Agent Creation for Agentic Orchestrationcited by 35
- 2025A Survey of Data Agents: Emerging Paradigm or Overstated Hype?cited by 30
- 2026AutoWebWorld: Synthesizing Infinite Verifiable Web Environments via Finite State Machinescited by 21
- 2026InfoPO: Information-Driven Policy Optimization for User-Centric Agentscited by 12
- 2026Demonstrating TOFFEE: A Learned System for Synthesizing Data Agent Trajectories at Scalecited by 0
- 2026Demonstrating TOFFEE: A Learned System for Synthesizing Data Agent Trajectories at Scale7 Jul 2026
- 2026InfoPO: Information-Driven Policy Optimization for User-Centric Agents28 Feb 2026
- 2026AutoWebWorld: Synthesizing Infinite Verifiable Web Environments via Finite State Machines15 Feb 2026
- 2026AOrchestra: Automating Sub-Agent Creation for Agentic Orchestration3 Feb 2026
- 2025A Survey of Data Agents: Emerging Paradigm or Overstated Hype?27 Oct 2025
- 2026SPICE: Submodular Penalized Information-Conflict Selection for Efficient Large Language Model Trainingcited by 7
- 2026Scalable Environments Drive Generalizable Agentscited by 2
- 2026From Instance Selection to Fixed-Pool Data Recipe Search for Supervised Fine-Tuningcited by 1
- 2026TuneAhead: Predicting Fine-tuning Performance Before Full Training Beginscited by 1
- 2026A Risk Decomposition Framework for Pre-Hoc Fine-Tuning Predictioncited by 0
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuningcited by 0
- 2026Every Batch Is Its Own Validation Set: Leave-One-Out Gradient Matching for Online Data Selection in LLM Fine-Tuning30 Sep 2026
- 2026TuneAhead: Predicting Fine-tuning Performance Before Full Training Begins16 Jun 2026
- 2026A Risk Decomposition Framework for Pre-Hoc Fine-Tuning Prediction16 Jun 2026
- 2026Scalable Environments Drive Generalizable Agents18 May 2026
- 2026From Instance Selection to Fixed-Pool Data Recipe Search for Supervised Fine-Tuning13 May 2026
- 2026SPICE: Submodular Penalized Information-Conflict Selection for Efficient Large Language Model Training30 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Models
- 2025Towards Alignment-Centric Paradigm: A Survey of Instruction Tuning in Large Language Models
- 2024ROSE: A Reward-Oriented Data Selection Framework for LLM Task-Specific Instruction Tuning
- 2024Take the essence and discard the dross: A Rethinking on Data Selection for Fine-Tuning Large Language Models
- 2024How to Train Data-Efficient LLMs
- 2024Selective Reflection-Tuning: Student-Selected Data Recycling for LLM Instruction-Tuning
- 2026Aligning the Query Space: Greedy Information Projection for Language Model Data Selection
- 2025Diversity-driven Data Selection for Language Model Tuning through Sparse Autoencoder