arXiv:2502.00899stat.MLcs.LG2025-02被引 3

提出统一框架HASSLE-free,高效压缩大模型参数

HASSLE-free: A unified Framework for Sparse plus Low-Rank Matrix Decomposition for LLMs

  • 引入局部逐层重构误差目标,优化稀疏+低秩分解
  • 在Llama3-8B上降低12%困惑度,零样本任务差距缩小15%
  • 适合追求高效推理的大模型部署者

大型基础模型的强大能力伴随着高昂的计算成本。压缩预训练模型可降低推理成本,推动其在更广泛社区中的应用。一种有前景的压缩方法是将模型密集权重分解为稀疏与低秩矩阵之和。本文提出名为HASSLE-free的统一框架,用于(半结构化)稀疏加低秩矩阵分解。该框架引入局部逐层重构误差目标,证明先前工作仅求解其松弛问题,并提供高效可扩展的方法来最小化精确优化问题。HASSLE-free在所提目标及多种LLM评估基准上显著优于现有方法。针对采用2:4稀疏性与64秩分解的Llama3-8B模型,其在WikiText-2数据集上测试困惑度降低12%,零样本任务平均性能差距较现有方法减少15%。

原文摘要 · Abstract (English)

The impressive capabilities of large foundation models come at a cost of substantial computing resources to serve them. Compressing these pre-trained models is of practical interest as it can democratize deploying them to the machine learning community at large by lowering the costs associated with inference. A promising compression scheme is to decompose foundation models' dense weights into a sum of sparse plus low-rank matrices. In this paper, we design a unified framework coined HASSLE-free for (semi-structured) sparse plus low-rank matrix decomposition of foundation models. Our framework introduces the local layer-wise reconstruction error objective for this decomposition, we demonstrate that prior work solves a relaxation of this optimization problem; and we provide efficient and scalable methods to minimize the exact introduced optimization problem. HASSLE-free substantially outperforms state-of-the-art methods in terms of the introduced objective and a wide range of LLM evaluation benchmarks. For the Llama3-8B model with a 2:4 sparsity component plus a 64-rank component decomposition, a compression scheme for which recent work shows important inference acceleration on GPUs, HASSLE-free reduces the test perplexity by 12% for the WikiText-2 dataset and reduces the gap (compared to the dense model) of the average of eight popular zero-shot tasks by 15% compared to existing methods.

大模型压缩稀疏分解低秩矩阵LLM推理

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