arXiv:2605.30857cs.CL2026-05

用大模型自身激活状态选数据,让小数据集也能高效训练大模型。

MADS: Model-Aware Diverse Core Set Selection for Instruction Tuning

论文配图:MADS: Model-Aware Diverse Core Set Selection for Instruction Tuning
图 1 · 摘自论文原文
  • 基于大模型推理时的神经激活特征筛选数据,实现更真实多样性的核心集
  • 仅用15%数据(52K中的7800条)就提升4个大模型平均2.5%性能
  • 适合资源有限但想高效微调大模型的研究者或开发者

指令微调用于提升大语言模型(LLM)的指令遵循能力。随着指令微调数据量增加,选择最优核心集变得尤为重要。然而,确保核心集的多样性仍是一大挑战。现有方法主要依据文本本身的特征区分数据,与大模型自身的理解与表征解耦。为此,我们提出模型感知的多样化核心集选择方法(MADS),通过大模型推理过程中的神经激活状态来区分数据特征,作为基于覆盖度的高效实例化,确保核心集的多样性。我们在六个涵盖五种不同任务的基准上进行了广泛评估。结果显示,由3B参数模型选出的核心集,在微调7B、8B和13B参数的大模型时表现良好。在包含52,000条指令-响应对的Alpaca-GPT4数据集上,由Llama-3.2-3B-Instruct选出的占原数据15%的核心集,在微调四个更大基础模型时,相比全量数据训练平均提升2.5%。实验表明,该方法在降低数据需求的同时提升了多下游任务性能。

原文摘要 · Abstract (English)

Instruction fine-tuning is employed to enhance the instruction-following ability of large language models (LLMs). As the amount of instruction fine-tuning data increases, selecting the optimal core set becomes particularly important. However, ensuring the diversity of the core set remains a significant challenge. Existing methods predominantly distinguish different training data based on the text features themselves, decoupled from LLMs' own understanding and representation of the data. To address this issue, we propose a Model-Aware Diverse Core Set Selection method, which distinguishes data features based on the neural activation states during LLM inference. This approach serves as an efficient instantiation of coverage-based selection using model-intrinsic activation features to ensure the diversity in the core set. We extensively evaluate our method on six benchmarks that cover five distinct tasks. In our method, the core set selected by the 3B-parameter LLM performs effectively when utilized to fine-tune larger models with 7B, 8B, and 13B parameters. Experimental results on the Alpaca-GPT4 dataset, which comprises 52K instruction-response pairs, show that the core set, sized at 15\% of the original dataset and selected by Llama-3.2-3B-Instruct, achieves an average improvement of 2.5\% when fine-tuning four larger base models compared with training on the full dataset. The experimental results demonstrate that our method enhances model performance on multiple downstream tasks while reducing data requirements.

指令微调核心集选择大模型训练多样性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。