arXiv:2411.06839cs.CLcs.AI2024-11被引 8

用低秩适配提升大模型知识蒸馏效率,压缩更优。

LLM-NEO: Parameter Efficient Knowledge Distillation for Large Language Models

  • 将低秩适配融入知识蒸馏,实现参数高效压缩。
  • 在Llama 2和Llama 3.2上性能超越多个基线方法。
  • 对多种LoRA变体鲁棒,适合资源受限场景部署。

知识蒸馏(KD)是压缩大语言模型(LLM)的主流方法。本文重新审视了KD与低秩适配(LoRA),发现二者遵循相同范式。受此启发,提出参数高效的蒸馏方法LLM-NEO,将LoRA引入KD以提升知识迁移效率。进一步总结了LLM-NEO的超参数调优指南。在压缩Llama 2和Llama 3.2的实验中,LLM-NEO显著优于多种基线方法。深入分析表明,该方法对不同LoRA变体具有强鲁棒性。代码与训练模型已开源于GitHub。

原文摘要 · Abstract (English)

Knowledge distillation (KD) has been a predominant method for compressing Large Language Models (LLMs). In this paper, we first revisit KD and Low-Rank Adaption (LoRA) and demonstrate that they follow the same paradigm. Inspired by this observation, we propose a parameter-efficient knowledge distillation method, LLM-NEO, which integrates LoRA into KD to improve the efficiency of knowledge transfer. After that, we summarize some valuable guidelines for the hyperparameters in LLM-NEO. Experimental results on compressing Llama 2 and Llama 3.2 show that LLM-NEO outperforms various baselines. Further analysis demonstrates the robustness of the proposed LLM-NEO on variants of LoRA. The code and trained models are available at [Github](https://github.com/yang3121099/LLM-Neo).

知识蒸馏大模型压缩低秩适配参数效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。