通过轨迹匹配生成可迁移的文本数据,显著减少大模型训练所需数据量。
Transferable text data distillation by trajectory matching
- 基于轨迹匹配学习伪提示数据,实现跨模型架构迁移
- 在ARC-Easy和MMLU上优于当前最优数据选择方法
- 适用于指令微调等文本生成任务,适合追求高效训练的研究者
在大语言模型(LLM)领域,随着模型规模增大,训练成本也随之上升。亟需降低训练数据规模。与数据选择方法相比,数据蒸馏旨在合成少量样本以达到全量数据的训练效果,具有更强灵活性。尽管在计算机视觉中已取得成功,但文本数据的离散性长期制约其在自然语言处理中的应用。本文提出一种基于轨迹匹配学习伪提示数据并寻找最近邻ID的方法,实现跨架构迁移。蒸馏过程中引入正则化损失以提升生成数据的鲁棒性。据我们所知,这是首个适用于文本生成任务(如指令微调)的数据蒸馏方法。在两个基准数据集(ARC-Easy 和 MMLU 指令微调数据集)上的实验表明,该方法优于当前最先进的数据选择方法 LESS。此外,该方法在不同模型结构间(如 OPT 到 Llama)展现出良好可迁移性。
原文摘要 · Abstract (English)
In the realm of large language model (LLM), as the size of large models increases, it also brings higher training costs. There is a urgent need to minimize the data size in LLM training. Compared with data selection method, the data distillation method aims to synthesize a small number of data samples to achieve the training effect of the full data set and has better flexibility. Despite its successes in computer vision, the discreteness of text data has hitherto stymied its exploration in natural language processing (NLP). In this work, we proposed a method that involves learning pseudo prompt data based on trajectory matching and finding its nearest neighbor ID to achieve cross-architecture transfer. During the distillation process, we introduce a regularization loss to improve the robustness of our distilled data. To our best knowledge, this is the first data distillation work suitable for text generation tasks such as instruction tuning. Evaluations on two benchmarks, including ARC-Easy and MMLU instruction tuning datasets, established the superiority of our distillation approach over the SOTA data selection method LESS. Furthermore, our method demonstrates a good transferability over LLM structures (i.e., OPT to Llama).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。