arXiv:2509.25622cs.LG2025-09被引 6

针对大模型压缩中各层信息不均的问题,提出分层动态秩分配方法,显著提升压缩后性能。

Layer-wise dynamic rank for compressing large language models

  • 按信息密度为每层动态分配压缩秩,避免均匀压缩导致的性能损失
  • 在20%压缩比下,LLaMA-3-8B模型困惑度降低15以上,40%压缩比下零样本推理准确率提升5%
  • 适用于带分组查询注意力的新一代大模型,兼顾性能与推理速度

大型语言模型(LLMs)规模快速膨胀,带来严重的内存和计算挑战,阻碍其部署。基于奇异值分解(SVD)的压缩技术成为有前景的后训练压缩方案,但现有方法对所有层采用统一压缩比,隐含假设各层信息量一致,忽略了大模型内部显著的层间异质性——中间层信息更丰富,而首尾层冗余度更高。本文重新审视现有SVD压缩方法,提出D-Rank框架,实现分层平衡的动态秩分配。首先引入有效秩作为衡量权重矩阵信息密度的合理指标,并通过拉格朗日乘子优化方案,在固定压缩比下自适应地将更多容量分配给信息密度更高的层组。进一步对注意力层进行重要性重平衡,并扩展至支持分组查询注意力的最新大模型。在多个不同规模的大模型上,于多种压缩比下的大量实验表明,D-Rank持续优于SVD-LLM、ASVD和基共享方法:在C4数据集上,以20%压缩比压缩LLaMA-3-8B时,困惑度降低超过15;以40%压缩比压缩LLaMA-7B时,零样本推理准确率提升达5%,同时获得更高吞吐量。

原文摘要 · Abstract (English)

Large language models (LLMs) have rapidly scaled in size, bringing severe memory and computational challenges that hinder their deployment. Singular Value Decomposition (SVD)-based compression has emerged as an appealing post-training compression technique for LLMs, yet most existing methods apply a uniform compression ratio across all layers, implicitly assuming homogeneous information included in various layers. This overlooks the substantial intra-layer heterogeneity observed in LLMs, where middle layers tend to encode richer information while early and late layers are more redundant. In this work, we revisit the existing SVD-based compression method and propose D-Rank, a framework with layer-wise balanced Dynamic Rank allocation for LLMs compression. We first introduce effective rank as a principled metric to measure the information density of weight matrices, and then allocate ranks via a Lagrange multiplier-based optimization scheme to adaptively assign more capacity to groups with higher information density under a fixed compression ratio. Moreover, we rebalance the allocated ranks across attention layers to account for their varying importance and extend D-Rank to latest LLMs with grouped-query attention. Extensive experiments on various LLMs with different scales across multiple compression ratios demonstrate that D-Rank consistently outperforms SVD-LLM, ASVD, and Basis Sharing, achieving more than 15 lower perplexity with LLaMA-3-8B model on C4 datasets at 20% compression ratio and up to 5% higher zero-shot reasoning accuracy with LLaMA-7B model at 40% compression ratio while achieving even higher throughput.

模型压缩SVD动态秩大模型

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