通过学生模型反馈筛选高质量数据,提升大模型压缩效率与效果
Less is More: Selective Reflection for Compatible and Efficient Knowledge Distillation in Large Language Models
- 用学生模型输出对比真实数据,自动挑选适合的训练样本
- 实验显示性能提升,训练时间最多减少39%
- 无需修改原有方法,可直接插入任一知识蒸馏流程
知识蒸馏(KD)是将大语言模型压缩为高效小模型的核心技术。现有白盒蒸馏方法多关注真实答案与学生模型输出的平衡,却忽视了训练数据质量与学生模型适配性。为此,我们提出选择性反思蒸馏(SRD),一个基于学生模型反思的数据优化框架。SRD通过比较真实数据与学生模型输出,动态评估并筛选出高质量、适配性强的提示-响应对,利用难度分级实现自动化数据精选。随后采用课程调度策略,在固定间隔逐步引入精选数据子集进行蒸馏。作为即插即用模块,SRD在多种白盒蒸馏方法和模型架构上均稳定提升效果,并显著降低训练计算开销。在多个语言模型基准测试中,SRD持续提升蒸馏模型性能,且训练耗时最多减少39%。结果表明,数据质量与兼容性是高效蒸馏的关键,SRD为此提供了一个系统性解决方案。
原文摘要 · Abstract (English)
Knowledge Distillation (KD) is a fundamental technique for compressing large language models (LLMs) into compact, efficient student models. However, existing white-box KD methods mainly focus on balancing ground truth and student-generated responses while overlooking two critical factors: training data quality and student-model compatibility. To address these limitations, we propose Selective Reflection Distillation (SRD), a novel data curation framework that leverages reflections from student models to systematically refine training data. SRD dynamically evaluates and selects prompt-response pairs by comparing ground truth data with student model outputs, selectively curating high-quality, student-compatible training instances through automated ranking based on difficulty. Furthermore, after selecting the training data, a curriculum scheduling strategy is employed to incrementally introduce these curated subsets into the distillation process at fixed intervals. As a plug-and-play enhancement, SRD consistently improves distillation outcomes across diverse white-box KD approaches and model architectures, as well as decreases computational cost significantly during KD training. Experiments on a range of language model benchmarks demonstrate SRD's consistent improvements in distilled model performance, as well as a reduction in training runtime by up to 39%, under diverse KD methods and model families. Notably, SRD operates as a plug-and-play module, enhancing sample efficiency without modifying underlying KD algorithms. Our findings highlight that data quality and compatibility are pivotal to effective and efficient distillation of LLMs, and SRD provides a principled framework to achieve both. This work advances the understanding of data-centric factors in KD and offers practical insights for enhancing the capability and efficiency of compressed LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。