用无标签数据让小模型高效学习大模型知识,提升训练效率。
Learning with Less: Knowledge Distillation from Large Language Models via Unlabeled Data
- 通过师生协同信号自适应选择优质样本
- 在少量标注数据下实现更优性能,数据效率更高
- 适合资源受限场景下小模型知识迁移
在真实NLP应用中,大型语言模型(LLMs)因其在海量数据上的训练而展现出巨大潜力。然而,其庞大的规模和高计算需求限制了实际部署,尤其在需要进一步微调时。为此,通常采用更小的模型进行部署,但其训练常受限于标注数据稀缺。相比之下,无标签数据往往丰富,可利用LLMs生成伪标签用于小模型(学生)训练,使小模型从大模型(教师)获取知识的同时降低计算成本。该过程面临伪标签噪声等挑战,因此高质量样本的选择至关重要。为此,我们提出LLKD,一种在减少计算资源和数据量的前提下实现知识蒸馏的方法。LLKD是一种自适应样本选择方法,融合教师与学生的信号:优先选择教师置信度高(标签可靠)且学生信息需求高的样本(学习难点)。大量实验表明,LLKD在多个数据集上均表现优异,具备更高的数据效率。
原文摘要 · Abstract (English)
In real-world NLP applications, Large Language Models (LLMs) offer promising solutions due to their extensive training on vast datasets. However, the large size and high computation demands of LLMs limit their practicality in many applications, especially when further fine-tuning is required. To address these limitations, smaller models are typically preferred for deployment. However, their training is hindered by the scarcity of labeled data. In contrast, unlabeled data is often readily which can be leveraged by using LLMs to generate pseudo-labels for training smaller models. This enables the smaller models (student) to acquire knowledge from LLMs(teacher) while reducing computational costs. This process introduces challenges, such as potential noisy pseudo-labels. Selecting high-quality and informative data is therefore critical to enhance model performance while improving the efficiency of data utilization. To address this, we propose LLKD that enables Learning with Less computational resources and less data for Knowledge Distillation from LLMs. LLKD is an adaptive sample selection method that incorporates signals from both the teacher and student. Specifically, it prioritizes samples where the teacher demonstrates high confidence in its labeling, indicating reliable labels, and where the student exhibits a high information need, identifying challenging samples that require further learning. Our comprehensive experiments show that LLKD achieves superior performance across various datasets with higher data efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。