arXiv:2502.06663cs.LG2025-02被引 15

通过感知剪枝的预训练,让小模型在边缘设备上达到大模型性能。

EfficientLLM: Scalable Pruning-Aware Pretraining for Architecture-Agnostic Edge Language Models

  • 在预训练阶段动态剪枝,自动设计模型结构以适应压缩需求。
  • 参数量1亿到10亿的小模型在常识任务上超越现有顶尖小型模型。
  • 不依赖特定架构,可推广至多种模型,适合资源受限场景部署。

当前大语言模型(LLM)遵循缩放定律,依赖庞大模型规模实现智能。然而,云成本、延迟与隐私问题促使对紧凑型边缘语言模型的需求日益迫切。本文提出一种剪枝感知的预训练方法,旨在保留经优化后更大模型的性能。该方法具备:1)数据可扩展性:在LLM中引入极少量参数组,并持续优化结构剪枝,将如LLM-Pruner和SparseGPT等后训练剪枝方法延伸至预训练阶段;2)架构无关性:基于显著性驱动的剪枝自动设计模型架构,首次在现代预训练中超越现有手工设计的SOTA LLM。所提出的EfficientLLM显著优于参数量为1亿至10亿的主流小型模型(如MobileLLM、SmolLM、Qwen2.5-0.5B、OLMo-1B、Llama3.2-1B),在通用常识基准测试中表现优异。EfficientLLM首次弥合了传统模型压缩与直接预训练之间的性能差距,项目将全开源,地址为https://github.com/Xingrun-Xing2/EfficientLLM。

原文摘要 · Abstract (English)

Modern large language models (LLMs) driven by scaling laws, achieve intelligence emergency in large model sizes. Recently, the increasing concerns about cloud costs, latency, and privacy make it an urgent requirement to develop compact edge language models. Distinguished from direct pretraining that bounded by the scaling law, this work proposes the pruning-aware pretraining, focusing on retaining performance of much larger optimized models. It features following characteristics: 1) Data-scalable: we introduce minimal parameter groups in LLM and continuously optimize structural pruning, extending post-training pruning methods like LLM-Pruner and SparseGPT into the pretraining phase. 2) Architecture-agnostic: the LLM architecture is auto-designed using saliency-driven pruning, which is the first time to exceed SoTA human-designed LLMs in modern pretraining. We reveal that it achieves top-quality edge language models, termed EfficientLLM, by scaling up LLM compression and extending its boundary. EfficientLLM significantly outperforms SoTA baselines with $100M \sim 1B$ parameters, such as MobileLLM, SmolLM, Qwen2.5-0.5B, OLMo-1B, Llama3.2-1B in common sense benchmarks. As the first attempt, EfficientLLM bridges the performance gap between traditional LLM compression and direct pretraining methods, and we will fully open source at https://github.com/Xingrun-Xing2/EfficientLLM.

边缘计算模型压缩自适应剪枝高效预训练

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