通过全局优化秩与稀疏性,实现大模型高效压缩。
Large Language Model Compression with Global Rank and Sparsity Optimization
- 分两阶段:先用主成分分析分解权重矩阵,再在低秩与稀疏空间中联合优化。
- 在多个基准上优于现有方法,压缩后模型性能损失小。
- 自动识别层间冗余,协同优化稀疏与低秩结构,适合部署场景。
低秩与稀疏复合近似是压缩大型语言模型的自然思路,但面临两大挑战:低秩与稀疏矩阵间的交互问题,以及不同层间权重分配难以确定,因各层冗余程度差异大。为此,我们提出一种两阶段的新型大模型压缩方法,具备全局秩与稀疏性资源分配能力。由于优化空间巨大,直接全面优化计算成本过高,因此第一阶段利用鲁棒主成分分析(RPCA)将模型权重矩阵分解为低秩与稀疏分量,分别覆盖其低维与稀疏空间。第二阶段提出概率化全局分配策略,在上述两个空间内联合识别低秩与稀疏结构。该方法能自动检测各层冗余,并有效管理稀疏与低秩组件间的交互。大量实验表明,该方法在稀疏化与复合近似方面显著超越当前最优技术。
原文摘要 · Abstract (English)
Low-rank and sparse composite approximation is a natural idea to compress Large Language Models (LLMs). However, such an idea faces two primary challenges that adversely affect the performance of existing methods. The first challenge relates to the interaction and cooperation between low-rank and sparse matrices, while the second involves determining weight allocation across different layers, as redundancy varies considerably among them. To address these challenges, we propose a novel two-stage LLM compression method with the capability of global resource allocation for rank and sparsity. It is noteworthy that the overall optimization space is vast, making comprehensive optimization computationally prohibitive. Therefore, to reduce the optimization space, our first stage utilizes robust principal component analysis to decompose the weight matrices of LLMs into low-rank and sparse components, which span the low dimensional and sparse spaces containing the resultant low-rank and sparse matrices, respectively. In the second stage, we propose a probabilistic global allocation strategy to jointly identify the low-rank and sparse structures within the above two spaces. The appealing feature of our approach is its ability to automatically detect the redundancy across different layers and to manage the interaction between the sparse and low-rank components. Extensive experimental results indicate that our method significantly surpasses state-of-the-art techniques for sparsification and composite approximation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。