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

构建用于云端低成本 LLM 微调的数据增强模型家族

Building a Family of Data Augmentation Models for Low-cost LLM Fine-tuning on the Cloud

School of Computer Science, Fudan University(复旦大学计算机科学学院);Alibaba Cloud Computing(阿里云计算)

Yuanhao Yue, Chengyu Wang, Jun Huang, Peng Wang

AI总结:

本文提出基于小型 LLM 的数据增强模型家族,通过自动数据收集、质量评估与教师模型蒸馏支持指令扩展、精炼及指令-响应对扩展,并集成到云端平台实现低成本微调。

摘要:

使 LLM 专门适配各种领域特定任务,已成为实现高性能的关键步骤。然而,特定领域数据集的构建与标注通常成本高昂。除了使用性能优越但昂贵的闭源 LLM API 来构建数据集外,一些开源模型已经足够强大,能够在许多场景中处理数据集构建工作。因此,我们提出了一个数据增强模型家族,旨在显著提升模型微调效率。这些模型基于足够小的 LLM 训练,能够以较低的推理成本支持关键功能:指令扩展、指令精炼以及指令-响应对扩展。为实现这一目标,我们首先构建了一个自动数据收集系统,其种子数据集来自公共代码仓库和我们的内部数据集。该系统利用强大的 LLM 对指令和响应进行扩展、精炼与重写,并结合质量评估技术。随后,我们介绍了模型的训练流程,该流程有效地从教师 LLM 中蒸馏任务求解和文本合成能力。最后,我们展示了如何将这些功能集成到机器学习平台中,从数据集准备和训练两个角度为用户支持低成本 LLM 微调。实验和一项应用研究证明了我们方法的有效性。

英文摘要:

Specializing LLMs in various domain-specific tasks has emerged as a critical step towards achieving high performance. However, the construction and annotation of datasets in specific domains are always very costly. Apart from using superior and expensive closed-source LLM APIs to construct datasets, some open-source models have become strong enough to handle dataset construction in many scenarios. Thus, we present a family of data augmentation models designed to significantly improve the efficiency for model fine-tuning. These models, trained based on sufficiently small LLMs, support key functionalities with low inference costs: instruction expansion, instruction refinement, and instruction-response pair expansion. To fulfill this goal, we first construct an automatic data collection system with seed datasets generated from both public repositories and our in-house datasets. This system leverages powerful LLMs to expand, refine and re-write the instructions and responses, incorporating quality assessment techniques. Following this, we introduce the training process of our models, which effectively distills task-solving and text synthesis abilities from teacher LLMs. Finally, we demonstrate how we integrate these functionalities into a machine learning platform to support low-cost LLM fine-tuning from both dataset preparation and training perspectives for users. Experiments and an application study prove the effectiveness of our approach.

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

↑