arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.04425cs.CL
收藏

DELIFT:数据高效的语言模型指令微调

DELIFT: Data Efficient Language model Instruction Fine Tuning

University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);IBM Research(IBM研究院)

Ishika Agarwal, Krishnateja Killamsetty, Lucian Popa, Marina Danilevksy

AI总结:

本研究提出DELIFT算法,通过成对效用度量与子模函数在指令微调、任务特定微调、持续微调三阶段优化数据选择,可减少70%微调数据量且不损失性能,效率与效果均优于现有方法。

摘要:

微调大语言模型(LLM)对提升其在特定任务上的表现至关重要,但由于数据存在冗余或信息含量低的问题,该过程往往资源消耗巨大。为解决这种低效问题,我们提出了DELIFT(数据高效语言模型指令微调),这是一种新型算法,可在微调的三个关键阶段系统性优化数据选择:(1)指令微调、(2)任务特定微调(如推理、问答)、(3)持续微调(如纳入新数据版本)。现有方法要么聚焦单阶段优化,要么依赖计算密集的梯度计算,而DELIFT可在所有阶段高效运行。我们方法的核心是成对效用度量,它量化单个数据样本对提升模型对其他样本的响应能力的益处,有效衡量其相对于模型当前能力的信息价值。通过将不同的子模函数应用于该度量,DELIFT能选出在所有微调阶段都有用的多样化最优子集。在不同任务和模型规模上的实验表明,DELIFT可将微调数据量减少多达70%且不损失性能,大幅节省计算资源,在效率和效果上均优于现有方法。

英文摘要:

Fine-tuning large language models (LLMs) is essential for enhancing their performance on specific tasks but is often resource-intensive due to redundant or uninformative data. To address this inefficiency, we introduce DELIFT (Data Efficient Language model Instruction Fine-Tuning), a novel algorithm that systematically optimizes data selection across the three key stages of fine-tuning: (1) instruction tuning, (2) task-specific fine-tuning (e.g., reasoning, question-answering), and (3) continual fine-tuning (e.g., incorporating new data versions). Unlike existing methods that focus on single-stage optimization or rely on computationally intensive gradient calculations, DELIFT operates efficiently across all stages. Central to our approach is a pairwise utility metric that quantifies how beneficial a data sample is for improving the model's responses to other samples, effectively measuring the informational value relative to the model's current capabilities. By leveraging different submodular functions applied to this metric, DELIFT selects diverse and optimal subsets that are useful across all stages of fine-tuning. Experiments across various tasks and model scales demonstrate that DELIFT can reduce the fine-tuning data size by up to 70% without compromising performance, offering significant computational savings and outperforming existing methods in both efficiency and efficacy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑