arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.14327cs.CLcs.AI

Dynosaur:一种用于指令微调数据策展的动态增长范式

Dynosaur: A Dynamic Growth Paradigm for Instruction-Tuning Data Curation

UCLA(加州大学洛杉矶分校);Peking University(北京大学);UIUC(伊利诺伊大学厄巴纳-香槟分校)

Da Yin, Xiao Liu, Fan Yin, Ming Zhong, Hritik Bansal, Jiawei Han, Kai-Wei Chang

AI总结:

本文提出Dynosaur动态增长范式,利用现有数据集元数据自动生成指令微调数据,以低成本获得高质量数据,并通过持续学习方案缓解遗忘并提升泛化能力。

摘要:

指令微调已出现以增强大型语言模型(LLMs)理解指令并生成适当响应的能力。现有方法要么手动标注,要么使用LLM(例如GPT系列)生成用于指令微调的数据。然而,它们往往忽视了将指令与现有的已标注数据集关联起来。在本文中,我们提出了Dynosaur,一种用于自动策展指令微调数据的动态增长范式。基于现有数据集的元数据,我们使用LLM通过识别相关数据字段并生成适当的指令来自动构建指令微调数据。通过利用现有的已标注数据集,Dynosaur具有以下几个优势:1)它降低了生成指令的API成本(例如,通过调用GPT-3.5-turbo生成800K个指令微调样本的成本不到12美元);2)它为指令微调提供了高质量的数据(例如,在数据量相当的情况下,它在Super-NI和Longform上的表现优于Alpaca和Flan);3)当新的已标注数据集可用时,它通过生成指令微调数据来支持模型的持续改进。我们进一步研究了一种持续学习方案,用于学习不断增长的指令微调数据集,并证明使用多样化的指令嵌入重放任务不仅有助于缓解遗忘问题,而且能更好地泛化到未见过的任务。代码和数据可在https://github.com/WadeYin9712/Dynosaur获取。

英文摘要:

Instruction tuning has emerged to enhance the capabilities of large language models (LLMs) to comprehend instructions and generate appropriate responses. Existing methods either manually annotate or employ LLM (e.g., GPT-series) to generate data for instruction tuning. However, they often overlook associating instructions with existing annotated datasets. In this paper, we propose Dynosaur, a dynamic growth paradigm for the automatic curation of instruction-tuning data. Based on the metadata of existing datasets, we use LLMs to automatically construct instruction-tuning data by identifying relevant data fields and generating appropriate instructions. By leveraging the existing annotated datasets, Dynosaur offers several advantages: 1) it reduces the API cost for generating instructions (e.g., it costs less than $12 USD by calling GPT-3.5-turbo for generating 800K instruction tuning samples; 2) it provides high-quality data for instruction tuning (e.g., it performs better than Alpaca and Flan on Super-NI and Longform with comparable data sizes); and 3) it supports the continuous improvement of models by generating instruction-tuning data when a new annotated dataset becomes available. We further investigate a continual learning scheme for learning with the ever-growing instruction-tuning dataset, and demonstrate that replaying tasks with diverse instruction embeddings not only helps mitigate forgetting issues but generalizes to unseen tasks better. Code and data are available at https://github.com/WadeYin9712/Dynosaur.

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

↑